jKj8<dZddlmZddlZddlZddlmZmZmZddl m Z ddl m Z mZmZddlmZmZmZddlmZmZmZmZmZmZmZmZmZmZdd lm Z dd l!m"Z"dd l#m$Z%m&Z&m'Z'd Z(dd Z)ddZ*dd ddZ+ ddZ,ddZ-ddZ.ddZ/ddZ0ddZ1y)uLow-level ``mcp.server.Server`` adapter. The low-level server keeps its handlers in a public ``request_handlers`` dict, so we wrap the ``CallToolRequest`` and ``ListToolsRequest`` entries directly. Unlike FastMCP, the low-level ``call_tool`` handler catches exceptions and returns a ``CallToolResult`` with ``isError=True`` rather than raising — so we detect errors from the result, not a ``try/except``. Session and client info are read from the server's ``request_context`` contextvar (the handler receives only the request). ) annotationsN)AnyOptionalTuple)add_context_parameter_to_schemaget_context_descriptionis_context_enabled)add_conversation_id_to_schemabuild_prompt_backresolve_conversation_id) _to_jsonableappend_get_more_toolsbuild_tool_call_request extract_toolsprepare_requestread_tool_categoryrecord_missing_capabilityrecord_tool_callrecord_tools_listrequest_to_dict)MCPAnalyticsData)log)GET_MORE_TOOLS_NAMEget_more_tools_result_text$resolve_missing_capability_tool_name__posthog_mcp_wrapped__ct|dd|_t|dd|_t||dt ||dy)uInstrument a raw ``mcp.server.Server``. ``context`` is injected as an optional schema property and NOT stripped — that schema is also the call's validation schema, and a typical ``(name, arguments)`` handler ignores extra keys.nameNversionF)strip_injectedcontext_required)getattr server_nameserver_version_wrap_call_tool_wrap_list_tools)serverdatas q/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/mcp/_instrument_lowlevel.pyinstrument_low_levelr,8s>vvt4D!&)T:DFD7VTE:ct|dd}| tdyt|ddxs t|dd|_t|ddxs t|dd|_t ||d|t ||dy) uInstrument jlowin's standalone ``fastmcp.FastMCP`` (FastMCP 2.0). It exposes a ``_mcp_server`` (a subclass of the official low-level Server) with the same ``request_handlers`` seam, but validates tool args against the function signature and rejects unexpected kwargs — so we STRIP the injected ``context``/``conversation_id`` before dispatch (like the official FastMCP path). _mcp_serverNz?Warning: fastmcp.FastMCP has no _mcp_server; cannot instrument.rr T)r! high_levelr")r$rr%r&r'r()r)r* low_levels r+instrument_fastmcp_v2r2Bs  t4I MNvvt4X 6SW8XD!&)T:g9d?DItDVLYterror duration_msr8r9rDr;rootrArB)rr>resultrGr8r9rDr;)paramsrdictr> _client_info_mcp_session_idrrroptionsreport_missingrget mcp_types ServerResultCallToolResult TextContentrr enable_conversation_id_tool_owned_injected_keyspoptime monotonic Exceptionrr$ isinstancelistappendr )reqrr>r8r9r7r:r;r5 missing_namerDmintedownedkeystartrIrFrG call_resultdelivered_conversation_idrAr*r0originalr)r!s r+handlerz _wrap_call_tool..handler\szz--34 &26&:# ^(0)$ :~.* )#)   W))#*;O*LV*T -1)  ##)5   Q  BF)  ##!^^-5='-(.O    % P sA7K :I4;AK I7BK -I:. K <z4_wrap_list_tools..handler..(s=u!qvv-usztools/list returned no tools) rqr:responserGrrrFr8r9r;)!rLrMrrrXrYrZrrr]rr$rstool_descriptionsrtool_categorieslenr rNr<r _GET_MORE_TOOLS_NAME_schema_has_paramrrUr rtrrOranyrr)r^r8r9r7r:r;r5rcrIrFrGtoolsrqtoolcategoryemptycontext_enabledrsschemar_r#r*rfr)s @r+rgz!_wrap_list_tools..handlers ;!#& &&26&:# ^(0!#&~.+ )#)     #C=(F~~'%/47 f%D LL #t]D1484D4D&&tyy1)$/H2:$$TYY/ E a,T\\-A-AB-dll.B.BC Dyy00T=$7F'8'K8DII{=M||22;L)<7vtyyIWT=$??P'-D$!( << & &? ML=u==%fl; \*  !&)#490t#)   y' ) #!^^-5='-     b!PB499+NOP sL JAL JL - J8J9J=BL C;L ?K"BL  L  L L J K#0KKKKL "!LL LL Trh)rjrPrQListToolsRequestr$rlrm)r)r*r#rnrgrfs``` @r+r(r(sc&&H||I667H78]EB``D G]D)+2HY ' '(r-cvt|txr(t|jdtxr||dvS)N properties)r[rKrP)rrs r+rr@s;64  ) vzz,/ 6 ) F<( (r-cBK| tS |j|d{}t|dd}|(ttj|j n t}dDchc] }||vs| c}S7Zcc}w#t $r tcYSwxYww)uIWhich of (``context``, ``conversation_id``) the jlowin FastMCP tool declares itself, read from its function signature. These are real tool arguments we must not strip. On any lookup failure, return empty (strip both) — same as the prior unconditional behaviour, so a flaky introspection never leaks an injected key.NfnrC)setget_toolr$inspect signature parametersrZ)r0rrrrJks r+rVrVHs u ((.. T4 &:<.W&&r*556ce9I9aQ&[9II/J u sQ BBA?A B. B8B<B>B?BBBBBBcF |jS#ttf$rYywxYwrw)request_context LookupErrorAttributeError)r)s r+_request_contextrXs*%%%  (s   ct|} |jj}|r8|jr,|jj|jj fSy#t $rYywxYw)N)NN)rsession client_params clientInforr rZ)r)ctxrs r+rLrL_sj 6 "C  11 ]55 ++00-2J2J2R2RR R     sAA A*)A*ct|} t|dd}t|dd}||jdS y#t$rYywxYw)Nr:headerszmcp-session-id)rr$rPrZ)r)rr:rs r+rMrMjsd 6 "C #y$/'9d3  ;;/0 0      s,< AA)r)rr*rriNone) r)rr*rr!boolr0rrir)r)rr*rr#rrir)rrrstrrir)r0rrrrir)r)rrir)r)rriz#Tuple[Optional[str], Optional[str]])r)rriz Optional[str])2__doc__ __future__rrrXtypingrrr mcp.typestypesrQ_context_parametersrr r _conversation_idr r r _instrumentationrrrrrrrrrr _internalrloggerrrrrrrrlr,r2r'r(rrVrrLrMr-r+rs # ''    ( * ;=&UY{2 {2'{2<@{2NQ{2 {2|k3 k3'k3>Bk3 k3\   r-