jdZddlZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZerdd lmZdd lmZeZeeefZneZeZGd de Zy)zO arize AI is OTEL compatible this file has Arize ai specific helper functions N)datetime) TYPE_CHECKINGAnyOptionalUnion)_utils)ArizeOTELAttributes) OpenTelemetry) ArizeConfig)ServiceLoggerPayload)StandardCallbackDynamicParams)Span)Protocolc*eZdZdedeefdZedefdZede fdZ dde d eed ee e efd ee e efd eef d Z dde deed eed ee e efd ee ee fd eef dZd e defdZdZdedeefdZy) ArizeLoggerspan response_objc2tj|||yN)rset_arize_attributes)selfrkwargsrs y/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/litellm/integrations/arize/arize.pyset_attributeszArizeLogger.set_attributess((v|Dc<tj|||tyr)rrr )rrrs rrz ArizeLogger.set_arize_attributes#sdFL:MNrreturncDtjjd}tjjd}tjjd}tjjd}d}d}|rd}|}n |rd}|}nd}d}t|||| S) z Helper function to get Arize configuration. Returns: ArizeConfig: A Pydantic model containing Arize configuration. Raises: ValueError: If required environment variables are not set. ARIZE_SPACE_KEY ARIZE_API_KEYARIZE_ENDPOINTARIZE_HTTP_ENDPOINTN otlp_grpc otlp_httpzhttps://otlp.arize.com/v1) space_keyapi_keyprotocolendpoint)osenvirongetr )r%r& grpc_endpoint http_endpointr(r's rget_arize_configzArizeLogger.get_arize_config(sJJNN#45 **..1 '78  '<= ( "H$H "H$H"H2H   rNpayloadparent_otel_span start_timeend_timeevent_metadatac KywzaArize is used mainly for LLM I/O tracing, sending router+caching metrics adds bloat to arize logsN)rr/r0r1r2r3s rasync_service_success_hookz&ArizeLogger.async_service_success_hookMs  errorc Kywr5r6)rr/r9r0r1r2r3s rasync_service_failure_hookz&ArizeLogger.async_service_failure_hookXs  r8headerscy)z_Arize is used mainly for LLM I/O tracing, sending Proxy Server Request adds bloat to arize logsNr6)rr1r<s r)create_litellm_proxy_request_started_spanz5ArizeLogger.create_litellm_proxy_request_started_spands rcK |j}|jsdddS|jsdddSdddS#t$r}ddt |dcYd }~Sd }~wwxYww) z Performs a health check for Arize integration. Returns: dict: Health check result with status and message unhealthyz,ARIZE_SPACE_KEY environment variable not set)status error_messagez*ARIZE_API_KEY environment variable not sethealthyz)Arize credentials are configured properly)rAmessagezArize health check failed: N)r.r%r& Exceptionstr)rconfiges rasync_health_checkzArizeLogger.async_health_checkls **,F##)%S >>)%Q $F   %#>s1vh!G  sCA" ;A";A";A" AAAA"AA" standard_callback_dynamic_paramsci}|jdr|jd|d<|jdr|jd|d<|jdr|jd|d<|S)z Construct dynamic Arize headers from standard callback dynamic params This is used for team/key based logging. Returns: dict: A dictionary of dynamic Arize headers arize_space_idzarize-space-idarize_space_key arize_api_keyr&)r+)rrJdynamic_headerss rconstruct_dynamic_otel_headersz*ArizeLogger.construct_dynamic_otel_headerss , / /0@ A0P0T0T 1O, - , / /0A B0P0T0T!1O, - , / / @)I)M)M*OI &r)NNNN)NNNN)__name__ __module__ __qualname__rrrr staticmethodrr r.r rrfloatdictr7rFr;r>rIr rPr6rrrrse4x}4" k" " N,07;59)-  %  #4.  U8U?34  5512  !   "+/7;59)-  %  }  #4.  U8U?34  512  !     B"0M" $"rr)__doc__r)rtypingrrrrlitellm.integrations.arizer!litellm.integrations.arize._utilsr "litellm.integrations.opentelemetryr litellm.types.integrations.arizer litellm.types.servicesr litellm.types.utilsr opentelemetry.tracer_Spanr _Protocolrr6rrrcs^  66-A<87=1FH  DH DQ-Qr