Ë
    tPêj  ã                   óÆ   — d Z ddlmZ ddlmZ ddlmZmZ ddlm	Z
 ddlmZmZmZ ddlmZ ddlmZ  G d	„ d
e«      Zddœdedz  defd„Zdedz  fd„Zdedeeef   fd„Zy)uC   Arize preset â€” OTLP exporter to Arize + OpenInference vocabulary.é    )ÚFinal)ÚField)ÚBaseSettingsÚSettingsConfigDict)ÚArizeLogger)ÚExporterOwnerÚExporterSpecÚOpenTelemetryV2Config)Úensure_mappers)ÚStandardCallbackDynamicParamsc                   óF   — e Zd ZU  edd¬«      Z edd¬«      Zedz  ed<   y)Ú_ArizeSettingsFÚignore)Úcase_sensitiveÚextraNÚ!OTEL_EXPORTER_OTLP_TRACES_HEADERS)ÚdefaultÚvalidation_aliasÚotlp_traces_headers)	Ú__name__Ú
__module__Ú__qualname__r   Úmodel_configr   r   ÚstrÚ__annotations__© ó    úl/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/otel/presets/arize.pyr   r      s'   … Ù%°UÀ(ÔK€Lñ ',°DÐKnÔ&oÐ˜˜t™Ôor   r   N)Úconfig_overridesr   Úreturnc           	      óŠ  — t        j                  «       }t        |«      }| xs
 t        «       }|j	                  g |j
                  ¢t        |j                  xs d|j                  xs d|t        j                  ¬«      ‘t        |j                  d«      i |j                  ¥|j                  rd|j                  ini ¥dœ¬«      S )NÚ	otlp_grpczhttps://otlp.arize.com/v1)ÚkindÚendpointÚheadersÚownerÚopeninferenceÚmodel_id)Ú	exportersÚmapper_namesÚresource_attributes)Úupdate)Ú_V1ArizeLoggerÚget_arize_configÚ_arize_headersr
   Ú
model_copyr)   r	   Úprotocolr$   r   ÚARIZE_AXr   r*   r+   Úproject_name)r   Ú	arize_cfgr%   Úbases       r   Úarize_presetr6      sÏ   € ô &×6Ñ6Ó8€IÜ# IÓ.€GØ"Ò=Ô&;Ó&=€DØ�?‰?ðØ—‘ðäØ"×+Ñ+Ò:¨{Ø&×/Ñ/ÒNÐ3NØ#Ü'×0Ñ0ô	ðô +¨4×+<Ñ+<¸oÓNð$Ø×*Ñ*ð$à;D×;QÒ;Q�J 	× 6Ñ 6Ñ7ÐWYð$ñ
ð ó ð r   c                 ó0  — g }| j                   s| j                  r,|j                  d| j                   xs | j                  › �«       | j                  r|j                  d| j                  › �«       |st	        «       j
                  S dj                  |«      S )Nz	space_id=zapi_key=ú,)Úspace_idÚ	space_keyÚappendÚapi_keyr   r   Újoin)r4   Úpiecess     r   r/   r/   5   sƒ   € Ø€FØ×Ò˜Y×0Ò0Ø�‰˜	 )×"4Ñ"4Ò"K¸	×8KÑ8KÐ!LÐMÔNØ×ÒØ�‰˜ ×!2Ñ!2Ð 3Ð4Ô5Ùô Ó×3Ñ3Ð3Ø�8‰8�FÓÐr   Úparamsc                 ó�   — i }| j                  d«      xs | j                  d«      }|r||d<   | j                  d«      }|r||d<   |S )z<Per-request Arize OTLP headers from team/key dynamic params.Úarize_space_keyÚarize_space_idzarize-space-idÚarize_api_keyr<   )Úget)r?   r%   Úspacer<   s       r   Úarize_dynamic_headersrF   B   sS   € à%'€Gà—:‘:Ð/Ó0ÒP°F·J±JÐ?OÓ4P€EÙØ$)ˆÐ Ñ!Ø—Z‘Z Ó0€GÙØ$ˆ�	ÑØ€Nr   )Ú__doc__Útypingr   Úpydanticr   Úpydantic_settingsr   r   Ú litellm.integrations.arize.arizer   r-   Ú&litellm.integrations.otel.model.configr   r	   r
   Ú'litellm.integrations.otel.presets.utilsr   Úlitellm.types.utilsr   r   r6   r   r/   ÚdictrF   r   r   r   ú<module>rP      s‚   ðÙ Iå å ß >å J÷ñ õ
 CÝ =ôp�\ô pð 6:òà+¨dÑ2ðð óð6
  t¡ó 
ð
Ð"?ð 
ÀDÈÈcÈÁNô 
r   