o ưi@sndZddlZddlmZddlmZmZmZddlm Z m Z ddl m Z eGdddZ Gd d d e ZdS) zQ AgentOps integration for LiteLLM - Provides OpenTelemetry tracing for LLM calls N) dataclass)OptionalDictAny) OpenTelemetryOpenTelemetryConfig)_get_httpx_clientc@sbeZdZUdZeed<dZeeed<dZeeed<dZ eeed<dZ eed<e d d Z dS) AgentOpsConfig%https://otlp.agentops.cloud/v1/tracesendpointNapi_key service_namedeployment_environment%https://api.agentops.ai/v3/auth/token auth_endpointcCs(|dtdtddtddddS) Nr ZAGENTOPS_API_KEYZAGENTOPS_SERVICE_NAMEagentopsZAGENTOPS_ENVIRONMENT productionr)r r r rr)osgetenv)clsr]/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/agentops/agentops.pyfrom_envs  zAgentOpsConfig.from_env) __name__ __module__ __qualname__r str__annotations__r rr rr classmethodrrrrrr s  r csLeZdZdZ d deeffdd Zdededeee ffd d Z Z S) AgentOpsa` AgentOps integration - built on top of OpenTelemetry Example usage: ```python import litellm litellm.success_callback = ["agentops"] response = litellm.completion( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "Hello, how are you?"}], ) ``` Nconfigcs|durt}d}d}|jr-z||j|j}|d}|d}Wn ty,Ynw|r4d|nd}td|j|d}t j |dd|j pJd|j pNd dd }|rX||d <||_ dS) Ntoken project_idzAuthorization=Bearer Z otlp_http)Zexporterr headersr)r Z callback_nameZlitellmr)z service.namezdeployment.environmentztelemetry.sdk.namez project.id)r rr _fetch_auth_tokenrget Exceptionrr super__init__r rZresource_attributes)selfr Z jwt_tokenr"responser#Z otel_configZresource_attrs __class__rrr(,s:   zAgentOps.__init__r rreturncCs^ddd}t}z!|j||d|idd}|jdkr!td|j|W|S|w) a Fetch JWT authentication token from AgentOps API Args: api_key: AgentOps API key auth_endpoint: Authentication endpoint Returns: Dict containing JWT token and project ID zapplication/jsonz keep-alive)z Content-Type Connectionr )urlr#jsontimeoutzFailed to fetch auth token: )rpost status_coder&textr1close)r)r rr#clientr*rrrr$Xs  zAgentOps._fetch_auth_token)N) rrr__doc__rr r(rrrr$ __classcell__rrr+rrs&,r)r9r dataclassesrtypingrrrZ"litellm.integrations.opentelemetryrrZ&litellm.llms.custom_httpx.http_handlerrr rrrrrs