fix unsorted errors

Author: inference-stack-llc

src/electripy/observability/ai_telemetry/__init__.py

@@ -21,7 +21,7 @@
 
 from __future__ import annotations
 
-from .adapters import JsonlTelemetrySinkAdapter, InMemoryTelemetryAdapter, OpenTelemetryAdapter
+from .adapters import InMemoryTelemetryAdapter, JsonlTelemetrySinkAdapter, OpenTelemetryAdapter
 from .domain import (
     CostRecord,
     HistogramObservation,

src/electripy/observability/ai_telemetry/adapters.py

@@ -18,11 +18,12 @@
 
 import json
 import threading
+from collections.abc import Callable
 from dataclasses import asdict, dataclass
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from hashlib import sha256
 from pathlib import Path
-from typing import Any, Callable
+from typing import Any
 
 from electripy.core.logging import get_logger
 
@@ -113,7 +114,7 @@ def __init__(
         self._previous_ctx: TelemetryContext | None = None
 
     def __enter__(self) -> TelemetryContext:
-        now = datetime.now(tz=timezone.utc)
+        now = datetime.now(tz=UTC)
         base_ctx = self._ctx or self._get_current()
         if base_ctx is None:
             # Minimal context; caller can create richer contexts via
@@ -147,7 +148,7 @@ def __enter__(self) -> TelemetryContext:
         return ctx
 
     def __exit__(self, exc_type, exc, tb) -> None:  # type: ignore[override]
-        end = datetime.now(tz=timezone.utc)
+        end = datetime.now(tz=UTC)
         ctx = self._get_current()
         if self._start_time is not None and ctx is not None:
             duration_ms = (end - self._start_time).total_seconds() * 1000.0
@@ -200,7 +201,7 @@ def emit_event(self, event: TelemetryEvent) -> None:
         record = {
             "type": "event",
             "name": event.name,
-            "timestamp": event.timestamp.astimezone(timezone.utc).isoformat(),
+            "timestamp": event.timestamp.astimezone(UTC).isoformat(),
             "severity": event.severity.value,
             "context": asdict(event.context),
             "attributes": _sanitize_attributes(dict(event.attributes)),
@@ -215,7 +216,7 @@ def increment(
         attrs: dict[str, object] | None = None,
         ctx: TelemetryContext | None = None,
     ) -> None:
-        now = datetime.now(tz=timezone.utc)
+        now = datetime.now(tz=UTC)
         record = CounterIncrement(
             name=name,
             value=value,
@@ -236,7 +237,7 @@ def observe(
         attrs: dict[str, object] | None = None,
         ctx: TelemetryContext | None = None,
     ) -> None:
-        now = datetime.now(tz=timezone.utc)
+        now = datetime.now(tz=UTC)
         record = HistogramObservation(
             name=name,
             value=value,
@@ -309,7 +310,7 @@ def increment(
             value=value,
             attributes=_sanitize_attributes(attrs or {}),
             context=ctx,
-            timestamp=datetime.now(tz=timezone.utc),
+            timestamp=datetime.now(tz=UTC),
         )
         self.counters.append(record)
 
@@ -326,7 +327,7 @@ def observe(
             value=value,
             attributes=_sanitize_attributes(attrs or {}),
             context=ctx,
-            timestamp=datetime.now(tz=timezone.utc),
+            timestamp=datetime.now(tz=UTC),
         )
         self.histograms.append(record)
 
@@ -396,7 +397,7 @@ def increment(
     ) -> None:  # pragma: no cover - simple mapping
         # For now, model counters as events; a production deployment can
         # attach these to a metrics exporter.
-        now = datetime.now(tz=timezone.utc)
+        now = datetime.now(tz=UTC)
         event = TelemetryEvent(
             name=f"metric.counter.{name}",
             timestamp=now,
@@ -424,7 +425,7 @@ def observe(
         attrs: dict[str, object] | None = None,
         ctx: TelemetryContext | None = None,
     ) -> None:  # pragma: no cover - simple mapping
-        now = datetime.now(tz=timezone.utc)
+        now = datetime.now(tz=UTC)
         event = TelemetryEvent(
             name=f"metric.histogram.{name}",
             timestamp=now,
@@ -456,7 +457,7 @@ def span(
         # not automatically mapped; callers can still emit explicit
         # events if they need a bridge.
         class _OtelSpan(SpanContextManager):
-            def __init__(self, adapter: "OpenTelemetryAdapter") -> None:
+            def __init__(self, adapter: OpenTelemetryAdapter) -> None:
                 self._adapter = adapter
                 self._cm = adapter._tracer.start_as_current_span(name)  # type: ignore[union-attr]
 

src/electripy/observability/ai_telemetry/domain.py

@@ -31,10 +31,11 @@
 
 from __future__ import annotations
 
+from collections.abc import MutableMapping
 from dataclasses import dataclass, field
 from datetime import datetime
 from enum import StrEnum
-from typing import Mapping, MutableMapping, TypeAlias
+from typing import TypeAlias
 
 AttributeValue: TypeAlias = str | int | float | bool | None
 Attributes: TypeAlias = MutableMapping[str, AttributeValue]
@@ -95,7 +96,7 @@ class TelemetryContext:
     environment: str | None
     tags: dict[str, str] = field(default_factory=dict)
 
-    def child(self, *, span_id: str) -> "TelemetryContext":
+    def child(self, *, span_id: str) -> TelemetryContext:
         """Return a child context for a nested span.
 
         Args:

src/electripy/observability/ai_telemetry/services.py

@@ -13,10 +13,10 @@
 
 import contextlib
 import os
+from collections.abc import Iterator, Mapping, MutableMapping
 from contextvars import ContextVar
 from dataclasses import dataclass
-from datetime import datetime, timezone
-from typing import Iterator, Mapping, MutableMapping
+from datetime import UTC, datetime
 
 from electripy.core.logging import get_logger
 
@@ -209,7 +209,7 @@ def record_http_retry_attempt(
 
     event = TelemetryEvent(
         name="http.retry_attempt",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=ctx or current_telemetry_context() or create_telemetry_context(),
         attributes={
             "attempt": attempt,
@@ -232,7 +232,7 @@ def record_http_circuit_opened(
 
     event = TelemetryEvent(
         name="http.breaker_opened",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=ctx or current_telemetry_context() or create_telemetry_context(),
         attributes={"url": url},
         severity=Severity.WARNING,
@@ -268,7 +268,7 @@ def record_llm_call(
     )
     event = TelemetryEvent(
         name="llm.call",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=context,
         attributes={
             "provider": provider,
@@ -295,7 +295,7 @@ def record_policy_decision(
 
     event = TelemetryEvent(
         name="policy.decision",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=ctx or current_telemetry_context() or create_telemetry_context(),
         attributes={
             "decision": decision,
@@ -317,7 +317,7 @@ def record_rag_experiment_started(
 
     event = TelemetryEvent(
         name="rag_eval.experiment_started",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=ctx or current_telemetry_context() or create_telemetry_context(),
         attributes={"experiment_id": experiment_id},
         severity=Severity.INFO,
@@ -336,7 +336,7 @@ def record_rag_experiment_finished(
 
     event = TelemetryEvent(
         name="rag_eval.experiment_finished",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=ctx or current_telemetry_context() or create_telemetry_context(),
         attributes={"experiment_id": experiment_id, **dict(metrics_summary)},
         severity=Severity.INFO,

tests/observability/ai_telemetry/test_ai_telemetry.py

@@ -1,9 +1,8 @@
 from __future__ import annotations
 
-from datetime import datetime, timezone
-from pathlib import Path
-
 import json
+from datetime import UTC, datetime
+from pathlib import Path
 
 from electripy.observability.ai_telemetry import (
     InMemoryTelemetryAdapter,
@@ -61,7 +60,7 @@ def test_jsonl_adapter_writes_events(tmp_path: Path) -> None:
 
     event = TelemetryEvent(
         name="example",
-        timestamp=datetime.now(tz=timezone.utc),
+        timestamp=datetime.now(tz=UTC),
         context=ctx,
         attributes={"provider": "fake"},
         severity=Severity.INFO,