jKj ~dZddlmZddlmZddlmZmZmZddl m Z ddl m Z ddl mZd d lmZmZGd d eZy )zPostHog span processor for OpenTelemetry. Provides a self-contained SpanProcessor that filters AI-related spans and exports them to PostHog's OTLP endpoint. This is the recommended integration for setups using TracerProvider.add_span_processor(). )Optional)Context) ReadableSpanSpan SpanProcessor)BatchSpanProcessor)OTLPSpanExporter)*warn_if_posthog_ai_gateway_otel_attributes) DEFAULT_HOST is_ai_spancpeZdZdZefdedefdZddedee ddfd Z de ddfd Z dd Z dd eedefd Zy)PostHogSpanProcessora+Span processor that filters AI spans and exports them to PostHog. Wraps a BatchSpanProcessor and OTLPSpanExporter internally, configured to send to PostHog's OTLP traces endpoint. Only spans identified as AI-related (by name or attribute prefix) are forwarded for export. Usage:: from opentelemetry.sdk.trace import TracerProvider from posthog.ai.otel import PostHogSpanProcessor provider = TracerProvider() provider.add_span_processor( PostHogSpanProcessor(api_key="phc_...") ) api_keyhostc||_|jd|_t|jddd|ji}t ||_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 r _processor)selfrrexporters j/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/otel/processor.py__init__zPostHogSpanProcessor.__init__%sT  [[% # |=1$ &?@ -X6Nspanparent_contextreturncy)z Handle span start notifications. This processor does not need to do work at span start; filtering happens in ``on_end``. N)rr!r"s ron_startzPostHogSpanProcessor.on_start8s r c|t|syt|j|jj |y)zz Export an ended span if it is AI-related. Args: span: The ended OpenTelemetry span. N)rr attributesron_end)rr!s rr)zPostHogSpanProcessor.on_endAs-$ 24??C t$r c8|jjy)z.Shut down the underlying batch span processor.N)rshutdown)rs rr+zPostHogSpanProcessor.shutdownMs   "r timeout_milliscp||jj|S|jjS)a Flush pending spans from the underlying batch span processor. 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 PostHogSpanProcessor.force_flushQs2  %??..~> >**,,r )N)r#N)__name__ __module__ __qualname____doc__r strrrrrr&rr)r+intboolr.r%r rrrsr(!777& T 8G3D PT  %< %D %# -(3- -4 -r rN)r2typingropentelemetry.contextropentelemetry.sdk.tracerrropentelemetry.sdk.trace.exportr5opentelemetry.exporter.otlp.proto.http.trace_exporterr gatewayr spansr rrr%r rr=s2)EE=R@+J-=J-r