Instrumentation()

from respan import Instrumentation

class Instrumentation(Protocol):
    name: str

    def activate(self) -> None:
        """Start intercepting spans and injecting them into the OTEL pipeline."""
        ...

    def deactivate(self) -> None:
        """Stop intercepting spans and clean up resources."""
        ...

from respan import Respan
from respan_tracing.utils.span_factory import build_readable_span, inject_span

class MySDKInstrumentor:
    name = "my-sdk"

    def activate(self) -> None:
        # Register a callback with the vendor SDK
        import my_sdk
        my_sdk.on_completion(self._on_completion)

    def deactivate(self) -> None:
        import my_sdk
        my_sdk.remove_callback(self._on_completion)

    def _on_completion(self, event):
        span = build_readable_span(
            name=f"my-sdk.completion",
            trace_id=event.trace_id,
            attributes={
                "llm.request.type": "chat",
                "gen_ai.request.model": event.model,
                "gen_ai.usage.prompt_tokens": event.input_tokens,
                "gen_ai.usage.completion_tokens": event.output_tokens,
                "traceloop.span.kind": "task",
                "traceloop.entity.name": "completion",
            },
        )
        inject_span(span)

# Use it
respan = Respan(
    api_key="your-api-key",
    instrumentations=[MySDKInstrumentor()],
)