3838from uuid import UUID
3939
4040from opentelemetry import trace
41- from opentelemetry .instrumentation ._semconv import (
42- _OpenTelemetrySemanticConventionStability ,
43- _OpenTelemetryStabilitySignalType ,
44- _StabilityMode ,
45- )
4641from opentelemetry .semconv ._incubating .attributes import (
4742 gen_ai_attributes as GenAI ,
4843)
6055from opentelemetry .util .genai .utils import (
6156 ContentCapturingMode ,
6257 get_content_capturing_mode ,
58+ is_experimental_mode ,
6359)
6460from 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-
142129def _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(
157144def _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 ):
0 commit comments