Skip to content

Commit fdc71a3

Browse files
committed
linting fixes
1 parent 11d937b commit fdc71a3

File tree

7 files changed

+16
-16
lines changed

7 files changed

+16
-16
lines changed

util/opentelemetry-util-genai/__init__.py

Lines changed: 0 additions & 1 deletion
This file was deleted.

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

Lines changed: 0 additions & 1 deletion
This file was deleted.

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

Lines changed: 0 additions & 1 deletion
This file was deleted.

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

Lines changed: 0 additions & 1 deletion
This file was deleted.

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,8 @@ class _SpanState:
7474

7575
def _get_property_value(obj: Any, property_name: str) -> Any:
7676
if isinstance(obj, Mapping):
77-
m = cast(Mapping[str, Any], obj)
78-
return m.get(property_name, None)
77+
mapping = cast(Mapping[str, Any], obj)
78+
return mapping.get(property_name)
7979

8080
return cast(Any, getattr(obj, property_name, None))
8181

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

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -156,19 +156,21 @@ def fail_llm(
156156
return invocation
157157

158158

159-
# Singleton accessor
160-
_default_handler: Optional[TelemetryHandler] = None
159+
# Singleton accessor (avoid global statement by storing on function attribute)
161160

162161

163162
def get_telemetry_handler(
164163
emitter_type_full: bool = True, **kwargs: Any
165164
) -> TelemetryHandler:
166-
global _default_handler
167-
if _default_handler is None:
168-
_default_handler = TelemetryHandler(
165+
handler: Optional[TelemetryHandler] = getattr(
166+
get_telemetry_handler, "_default_handler", None
167+
)
168+
if handler is None:
169+
handler = TelemetryHandler(
169170
emitter_type_full=emitter_type_full, **kwargs
170171
)
171-
return _default_handler
172+
setattr(get_telemetry_handler, "_default_handler", handler)
173+
return handler
172174

173175

174176
# Module‐level convenience functions

util/opentelemetry-util-genai/tests/test_utils.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818
)
1919

2020

21-
@pytest.fixture
22-
def telemetry_setup():
21+
@pytest.fixture(name="span_exporter")
22+
def span_exporter_fixture():
2323
"""Set up telemetry providers for testing"""
2424
# Set up in-memory span exporter to capture spans
2525
memory_exporter = InMemorySpanExporter()
@@ -37,8 +37,9 @@ def telemetry_setup():
3737
trace.set_tracer_provider(trace.NoOpTracerProvider())
3838

3939

40+
@pytest.mark.usefixtures("span_exporter")
4041
def test_llm_start_and_stop_creates_span(
41-
telemetry_setup: InMemorySpanExporter,
42+
request: pytest.FixtureRequest,
4243
):
4344
run_id = uuid4()
4445
message = Message(content="hello world", type="Human", name="message name")
@@ -53,7 +54,8 @@ def test_llm_start_and_stop_creates_span(
5354
)
5455

5556
# Get the spans that were created
56-
spans = telemetry_setup.get_finished_spans()
57+
exporter: InMemorySpanExporter = request.getfixturevalue("span_exporter")
58+
spans = exporter.get_finished_spans()
5759

5860
# Verify span was created
5961
assert len(spans) == 1

0 commit comments

Comments
 (0)