o ưiD@sddlZddlZddlZddlmZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZmZddlmZdd lmZerSdd lmZeeefZneZd Zd ZGd ddeZdS)N)datetime) TYPE_CHECKINGAnyOptionalUnion)verbose_logger)_utils)LangfuseLLMObsOTELAttributes) OpenTelemetryOpenTelemetryConfig)LangfuseSpanAttributes)StandardCallbackDynamicParams)Spanz*https://cloud.langfuse.com/api/public/otelz-https://us.cloud.langfuse.com/api/public/otelcs eZdZd(fdd ZedefddZededefd d Zeded efd d Z edefddZ edefddZ ede e fddZdefddZed)ddZede de de fddZdede efddZd ed!ede efd"d#Zd$d%Zd&d'ZZS)*LangfuseOtelLoggerNcs*|dur|}tj|d|i|dS)Nconfig)/_create_open_telemetry_config_from_langfuse_envsuper__init__)selfrargskwargs __class__b/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/langfuse/langfuse_otel.pyrszLangfuseOtelLogger.__init__spancCs$t|||ttj|||ddS)z Sets OpenTelemetry span attributes for Langfuse observability. Uses the same attribute setting logic as Arize Phoenix for consistency. rr response_objN)rZset_attributesr r!_set_langfuse_specific_attributesrrrrset_langfuse_otel_attributes&s z/LangfuseOtelLogger.set_langfuse_otel_attributesrreturncCsh|dipi}|dpi}|dust|tsi}zddlm}|||}W|Sty3Y|Sw)a Extracts Langfuse metadata from the standard LiteLLM kwargs structure. 1. Reads kwargs["litellm_params"]["metadata"] if present and is a dict. 2. Enriches it with any `langfuse_*` request-header params via the existing LangFuseLogger.add_metadata_from_header helper so that proxy users get identical behaviour across vanilla and OTEL integrations. litellm_paramsmetadataNr)LangFuseLogger)get isinstancedictZ&litellm.integrations.langfuse.langfuser#Zadd_metadata_from_header Exception)rr!r"Z _LFLoggerrrr_extract_langfuse_metadata7s   z-LangfuseOtelLogger._extract_langfuse_metadatar"c Cs.ddlm}idtjdtjdtjdtjdtjdtjd tj d tj d tj d tj d tj dtjdtjdtjdtjdtjdtj}|D]D\}}||vr||dur||}|d krqt|trq|dd}t|ttfrzt|}Wn tyt|}Ynw|||j|qPdS)z/Helper to set metadata attributes from mapping.rsafe_set_attributeZgeneration_nameZ generation_idZparent_observation_idversionZ mask_inputZ mask_outputZ trace_user_idZ session_idtagsZ trace_nameZtrace_idZtrace_metadataZ trace_versionZ trace_releaseZexisting_trace_idZupdate_trace_keysZdebug_langfuseN-)!litellm.integrations.arize._utilsr*r ZGENERATION_NAMEZ GENERATION_IDZPARENT_OBSERVATION_IDZGENERATION_VERSIONZ MASK_INPUTZ MASK_OUTPUTZ TRACE_USER_IDZ SESSION_IDZTAGSZ TRACE_NAMEZTRACE_IDZTRACE_METADATAZ TRACE_VERSIONZ TRACE_RELEASEZEXISTING_TRACE_IDZUPDATE_TRACE_KEYSZDEBUG_LANGFUSEitemsr%strreplacelistr&jsondumpsr'value)rr"r*mappingkeyZ enum_attrr6rrr_set_metadata_attributesTsd         z+LangfuseOtelLogger._set_metadata_attributesc Cs~ddlm}ddlm}|rt|dsdS|dg}|r|d}|di}|d}|rg}|D]B} | d i} | d d } zt| trKt | n| } Wn tj yZi} Ynw|d d | dd | d d d| d} | | q2||t j j||n(i}|dr|d|d<|ddur|d|d<|r||t j j|||dg}|r;g}|D]v}t|dr+|j}|dkrt|dr|jD]}t|dr| d|jdqq|dkr| t|ddtt|digddd dq|dkr+t|d d } t| trt | n| } t|d d t|dd t|dd d| d} | | q|r=||t j j||dSdSdS)z,Helper to set observation output attributes.rr) safe_dumpsr$Nchoicesmessage tool_callsfunction argumentsz{}idr.nameZ function_call)rArBcall_idtyper@rolecontentoutputrDZ reasoningsummarytextZreasoning_summary)rErFZ assistantrC)r/r**litellm.litellm_core_utils.safe_json_dumpsr;hasattrr$r%r1r4loadsJSONDecodeErrorappendr ZOBSERVATION_OUTPUTr6rDrHrIgetattr)rrr*r;r<Z first_choicer=r>Ztransformed_tool_callsZ tool_callr?Z arguments_strZ arguments_objZlangfuse_tool_callZ output_datarGZoutput_items_dataitemZ item_typerHrrr_set_observation_output{s                        &z*LangfuseOtelLogger._set_observation_outputcCsddlm}ddlm}tjd}|r||tjj |t |}t j ||d|d}|r9||tj j ||t j||ddS) aH Sets Langfuse specific metadata attributes onto the OTEL span. All keys supported by the vanilla Langfuse integration are mapped to OTEL-safe attribute names defined in LangfuseSpanAttributes. Complex values (lists/dicts) are serialised to JSON strings for OTEL compatibility. rr)r:ZLANGFUSE_TRACING_ENVIRONMENT)rr"messages)rrN)r/r*rJr;osenvironr$r ZLANGFUSE_ENVIRONMENTr6rr(r9ZOBSERVATION_INPUTrQ)rrrr*r;Zlangfuse_environmentr"rRrrrrs&    z4LangfuseOtelLogger._set_langfuse_specific_attributescCstjdp tjdS)z Returns the Langfuse OTEL host based on environment variables. Returned in the following order of precedence: 1. LANGFUSE_OTEL_HOST 2. LANGFUSE_HOST ZLANGFUSE_OTEL_HOSTZ LANGFUSE_HOST)rSrTr$rrrr_get_langfuse_otel_hosts z*LangfuseOtelLogger._get_langfuse_otel_hostcCsddlm}tjdd}tjdd}|r|s|St}|r<|ds+d|}| dd }t d |n t }t d |tj ||d }d |}|d||dS)z Creates OpenTelemetryConfig from Langfuse environment variables. Does NOT modify global environment variables. r)r LANGFUSE_PUBLIC_KEYNLANGFUSE_SECRET_KEYhttphttps:////api/public/otel(Using Langfuse OTEL endpoint from host: "Using Langfuse US cloud endpoint:  public_key secret_keyAuthorization= otlp_httpZexporterendpointheaders)"litellm.integrations.opentelemetryr rSrTr$Zfrom_envrrU startswithrstriprdebugLANGFUSE_CLOUD_US_ENDPOINT"_get_langfuse_authorization_header)rr r_r` langfuse_hostrd auth_headerotlp_auth_headersrrrr s,   zBLangfuseOtelLogger._create_open_telemetry_config_from_langfuse_envr cCstjdd}tjdd}|r|stdt}|r6|ds%d|}|dd}t d |n t }t d |tj ||d }d |}t d ||dS)a2 Retrieves the Langfuse OpenTelemetry configuration based on environment variables. Environment Variables: LANGFUSE_PUBLIC_KEY: Required. Langfuse public key for authentication. LANGFUSE_SECRET_KEY: Required. Langfuse secret key for authentication. LANGFUSE_HOST: Optional. Custom Langfuse host URL. Defaults to US cloud. Returns: OpenTelemetryConfig: A Pydantic model containing Langfuse OTEL configuration. Raises: ValueError: If required keys are missing. rVNrWz_LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY must be set for Langfuse OpenTelemetry integration.rXrYrZr[r\r]r^rarbrc) rSrTr$ ValueErrorrrUrgrhrrirjrkr )r_r`rlrdrmrnrrrget_langfuse_otel_config2s.  z+LangfuseOtelLogger.get_langfuse_otel_configr_r`cCs*|d|}t|}d|S)zJ Get the authorization header for Langfuse OpenTelemetry. :zBasic )base64 b64encodeencodedecode)r_r`Z auth_stringrmrrrrkgs z5LangfuseOtelLogger._get_langfuse_authorization_header standard_callback_dynamic_paramscCs:i}|d}|d}|r|rtj||d}||d<|S)z Construct dynamic Langfuse headers from standard callback dynamic params This is used for team/key based logging. Returns: dict: A dictionary of dynamic Langfuse headers Zlangfuse_public_keyZlangfuse_secret_keyr^ Authorization)r$rrk)rrvZdynamic_headersZdynamic_langfuse_public_keyZdynamic_langfuse_secret_keyrmrrrconstruct_dynamic_otel_headersps z1LangfuseOtelLogger.construct_dynamic_otel_headers start_timerecCsdS)aq Override to prevent creating empty proxy request spans. Langfuse should only receive spans for actual LLM calls, not for internal proxy operations (auth, postgres, proxy_pre_call, etc.). By returning None, we prevent the parent span from being created, which in turn prevents empty traces from being sent to Langfuse. Nr)rryrerrr)create_litellm_proxy_request_started_spanszs$