diff --git a/dd-java-agent/agent-profiling/profiling-ddprof/src/main/java/com/datadog/profiling/ddprof/DatadogProfilerConfig.java b/dd-java-agent/agent-profiling/profiling-ddprof/src/main/java/com/datadog/profiling/ddprof/DatadogProfilerConfig.java index 60c0d5c2e60..3fba25def6b 100644 --- a/dd-java-agent/agent-profiling/profiling-ddprof/src/main/java/com/datadog/profiling/ddprof/DatadogProfilerConfig.java +++ b/dd-java-agent/agent-profiling/profiling-ddprof/src/main/java/com/datadog/profiling/ddprof/DatadogProfilerConfig.java @@ -4,6 +4,7 @@ import static datadog.environment.JavaVirtualMachine.isJ9; import static datadog.trace.api.config.ProfilingConfig.PROFILING_ALLOCATION_ENABLED; import static datadog.trace.api.config.ProfilingConfig.PROFILING_CONTEXT_ATTRIBUTES; +import static datadog.trace.api.config.ProfilingConfig.PROFILING_CONTEXT_ATTRIBUTES_LLM_PHASE_ENABLED; import static datadog.trace.api.config.ProfilingConfig.PROFILING_CONTEXT_ATTRIBUTES_RESOURCE_NAME_ENABLED; import static datadog.trace.api.config.ProfilingConfig.PROFILING_CONTEXT_ATTRIBUTES_SPAN_NAME_ENABLED; import static datadog.trace.api.config.ProfilingConfig.PROFILING_DATADOG_PROFILER_ALLOC_ENABLED; @@ -396,6 +397,10 @@ public static boolean isResourceNameContextAttributeEnabled(ConfigProvider confi return configProvider.getBoolean(PROFILING_CONTEXT_ATTRIBUTES_RESOURCE_NAME_ENABLED, false); } + public static boolean isLlmPhaseAttributeEnabled(ConfigProvider configProvider) { + return configProvider.getBoolean(PROFILING_CONTEXT_ATTRIBUTES_LLM_PHASE_ENABLED, true); + } + public static boolean isTrackingGenerations(ConfigProvider configProvider) { return getBoolean( configProvider, PROFILING_HEAP_TRACK_GENERATIONS, PROFILING_HEAP_TRACK_GENERATIONS_DEFAULT); diff --git a/dd-trace-api/src/main/java/datadog/trace/api/config/ProfilingConfig.java b/dd-trace-api/src/main/java/datadog/trace/api/config/ProfilingConfig.java index 4076f4aae30..8fa4f3e7513 100644 --- a/dd-trace-api/src/main/java/datadog/trace/api/config/ProfilingConfig.java +++ b/dd-trace-api/src/main/java/datadog/trace/api/config/ProfilingConfig.java @@ -223,6 +223,9 @@ public final class ProfilingConfig { public static final String PROFILING_CONTEXT_ATTRIBUTES_RESOURCE_NAME_ENABLED = "profiling.context.attributes.resource.name.enabled"; + public static final String PROFILING_CONTEXT_ATTRIBUTES_LLM_PHASE_ENABLED = + "profiling.context.attributes.llm.phase.enabled"; + public static final String PROFILING_QUEUEING_TIME_ENABLED = "profiling.queueing.time.enabled"; public static final boolean PROFILING_QUEUEING_TIME_ENABLED_DEFAULT = true;