Skip to content

Commit 381d38d

Browse files
committed
move utility function to utils
1 parent 5ba84ef commit 381d38d

File tree

2 files changed

+12
-16
lines changed

2 files changed

+12
-16
lines changed

util/opentelemetry-util-genai/src/opentelemetry/util/genai/generators.py

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,6 @@
3838
from uuid import UUID
3939

4040
from opentelemetry import trace
41-
from opentelemetry.instrumentation._semconv import (
42-
_OpenTelemetrySemanticConventionStability,
43-
_OpenTelemetryStabilitySignalType,
44-
_StabilityMode,
45-
)
4641
from opentelemetry.semconv._incubating.attributes import (
4742
gen_ai_attributes as GenAI,
4843
)
@@ -60,6 +55,7 @@
6055
from opentelemetry.util.genai.utils import (
6156
ContentCapturingMode,
6257
get_content_capturing_mode,
58+
is_experimental_mode,
6359
)
6460
from opentelemetry.util.types import AttributeValue
6561

@@ -130,19 +126,10 @@ def _collect_finish_reasons(generations: List[OutputMessage]) -> List[str]:
130126
return finish_reasons
131127

132128

133-
def _is_experimental_mode() -> bool:
134-
return (
135-
_OpenTelemetrySemanticConventionStability._get_opentelemetry_stability_opt_in_mode(
136-
_OpenTelemetryStabilitySignalType.GEN_AI,
137-
)
138-
is _StabilityMode.GEN_AI_LATEST_EXPERIMENTAL
139-
)
140-
141-
142129
def _maybe_set_span_input_messages(
143130
span: Span, messages: List[InputMessage]
144131
) -> None:
145-
if not _is_experimental_mode() or get_content_capturing_mode() not in (
132+
if not is_experimental_mode() or get_content_capturing_mode() not in (
146133
ContentCapturingMode.SPAN_ONLY,
147134
ContentCapturingMode.SPAN_AND_EVENT,
148135
):
@@ -157,7 +144,7 @@ def _maybe_set_span_input_messages(
157144
def _maybe_set_span_output_messages(
158145
span: Span, generations: List[OutputMessage]
159146
) -> None:
160-
if not _is_experimental_mode() or get_content_capturing_mode() not in (
147+
if not is_experimental_mode() or get_content_capturing_mode() not in (
161148
ContentCapturingMode.SPAN_ONLY,
162149
ContentCapturingMode.SPAN_AND_EVENT,
163150
):

util/opentelemetry-util-genai/src/opentelemetry/util/genai/utils.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,15 @@
2828
logger = logging.getLogger(__name__)
2929

3030

31+
def is_experimental_mode() -> bool:
32+
return (
33+
_OpenTelemetrySemanticConventionStability._get_opentelemetry_stability_opt_in_mode(
34+
_OpenTelemetryStabilitySignalType.GEN_AI,
35+
)
36+
is _StabilityMode.GEN_AI_LATEST_EXPERIMENTAL
37+
)
38+
39+
3140
def get_content_capturing_mode() -> ContentCapturingMode:
3241
"""This function should not be called when GEN_AI stability mode is set to DEFAULT.
3342

0 commit comments

Comments
 (0)