$jY>dZddlmZddlmZddlmZddlmZddl m Z m Z m Z m Z ddlmZmZdeeeffd Zdeeeffd Zded zfd Ze dd edededededed zdee d d ffdZed edededee d d ffdZgdZd S)zServer-side telemetry helpers.) Generator)contextmanager) request_ctx)Context)SpanSpanKindStatus StatusCode)extract_trace_context get_tracerreturncddlm}i} |}|r5|jr |j|d<|jrd|j|d<n#t $rYnwxYw|S)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 `/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/telemetry.pyget_auth_span_attributesr s<<<<<<E   ""  @ 6&+ol#| @),%,)?)?o&      LsAA AAc~ddlm}i} |}|j|j |j|d<n#t$rYnwxYw|S)z/Get session attributes for the current request.r) get_contextNzmcp.session.id)rrrequest_context session_idr)rrctxs rget_session_span_attributesrsl777777E kmm   *s~/I&)nE" #      Ls "- ::Nc tj}|r8t|dr(|jr!t t |jSn#t $rYnwxYwdS)zCGet parent trace context from request meta for distributed tracing.metaN)rgethasattrr!r dict LookupError)req_ctxs r_get_parent_trace_contextr',st /##  =ww// =GL =(gl););<< <      4sA A AAnamemethod server_namecomponent_type component_key resource_uric #Kt}||ttj5}d||||||dt t }|||d<|| |VnT#t$rG} | | | ttj d} ~ wwxYw ddddS#1swxYwYdS)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=7ss\\F  % % )++ _ &    & %#.&4%2 ! ! '(( ! *++ !   #(4E$ % E""" JJJJ     ! !! $ $ $ OOF:#344 5 5 5   /s7AC*BC* CACCC**C.1C. provider_typec#nKt}|d|5}|||d |VnT#t$rG}|||t tjd}~wwxYw ddddS#1swxYwYdS)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$&8&8 9 9 T )6)6      JJJJ     ! !! $ $ $ OOF:#344 5 5 5                     s6B*AB* BABBB**B.1B.)r@rrr=)N)__doc__collections.abcr contextlibrmcp.server.lowlevel.serverropentelemetry.contextropentelemetry.tracerrr r fastmcp.telemetryr r r$strrrr'r=r@__all__rrLs$$%%%%%%%%%%%%222222))))))BBBBBBBBBBBB????????$sCx." T#s(^    7T> $ (( ( (( (  ( * (tT4 ((((V tT4 4   rK