o +iC@sdZddlZddlZddlmZddlmZddlmZ zddl m Z ddl m Z dZWn(eyTejd ed d Gd d d Z GdddZ GdddZdZYnwGddde ZGdddZdS)z8OpenTelemetry span processor and exporter for LangSmith.N)Optional)urljoin)utils)OTLPSpanExporter)BatchSpanProcessorTwOpenTelemetry packages are not installed. Install optional OpenTelemetry dependencies with: pip install langsmith[otel]) stacklevelc@eZdZdZddZdS)rzMock otlp span exporter class.cOtdzMock init method.r ImportErrorselfargskwargsrb/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/langsmith/integrations/otel/processor.py__init__zOTLPSpanExporter.__init__N__name__ __module__ __qualname____doc__rrrrrr rc@r )rz Mock batch span processor class.cOr r r rrrrr&rzBatchSpanProcessor.__init__Nrrrrrr#rrc@seZdZdZeddZdS)tracezMock trace class.cCr )z Mock get tracer provider method.rr rrrrget_tracer_provider1sztrace.get_tracer_providerN)rrrr staticmethodrrrrrr.srFc sTeZdZdZ    d deedeedeedeeeefffdd ZZS) OtelExportera&A subclass of `OTLPSpanExporter` configured for LangSmith. Environment Variables: - `LANGSMITH_API_KEY`: Your LangSmith API key. - `LANGSMITH_ENDPOINT`: Base URL for LangSmith API (defaults to `https://api.smith.langchain.com`). - `LANGSMITH_PROJECT`: Project identifier. Nurlapi_keyprojectheadersc  std}|ds|d7}|pt|d}|ptd}|p!t}|p%i}|s,td|s5d}tdd|i|}|rA||d<||_ t j d ||d |dS) aInitialize the `OtelExporter`. Args: url: OTLP endpoint URL. Defaults to `{LANGSMITH_ENDPOINT}/otel/v1/traces`. api_key: LangSmith API key. Defaults to `LANGSMITH_API_KEY` env var. parent: Parent identifier (e.g., `'project_name:test'`). Defaults to `LANGSMITH_PARENT` env var. headers: Additional headers to include in requests. **kwargs: Additional arguments passed to `OTLPSpanExporter`. N/zotel/v1/tracesz`API key is required. Provide it via api_key parameter or LANGSMITH_API_KEY environment variable.defaultzpNo project specified, using default. Configure with LANGSMITH_PROJECT environment variable or project parameter.z x-api-keyzLangsmith-Project)endpointr$r) ls_utils get_api_urlendswithr get_api_keyget_tracer_project ValueErrorlogginginfor#superr) rr!r"r#r$rbase_urlr'exporter_headers __class__rrrGs0   zOtelExporter.__init__)NNNN) rrrrrstrdictr __classcell__rrr3rr =s r c @seZdZdZ     ddeedeedeedeeeefdeef dd Zdd d Z d d Z ddZ dddZ e ddZe ddZdS)OtelSpanProcessora&A span processor for adding LangSmith to OpenTelemetry setups. This class combines the `OtelExporter` and `BatchSpanProcessor` into a single processor that can be added to any `TracerProvider`. Use this when: 1. You already have OpenTelemetry initialized with other tools 2. You want to add LangSmith alongside existing OTEL exporters Examples: # Fresh OpenTelemetry setup (LangSmith only): from langsmith.integrations.otel import configure configure(api_key="your-key", project="your-project") # Add LangSmith to existing OpenTelemetry setup: from opentelemetry import trace from langsmith.integrations.otel.processor import OtelSpanProcessor # Get your existing TracerProvider (already set by other tools) provider = trace.get_tracer_provider() # Add LangSmith processor alongside existing processors langsmith_processor = OtelSpanProcessor( project="your-project", ) provider.add_span_processor(langsmith_processor) Nr"r#r!r$ SpanProcessorcCsRd}|r |dd}t||||d|_tstd|dur!t}||j|_dS)aInitialize the `OtelSpanProcessor`. Args: api_key: LangSmith API key. Defaults to `LANGSMITH_API_KEY` env var. project: Project identifier. Defaults to `LANGSMITH_PROJECT` env var. url: Base URL for LangSmith API. Defaults to `LANGSMITH_ENDPOINT` env var or `https://api.smith.langchain.com`. headers: Additional headers to include in requests. SpanProcessor: Optional span processor class. Defaults to `BatchSpanProcessor`. Nr%z/otel/v1/traces)r!r"r#r$r)rstripr _exporterOTEL_AVAILABLErr _processor)rr"r#r!r$r9 exporter_urlrrrrszOtelSpanProcessor.__init__cCs|j||dS)z1Forward span start events to the inner processor.N)r=on_start)rspanparent_contextrrrr?szOtelSpanProcessor.on_startcCs|j|dS)z/Forward span end events to the inner processor.N)r=on_end)rr@rrrrBszOtelSpanProcessor.on_endcCs|jdS)zShutdown processor.N)r=shutdownrrrrrCszOtelSpanProcessor.shutdown0ucCs |j|S)z Force flush the inner processor.)r= force_flush)rtimeout_millisrrrrFs zOtelSpanProcessor.force_flushcC|jS)zThe underlying OtelExporter.)r;rDrrrexporterzOtelSpanProcessor.exportercCrH)zThe underlying span processor.)r=rDrrr processorrJzOtelSpanProcessor.processor)NNNNN)N)rE)rrrrrr5r6typerr?rBrCrFpropertyrIrKrrrrr8~s4 *  r8)rr.warningstypingr urllib.parser langsmithrr(5opentelemetry.exporter.otlp.proto.http.trace_exporterropentelemetry.sdk.trace.exportrr<rwarn UserWarningrr r8rrrrs.         .A