o i*@sUddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z eeee fZ ej ddd Zej e ed <Gd d d eZdS) N)Optional)Union)Context)DefaultContextProvider)Span) SpanTypesdatadog_llmobs_contextvar)default_DD_LLMOBS_CONTEXTVARcsjeZdZdZdfdd ZdefddZdedeefd d Z d e ddffd d Z de fddZ Z S)LLMObsContextProviderzContext provider that retrieves contexts from a context variable. It is suitable for synchronous programming and for asynchronous executors that support contextvars. returnNcstt|tddS)N)superr__init__r set)self __class__K/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/llmobs/_context.pyrszLLMObsContextProvider.__init__cCst}|duS)zDReturns whether there is an active context in the current execution.N)r getrctxrrr_has_active_contextsz)LLMObsContextProvider._has_active_contextspancCsL|js|S|j}|r|js|jtjkr|||S|j}|s |ddS)zUpdates the active LLMObs span. The active span is updated to be the span's closest unfinished LLMObs ancestor span. N)finished_parent span_typerLLMactivate)rr new_activerrr_update_active#s  z$LLMObsContextProvider._update_activercst|tt||dS)z8Makes the given context active in the current execution.N)r rr rrrrrrr2s zLLMObsContextProvider.activatecCs t}t|tr||S|S)z=Returns the active span or context for the current execution.)r r isinstancerr )ritemrrractive7s  zLLMObsContextProvider.active)r N)__name__ __module__ __qualname____doc__rboolrrrr ContextTypeValuerr# __classcell__rrrrr sr ) contextvarstypingrrddtrace._trace.contextrddtrace._trace.providerrddtrace._trace.spanr ddtrace.extrr) ContextVarr __annotations__r rrrrs