tPjUdZddlmZmZddlZddlmZddlmZm Z ddl m Z ddl m Z mZmZddlmZd Zeed <d Zeed <d Zeed<GddeZdddedzdefdZdedefdZdedeeeffdZeeey)uyAgentOps preset — OTLP/HTTP to AgentOps' endpoint with a lazily-fetched JWT. AgentOps authenticates with a short-lived JWT minted from the API key. Fetching it is blocking network I/O, so it must never run on the event loop: callback construction (where presets are built) can run inside the proxy's async startup or, in the SDK, on the first request. Instead of fetching at config-build time, this preset registers a custom exporter (``kind="agentops"``) that mints the JWT **on its first export** — which the ``BatchSpanProcessor`` runs in its own worker thread, off any event loop — and caches it for the process lifetime. )AnyFinalN)Field) BaseSettingsSettingsConfigDict)verbose_logger) ExporterOwner ExporterSpecOpenTelemetryV2Config)register_exporter_factoryz"https://otlp.agentops.ai/v1/traces_AGENTOPS_ENDPOINTz%https://api.agentops.ai/v3/auth/token_AGENTOPS_AUTH_ENDPOINTagentops_AGENTOPS_EXPORTER_KINDceZdZUeddZeddZedzed<edd Z eed <edd Z edzed <y) _AgentOpsSettingsFignore)case_sensitiveextraNAGENTOPS_API_KEY)defaultvalidation_aliasapi_keyrAGENTOPS_SERVICE_NAME service_nameAGENTOPS_ENVIRONMENT environment) __name__ __module__ __qualname__r model_configrrstr__annotations__rro/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/otel/presets/agentops.pyrrsL%U(KL?QRGS4ZRjCZ[L#[#DCYZKtZr%r)config_overridesr'returnc `t}|xs t}|jg|jt t t |jrd|jindtji|j|jdd|jrd|jinidS) uEBuild the AgentOps config without any network I/O. The ``agentops`` exporter mints (and caches) the JWT lazily on its first export, so this stays non-blocking. ``project.id`` is therefore not a resource attribute — it is encoded in the JWT, which AgentOps uses to route the trace to the right project. rN)kindendpointoptionsownerr)z service.nameztelemetry.sdk.namezdeployment.environment) exportersresource_attributes)update) rr model_copyr.r rr rr AGENTOPSr/rr)r'settingsbases r&agentops_presetr5's()H"=&;&=D ??0/>F>N>Ni)9)9:TX'00 $**$ ( 5 5&0$HPG[G[,h.B.BCac $   r%speccddlm}Gdd|}|jxsi}||j|j dS)zKFactory for the ``agentops`` exporter kind: a lazy-auth OTLP/HTTP exporter.r)OTLPSpanExportercXeZdZdZdedzdedzddffd Zd dZdedeffd ZxZ S) ;_build_agentops_exporter.._LazyAuthAgentOpsExportera6OTLP/HTTP exporter that mints the AgentOps JWT on its first export. ``export`` runs in the ``BatchSpanProcessor`` worker thread, so the blocking token fetch never touches an event loop. The result is cached after the first attempt (success or failure) so it runs at most once. r+Nrr(cBt||||_d|_y)N)r+F)super__init___agentops_api_key_auth_resolved)selfr+r __class__s r&r=zD_build_agentops_exporter.._LazyAuthAgentOpsExporter.__init__Ws# G h  /%,D ""'D r%c&|jryd|_|jsy t|jjd}|rd||jj d<yy#t $r }tjd|Yd}~yd}~wwxYw)NTtokenzBearer AuthorizationzAgentOps JWT fetch failed: %s) r?r>_fetch_agentops_jwtget_sessionheaders Exceptionrdebug)r@rCes r&_ensure_authenticatedzQ_build_agentops_exporter.._LazyAuthAgentOpsExporter._ensure_authenticated\s"""&D )) I243I3IJNNwW@Gug=NDMM))/:   I$$%DaHH IsAA'' B0B  BspanscB|jt| |S)N)rLr<export)r@rMrAs r&rOzB_build_agentops_exporter.._LazyAuthAgentOpsExporter.exportls  & & (7>%( (r%)r(N) rrr __doc__r"r=rLrrO __classcell__)rAs@r&_LazyAuthAgentOpsExporterr:OsH  (d  (S4Z (D (  I  ) ) ) )r%rRr)r+r)5opentelemetry.exporter.otlp.proto.http.trace_exporterr8r,r+rF)r6r8rRr,s r&_build_agentops_exporterrTIs@)$4)B\\'RG $dmmW[[QZE[ \\r%rctjd5}|jtdddd|i}|jdk7rt d |j |jcdddS#1swYyxYw) N )timeoutzapplication/jsonz keep-alive)z Content-Type Connectionr)urlrHjsonz Failed to fetch AgentOps token: )httpxClientpostr status_code RuntimeErrortextrZ)rclientresponses r&rErEts} b !V ++'%7|TW%&    3 &!A(--QR R}}s AA55A>)rPtypingrrr\pydanticrpydantic_settingsrrlitellm._loggingr&litellm.integrations.otel.model.configr r r ,litellm.integrations.otel.plumbing.providersr r r#rrrr5rTr"dictrEr$r%r&rks  >+ S@E@!HH!++[ [6:+d2D(]<(]C(]V  c3h  13KLr%