iYdZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z ddlmZmZdeeeffd Zdeeeffd Zded zfd Ze dd edededededed zdee d d ffdZed edededee d d ffdZgdZy )zServer-side telemetry helpers.) Generator)contextmanager) request_ctx)Context)SpanSpanKindStatus StatusCode)extract_trace_context get_tracerreturncddlm}i} |}|rE|jr|j|d<|jrdj |j|d<|S#t $rY|SwxYw)z>Get auth attributes for the current request, if authenticated.r)get_access_tokenz enduser.id z enduser.scope)fastmcp.server.dependenciesr client_idscopesjoin RuntimeError)rattrstokens h/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/server/telemetry.pyget_auth_span_attributesr sm<E  " &+ool#||),%,,)?o& L   L sAA A'&A'cddlm}i} |}|j|j|j|d<|S#t$rY|SwxYw)z/Get session attributes for the current request.r) get_contextzmcp.session.id)rrrequest_context session_idr)rrctxs rget_session_span_attributesrsZ7E m    *s~~/I&)nnE" # L   L s.: AANc tj}|r6t|dr*|jrt t |jSy#t $rYywxYw)zCGet parent trace context from request meta for distributed tracing.metaN)rgethasattrr!r dict LookupError)req_ctxs r_get_parent_trace_contextr',sU //# ww/GLL(gll);< <     sA A AAnamemethod server_namecomponent_type component_key resource_uric #Kt}|j|ttj5}d||||||dt t }|||d<|j| | dddy#t$r?} |j| |jttjd} ~ wwxYw#1swYyxYww)zCreate a SERVER span with standard MCP attributes and auth context. Automatically records any exception on the span and sets error status. )contextkindmcp)z rpc.systemz rpc.servicez rpc.methodzmcp.method.namezfastmcp.server.namezfastmcp.component.typefastmcp.component.keyNzmcp.resource.uri)r start_as_current_spanr'rSERVERrrset_attributes Exceptionrecord_exception set_statusr r ERROR) r(r)r*r+r,r-tracerspanres r server_spanr=7s\F  % % )+ __ &   & %#.&4%2 ! '( ! *+ !   #(4E$ % E" J/  0   ! !! $ OOF:#3#34 5  1  s:5C7C/A=4 C= C:CCCC C provider_typec#<Kt}|jd|5}|j||d | dddy#t$r?}|j ||j t tjd}~wwxYw#1swYyxYww)zCreate an INTERNAL span for provider delegation. Used by FastMCPProvider when delegating to mounted servers. Automatically records any exception on the span and sets error status. z delegate )zfastmcp.provider.typer2N) r r3r5r6r7r8r r r9)r(r>r,r:r;r<s r delegate_spanr@cs\F  % % $&8 9T )6)6   J : 9   ! !! $ OOF:#3#34 5   : 9s8BBA B B :BB  BBB)r@rrr=)N)__doc__collections.abcr contextlibrmcp.server.lowlevel.serverropentelemetry.contextropentelemetry.tracerrr r fastmcp.telemetryr r r$strrrr'r=r@__all__rrLs $%%2)BB?$sCx." T#s(^ 7T> $ ( ( (( (  ( * (tT4 ((V tT4 4 rK