jKj rdZddlmZmZddlmZddlmZmZddl m Z ddl m Z dd l mZmZGd d eZy ) zPostHog trace exporter for OpenTelemetry. Provides a SpanExporter that filters AI-related spans before forwarding them to PostHog's OTLP endpoint. Use this when your setup only accepts a SpanExporter (e.g. as an argument to BatchSpanProcessor). )OptionalSequence) ReadableSpan) SpanExporterSpanExportResult)OTLPSpanExporter)*warn_if_posthog_ai_gateway_otel_attributes) DEFAULT_HOST is_ai_spancZeZdZdZefdedefdZdeede fdZ d d Z d d e e defd Zy)PostHogTraceExporterafSpan exporter that filters AI spans and forwards them to PostHog. Wraps an OTLPSpanExporter configured for PostHog's OTLP endpoint. Spans that are not AI-related are silently dropped, returning SUCCESS immediately. Usage:: from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import BatchSpanProcessor from posthog.ai.otel import PostHogTraceExporter provider = TracerProvider() provider.add_span_processor( BatchSpanProcessor( PostHogTraceExporter(api_key="phc_...") ) ) api_keyhostc||_|jd|_t|jddd|ji|_y)z Args: api_key: PostHog project API key. host: PostHog host URL. Defaults to US cloud. /z /i/v0/ai/otel AuthorizationzBearer )endpointheadersN)_api_keyrstrip_hostr _exporter)selfrrs i/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/otel/exporter.py__init__zPostHogTraceExporter.__init__&sH  [[% ) |=1$ &?@ spansreturnc|Dcgc]}t|s|}}|stjS|D]}t|j|j j |Scc}w)z Export AI-related spans to PostHog and drop non-AI spans. Args: spans: Readable OpenTelemetry spans to filter and export. Returns: The OpenTelemetry export result. )r rSUCCESSr attributesrexport)rrspanai_spanss rr$zPostHogTraceExporter.export8s_&+?UTj.>DU?#++ +D 6t G~~$$X.. @s A%A%Nc8|jjy)z'Shut down the underlying OTLP exporter.N)rshutdown)rs rr(zPostHogTraceExporter.shutdownIs !rtimeout_milliscp||jj|S|jjS)z Flush pending spans from the underlying OTLP exporter. Args: timeout_millis: Optional flush timeout in milliseconds. Returns: True if the flush succeeded within the timeout, False otherwise. )r force_flush)rr)s rr+z PostHogTraceExporter.force_flushMs2  %>>--n= =~~))++r)r N)N)__name__ __module__ __qualname____doc__r strrrrrr$r(rintboolr+rrrrsX,!   $/H\2/7G/"" ,(3- ,4 ,rrN)r/typingrropentelemetry.sdk.traceropentelemetry.sdk.trace.exportrr5opentelemetry.exporter.otlp.proto.http.trace_exporterrgatewayr rr r rr3rrr9s,&0IR@+G,<G,r