~tj 2dZddlZddlZddlmZmZddlmZddlm Z ddl m Z  ddl mZddlmZd ZnG#e$r?ejd ed Gd dZGddZGddZdZYnwxYwGddeZGddZdS)z8OpenTelemetry span processor and exporter for LangSmith.N)AnyOptional)urljoin)utils) set_langsmith_metadata_attribute)OTLPSpanExporter)BatchSpanProcessorTwOpenTelemetry packages are not installed. Install optional OpenTelemetry dependencies with: pip install langsmith[otel]) stacklevelceZdZdZdZdS)rzMock otlp span exporter class.c tdzMock init method.r  ImportErrorselfargskwargss c/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/integrations/otel/processor.py__init__zOTLPSpanExporter.__init__. N__name__ __module__ __qualname____doc__rrrrrs),,     rrceZdZdZdZdS)r z Mock batch span processor class.c tdrrrs rrzBatchSpanProcessor.__init__'rrNrrrrr r $s)..     rr c(eZdZdZedZdS)tracezMock trace class.c td)z Mock get tracer provider method.r rrrrget_tracer_providerztrace.get_tracer_provider2s. rN)rrrr staticmethodr%rrrr#r#/s3       rr#Fc ~eZdZdZ ddeedeedeedeeeefffd ZxZS) 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 tjd}|ds|dz }|pt|d}|ptjd}|ptj}|pi}|st d|sd}tjdd|i|}|r||d<||_ tj 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_urlr0exporter_headers __class__s rrzOtelExporter.__init__Hs &'--  %%  OH='(,<==7X1$77:X8::-R :   G L%        <4; 0 1 O(4DOOOOOOOr)NNNN) rrrrrstrdictr __classcell__)r<s@rr(r(>s"!%!%,0 4P4P c]4P#4P# 4P $sCx.) 4P4P4P4P4P4P4P4P4P4Prr(c eZdZdZ ddeedeedeedeeeefdeef dZd eee fd dfd Z dd Z d Z dZ dZddZedZedZdS)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, SpanProcessorcd}|r|dd}t|||||_tst d|t }||j|_i|_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 _metadata)rr*r+r)r,rB exporter_urls rrzOtelSpanProcessor.__init__s*  ?!jjoo>>>L%gw     .   .M'-77)+rmetadatareturnc8||_dS)a"Set metadata attributes to propagate to all spans. Use this to ensure metadata like ``thread_id`` appears on every span in a trace, not just the root span. This is required for LangSmith features (e.g. the threads view) that query runs by metadata fields. N)copyrH)rrJs r set_metadatazOtelSpanProcessor.set_metadatas"rc|jr0|jD]\}}t||||j||dS)z1Forward span start events to the inner processor.N)rHitemsrrGon_start)rspanparent_contextkeyvalues rrQzOtelSpanProcessor.on_startsc > C"n2244 C C U0sEBBBB   ~66666rc:|j|dS)z/Forward span end events to the inner processor.N)rGon_endrrRs rrWzOtelSpanProcessor.on_ends t$$$$$rcht|jdr|j|dSdS)z2Forward span ending events to the inner processor. _on_endingN)hasattrrGrZrXs rrZzOtelSpanProcessor._on_endings< 4?L 1 1 - O & &t , , , , , - -rc8|jdS)zShutdown processor.N)rGshutdownrs rr]zOtelSpanProcessor.shutdowns   """""r0uc6|j|S)z Force flush the inner processor.)rG force_flush)rtimeout_milliss rrazOtelSpanProcessor.force_flushs**>:::rc|jS)zThe underlying OtelExporter.)rEr^s rexporterzOtelSpanProcessor.exporters ~rc|jS)zThe underlying span processor.)rGr^s r processorzOtelSpanProcessor.processors r)NNNNN)N)r_)rrrrrr=r>typerrrNrQrWrZr]rapropertyrdrfrrrrArAsK>"&!%!,0(, ),),#),#),c] ), $sCx.) ),  ~ ),),),),V)T#s(^)))))7777%%%--- ###;;;;XXrrA)rr7warningstypingrr urllib.parser langsmithrr1"langsmith.integrations.otel._utilsr5opentelemetry.exporter.otlp.proto.http.trace_exporterropentelemetry.sdk.trace.exportr rFrwarn UserWarningr#r(rArrrrrs>> ''''''OOOOOO0VVVVVVAAAAAANN+++HM &                          NNNW+\>P>P>P>P>P#>P>P>PBrrrrrrrrrrs5AA98A9