^hdSrSSKrSSKJrJrJr SSKJr \(aSSKJ r SSK J r \ r \ r O\r \r SSKrSSK7 "S S 5rg) zO arize AI is OTEL compatible this file has Arize ai specific helper functions N) TYPE_CHECKINGAnyOptional)verbose_logger)SpanOpenTelemetryConfig)*c~\rSrSr\S\4Sj5r\S\4Sj5r\S\ \ 4Sj5r \S\ S\ 4Sj5r S rg ) ArizeLoggerspancSSKJnJnJn UR S05nUR S5(a+UR UR UR S55 UR URURR5 UR S5nU(aUR URUSR SS55 [U5HtupUR URS US UR3U S 5 UR URS US UR3U R SS55 Mv [R!U5n ["R$"U 5n UR UR&U 5 UR S 5(a+UR UR(UR S 55 UR S 5Hn U R S 05n UR UR*U R SS55 UR UR,SUR3U S 5 UR UR,SUR3U R SS55 M UR S5nU(aUR UR.UR S55 UR UR0UR S55 UR UR2UR S55 g![4a#n[6R8"SU35 SnAgSnAff=f)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 W/home/james-whalen/.local/lib/python3.13/site-packages/litellm/integrations/arize_ai.pyset_arize_ai_attributes#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 IsL-L:: M'M""M'returnc[RRS5n[RRS5nU(d [S5eU(d [S5e[RRS5n[RRS5nUcUc[R "S5 Sn[ UUUUS 9$) 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 ArizeConfigrMs rD_get_arize_configArizeLogger._get_arize_configsJJNN#45 **..1QR ROP P '78  '<=  ]%:  s 8M''   cSSKJn [R5nUR(aU"SURS9$U"SUR S9$)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 rWrQrP)r arize_configs rDget_arize_opentelemetry_config*ArizeLogger.get_arize_opentelemetry_configsR 7"446  % %&$%33  # !//  rYpayloadc PUR5Hup[U[5(a'[R UR 55X'MB[U[ [[[[S545(d[ U5X'MM U$![a Mf=f)N) items isinstancedictr r/copystrintfloatbooltyper9)rckeyr)s rDr/"ArizeLogger.make_json_serializables!--/JC eT**#.#E#Eejjl#SGL#ECeT4:+NOO#&u:GLP *  s:Br~sI  //+1J D. D .yyrY