From 999dc006da32793d3c6002b8f1b2a956087aec59 Mon Sep 17 00:00:00 2001 From: Jaroslav Bachorik Date: Tue, 2 Jun 2026 14:05:57 +0200 Subject: [PATCH] feat: add llm phase profiling feature flag --- .../com/datadog/profiling/ddprof/DatadogProfilerConfig.java | 5 +++++ .../main/java/datadog/trace/api/config/ProfilingConfig.java | 3 +++ 2 files changed, 8 insertions(+) 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;