o +iT@sdZGdddZdS)z:Middleware for making it easier to do distributed tracing.c@s&eZdZdZddZdefddZdS)TracingMiddlewareaWMiddleware for propagating distributed tracing context using LangSmith. This middleware checks for the `'langsmith-trace'` header and propagates the tracing context if present. It does not start new traces by default. Designed to work with ASGI applications. Attributes: app: The ASGI application being wrapped. cCsddlm}||_||_dS)zInitialize the middleware.)tracing_contextN)langsmith.run_helpersr _with_headersapp)selfrrrQ/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/langsmith/middleware.py__init__s  zTracingMiddleware.__init__scopecs|ddkr8d|vr8t|d}d|vr8|j|d||||IdHWddS1s1wYdS||||IdHdS)a*Handle incoming requests and propagate tracing context if applicable. Args: scope: A dict containing ASGI connection scope. receive (callable): An awaitable callable for receiving ASGI events. send (callable): An awaitable callable for sending ASGI events. If the request is HTTP and contains the `'langsmith-trace'` header, it propagates the tracing context before calling the wrapped application. Otherwise, it calls the application directly without modifying the context. typehttpheadersslangsmith-trace)parentN)dictrr)rr receivesendrrrr __call__s  zTracingMiddleware.__call__N)__name__ __module__ __qualname____doc__r rrrrrr rs rN)rrrrrr s