h'ddlZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z mZerddlmZddlmZdd lmZeZeZeeefZneZeZeZd Zd ZGd d Zy)N) TYPE_CHECKINGAnyUnion)quote)verbose_logger)_utils)LangfuseOtelConfigLangfuseSpanAttributes)Span)OpenTelemetryConfig)Protocolz*https://cloud.langfuse.com/api/public/otelz-https://us.cloud.langfuse.com/api/public/otelcNeZdZedefdZedefdZedefdZy)LangfuseOtelLoggerspanc`tj|||tj||y)z Sets OpenTelemetry span attributes for Langfuse observability. Uses the same attribute setting logic as Arize Phoenix for consistency. )rkwargsN)rset_attributesr!_set_langfuse_specific_attributes)rr response_objs g/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/integrations/langfuse/langfuse_otel.pyset_langfuse_otel_attributesz/LangfuseOtelLogger.set_langfuse_otel_attributes$s6 dFL9 << =  cddlm}tjj dd}|r$||t j j|yy)z@ Sets Langfuse specific attributes to the span. r)safe_set_attributeLANGFUSE_TRACING_ENVIRONMENTN)rkeyvalue)!litellm.integrations.arize._utilsrosenvirongetr LANGFUSE_ENVIRONMENTr)rrrlangfuse_environments rrz4LangfuseOtelLogger._set_langfuse_specific_attributes5sC I!zz~~.LdS  *??EE*  rreturnctjjdd}tjjdd}|r|s tdtjjdd}|rC|j dsd|z}|j dd }t jd |nt}t jd ||d |}tj|jj}d td|}|tjd<|tjd<t|dS)a1 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: LangfuseOtelConfig: A Pydantic model containing Langfuse OTEL configuration. Raises: ValueError: If required keys are missing. LANGFUSE_PUBLIC_KEYNLANGFUSE_SECRET_KEYz_LANGFUSE_PUBLIC_KEY and LANGFUSE_SECRET_KEY must be set for Langfuse OpenTelemetry integration. LANGFUSE_HOSThttpzhttps:///z/api/public/otelz(Using Langfuse OTEL endpoint from host: z"Using Langfuse US cloud endpoint: :zAuthorization=zBasic OTEL_EXPORTER_OTLP_ENDPOINTOTEL_EXPORTER_OTLP_HEADERS otlp_http)otlp_auth_headersprotocol)rr r! ValueError startswithrstriprdebugLANGFUSE_CLOUD_US_ENDPOINTbase64 b64encodeencodedecoderr ) public_key secret_key langfuse_hostendpoint auth_string auth_headerr/s rget_langfuse_otel_configz+LangfuseOtelLogger.get_langfuse_otel_configCs@ ZZ^^$94@ ZZ^^$94@ q   =  ++F3 *] : '..s344DEH  #KH:!V W2H  #EhZ!P Q$ Aj\2 &&{'9'9';<CCE ,UVK=3I-J,KL5= 013D /0!/+  rN) __name__ __module__ __qualname__ staticmethodr rrr r@rrrr#sR4    1 &81 1 rr)r6rtypingrrr urllib.parserlitellm._loggingrlitellm.integrations.arizer(litellm.types.integrations.langfuse_otelr r opentelemetry.tracer _Span"litellm.integrations.opentelemetryr _OpenTelemetryConfig litellm.types.integrations.arizer _ProtocolLANGFUSE_CLOUD_EU_ENDPOINTr5rrErrrRsv ,,+- 1GH.  DH DJLR R r