tPj0>UddlmZddlZddlZddlZddlmZmZmZddl m Z m Z ddl m Z ddlmZddlmZddlmZdd lmZmZdd lmZmZdd lmZdd lmZmZdd l m!Z!erddl m"Z"dZ#de$d<dZ%de$d<GddeZ&ddZ'ddZ(ddZ)ddZ*GddeZ+y)) annotationsN) TYPE_CHECKINGAnyFinal)Status StatusCode)override)verbose_logger)SpanAttributes)_utils) OpenTelemetryOpenTelemetryConfig)BaseLLMObsOTELAttributessafe_set_attribute) safe_dumps)WeaveOtelConfigWeaveSpanAttributes)StandardCallbackDynamicParams)Spanzhttps://trace.wandb.airWEAVE_BASE_URLz/otel/v1/tracesWEAVE_OTEL_ENDPOINTc,eZdZdZeeddZy)WeaveLLMObsOTELAttributesz Weave-specific LLM observability OTEL attributes. Weave automatically maps attributes from multiple frameworks including GenAI, OpenInference, Langfuse, and others. c|jdxsg}|jdxsi}d|i}|jd}|jd}|||d<|||d<t|tjt j |y)zFSet input messages as span attributes using OpenInference conventions.messagesoptional_params functionstoolsN)getrOpenInferenceSpanAttributes INPUT_VALUEjsondumps)spankwargsrrpromptrrs j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/weave/weave_otel.py set_messagesz&WeaveLLMObsOTELAttributes.set_messages,s !**Z06B!',=!>!D"#X.*..{; &**73  "+F;   #F7O4!sh#JJ'78>BNjj,2H::g&,"E!/!3!34I!J!Pb<</L E 12!E7;L L#++C6 4!4!A!A!G!GVll<00 = j4, /#J/J4!4!>!>!D!DjQ4!4! zapi:zBasic )base64 b64encodeencodedecode)api_key auth_string auth_headers r'_get_weave_authorization_headerrQls?  y)K))+*<*<*>?FFHK K= !!r)ctjd}tjd}tjd}|s td|s td|rE|jdsd|z}|j dt z}t jd |n#tt z}t jd |t| }d |d |}|tjd<|tjd<t|||dS)a Retrieves the Weave OpenTelemetry configuration based on environment variables. Environment Variables: WANDB_API_KEY: Required. W&B API key for authentication. WANDB_PROJECT_ID: Required. Project ID in format /. WANDB_HOST: Optional. Custom Weave host URL. Defaults to cloud endpoint. Returns: WeaveOtelConfig: A Pydantic model containing Weave OTEL configuration. Raises: ValueError: If required environment variables are missing. WANDB_API_KEYWANDB_PROJECT_ID WANDB_HOSTz>WANDB_API_KEY must be set for Weave OpenTelemetry integration.zaWANDB_PROJECT_ID must be set for Weave OpenTelemetry integration. Format: /httpzhttps://r8z'Using Weave OTEL endpoint from host: %szUsing Weave cloud endpoint: %srNzAuthorization=z ,project_id=OTEL_EXPORTER_OTLP_ENDPOINTOTEL_EXPORTER_OTLP_HEADERS otlp_http)otlp_auth_headersendpoint project_idprotocol) osgetenv ValueError startswithrstriprr debugrrQenvironr)rNr]hostr\rPr[s r'get_weave_otel_configrgwsYY/G "45J 99\ "D YZZ  o   v&$D;;s#&99FQ!$77=xH9IK!/ }L U19BJJ,-/@BJJ+, +  r)cXtj|||tt|||y)z Sets OpenTelemetry span attributes for Weave observability. Uses the same attribute setting logic as other OTEL integrations for consistency. r$r%rFN)r set_attributesrrHris r'set_weave_otel_attributesrks%  $ 6OP"V,Wr)cVeZdZdZ d dfd ZdZ d dZdZ d dZxZ S) WeaveOtelLoggera Weave (W&B) OpenTelemetry Logger for LiteLLM. Sends LLM traces to Weave via the OpenTelemetry Protocol (OTLP). Environment Variables: WANDB_API_KEY: Required. Weights & Biases API key for authentication. WANDB_PROJECT_ID: Required. Project ID in format /. WANDB_HOST: Optional. Custom Weave host URL. Defaults to cloud endpoint. Usage: litellm.callbacks = ["weave_otel"] Or manually: from litellm.integrations.weave.weave_otel import WeaveOtelLogger weave_logger = WeaveOtelLogger(callback_name="weave_otel") litellm.callbacks = [weave_logger] Reference: https://docs.wandb.ai/weave/guides/tracking/otel c |6t}t|j|j|j}t |d||d|y)z Initialize WeaveOtelLogger. If config is not provided, automatically configures from environment variables (WANDB_API_KEY, WANDB_PROJECT_ID, WANDB_HOST) via get_weave_otel_config(). N)exporterr\headers)config callback_namer0)rgrr^r\r[super__init__)selfrqrrr% weave_config __class__s r'rtzWeaveOtelLogger.__init__sP >"7"9L(%..%..$66F NmNvNr)cy)z Override to skip creating the raw_gen_ai_request child span. For Weave, we only want a single span per LLM call. The parent span already contains all the necessary attributes, so the child span is redundant. Nr0)rur%rF start_timeend_time parent_spans r'_maybe_log_raw_requestz&WeaveOtelLogger._maybe_log_raw_requestsr)cD|j|}|j|j||j||}|j t t j|j||||j|j||S)z Override to always create a child span instead of reusing the parent span. This ensures that wrapper spans (like "B", "C", "D", "E") remain separate from the LiteLLM LLM call spans, creating proper nesting in Weave. )namerycontext)rz) get_tracer_to_use_for_request start_span_get_span_name_to_ns set_statusrrOKrjend) rur%rFryrzrr{ otel_tracerr$s r'_start_primary_spanz#WeaveOtelLogger._start_primary_spans "??G ",,$$V,{{:.-  z}}-. D&,7 $++h/0 r)cjtjd||j|j|\}}d}|j ||||||}|j ||||||j |||j|||||jjr|j|||yy)z Override to prevent ending externally created parent spans. When wrapper spans (like "B", "C", "D", "E") are provided as parent spans, they should be managed by the user code, not ended by LiteLLM. zGWeave OpenTelemetry Logger: Logging kwargs: %s, OTEL config settings=%sN)r%r) r rdrq_get_span_contextrr|_create_guardrail_span_record_metrics enable_events_emit_semantic_logs) rur%rFryrzctxr{primary_span_parentr$s r'_handle_successzWeaveOtelLogger._handle_success s  U  KK  11&9[&*..v|ZQY[^`st ##FL*hPTU ##63#? V\:xH ;; $ $  $ $V\4 @ %r)ci}|jd}|jd}|rt|}||d<|r||d<|r|SdS)z Construct dynamic Weave headers from standard callback dynamic params. This is used for team/key based logging. Returns: dict: A dictionary of dynamic Weave headers wandb_api_keyweave_project_idrW Authorizationr]N)rrQ)ru standard_callback_dynamic_paramsdynamic_headersdynamic_wandb_api_keydynamic_weave_project_idrPs r'construct_dynamic_otel_headersz.WeaveOtelLogger.construct_dynamic_otel_headers1sb"$'G'K'KO'\*J*N*NOa*b !@-"K0;OO , #,DOL )"1;t;r))N weave_otel)rqzOpenTelemetryConfig | Nonerrz str | None)N)rrreturnz dict | None) r+r,r-r.rtr|rrr __classcell__)rws@r'rmrmsU0.2$0O*O"O0 : AN<0M< rs" ,,2&+.QBV=( 10.U.^ 8^4+h\"2jXU