gddZddlZddlmZmZmZddlmZerddlm Z ddl m Z e Z e Z neZ eZ ddlZddlGd d Zy) zO arize AI is OTEL compatible this file has Arize ai specific helper functions N) TYPE_CHECKINGAnyOptional)verbose_logger)SpanOpenTelemetryConfig)*cneZdZedefdZedefdZedee fdZ ede de fdZ y) ArizeLoggerspancddlm}m}m} |j di}|j dr+|j |j |j d|j |j|jj|j d}|r|j |j|dj ddt|D]t\}} |j |jd |d |j| d |j |jd |d |j| j ddvtj!|} t#j$| } |j |j&| |j d r+|j |j(|j d |j d D]} | j d i} |j |j*| j dd|j |j,d|j| d |j |j,d|j| j dd|j d}|r|j |j.|j d|j |j0|j d|j |j2|j dy#t4$r"}t7j8d|Yd}~yd}~wwxYw)Nr)MessageAttributesOpenInferenceSpanKindValuesSpanAttributesoptional_paramsmodelmessagescontent.roleuserchoicesmessagez.0.usage total_tokenscompletion_tokens prompt_tokensz Error setting arize attributes: )*litellm.integrations._types.open_inferencerrrget set_attributeLLM_MODEL_NAMEOPENINFERENCE_SPAN_KINDLLMvalue INPUT_VALUE enumerateLLM_INPUT_MESSAGES MESSAGE_ROLEMESSAGE_CONTENTr make_json_serializablejsondumpsLLM_INVOCATION_PARAMETERSUSER_ID OUTPUT_VALUELLM_OUTPUT_MESSAGESLLM_TOKEN_COUNT_TOTALLLM_TOKEN_COUNT_COMPLETIONLLM_TOKEN_COUNT_PROMPT Exceptionrerror)rkwargs response_objrrrrridxmsg_optional_params_json_optional_paramschoiceresponse_messageres R/var/www/openai/venv/lib/python3.12/site-packages/litellm/integrations/arize_ai.pyset_arize_ai_attributesz#ArizeLogger.set_arize_ai_attributess(  b I$jj):B?Ozz'""">#@#@&**WBUV   66+//55 zz*-H"""..RL$$Y3 !*( 3HC&&)<<=Qse1EVEcEcDdeF  &&)<<=Qse1EVEfEfDgh 2.!4 +AA/R $(JJ/?$@ !   88:O ""6*"">#9#9?;N;Nv;VW '**95#)::i#< """//1A1E1EiQS1T ""%99:#>O>\>\=]^$V,""%99:#>O>_>_=`a$((B76"!$$W-E"""88IIn- """==II12 """99IIo.  I  #CA3!G H H IsLL"" M +MM returnctjjd}tjjd}|s td|s tdtjjd}tjjd}||t j dd}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_KEYz2ARIZE_SPACE_KEY not found in environment variablesz0ARIZE_API_KEY not found in environment variablesARIZE_ENDPOINTARIZE_HTTP_ENDPOINTzaNo ARIZE_ENDPOINT or ARIZE_HTTP_ENDPOINT found, using default endpoint: https://otlp.arize.com/v1zhttps://otlp.arize.com/v1 space_keyapi_key grpc_endpoint http_endpoint)osenvironr# ValueErrorrdebug ArizeConfigrKs rC_get_arize_configzArizeLogger._get_arize_configsJJNN#45 **..1QR ROP P '78  '<=  ]%:  s 8M''   cddlm}tj}|jr|d|jS|d|j S)z Helper function to get OpenTelemetry configuration for Arize. Args: arize_config (ArizeConfig): Arize configuration object. Returns: OpenTelemetryConfig: Configuration for OpenTelemetry. rr otlp_http)exporterendpoint otlp_grpc) opentelemetryr r rUrOrN)r arize_configs rCget_arize_opentelemetry_configz*ArizeLogger.get_arize_opentelemetry_configsR 7"446  % %&$%33  # !//  rVpayloadc 4|jD]u\}} t|tr'tj |j ||<n8t|t ttttdfst |||<w|S#t$rYwxYw)N) items isinstancedictr r.copystrintfloatbooltyper8)r_keyr(s rCr.z"ArizeLogger.make_json_serializables!--/JC eT*#.#E#Eejjl#SGCL#ECeT4:+NO#&u:GCL*  sA/B  BBN) __name__ __module__ __qualname__ staticmethodrrDrTrUrr r^rcr.rVrCr r siIdiIiIZ  {    D H5H,I  2    rVr )__doc__r/typingrrrlitellm._loggingropentelemetry.tracer_Spanr\r _OpenTelemetryConfigrP litellm.types.integrations.arizer rorVrCrwsI  //+1J D. D .yyrV