`iV@dZddlZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZdd lmZ dd lmZdd lmZdd lmZn#e$r edwxYw ddlmZn #e$rdZYnwxYwerddlmZmZm Z m!Z!m"Z"ddl#m$Z$m%Z%m&Z&Gdde Z' dAdZ(dBdZ)de*de*ddfdZ+ddde*d e*d!e*d"d#d$d%d&d%d'e*ddfd(Z,dCd*Z-ddd)dd+d%de*ddf d,Z. dDde*d-d.d/d0dd1fd2Z/ dEde*d4d5d-d.d/d0d6d7d8e0ddfd9Z1d:d5de*d;ddZ3dFd?Z4dFd@Z5dS)Ga Sentry integration for MCP (Model Context Protocol) servers. This integration instruments MCP servers to create spans for tool, prompt, and resource handler execution, and captures errors that occur during execution. Supports the low-level `mcp.server.lowlevel.Server` API. Nwraps) TYPE_CHECKING)get_start_span_function)OPSPANDATA) Integration DidNotEnable)safe_serialize)should_send_default_pii) nullcontext)Server) request_ctx)StreamableHTTPServerTransportzMCP SDK not installed)FastMCP)AnyCallableOptionalTupleContextManager)ReceiveScopeSendc@eZdZdZdZd deddfdZed dZdS) MCPIntegrationmcpz auto.ai.mcpTinclude_promptsreturnNc||_dS)z Initialize the MCP integration. Args: include_prompts: Whether to include prompts (tool results and prompt content) in span data. Requires send_default_pii=True. Default is True. N)r)selfrs o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/sentry_sdk/integrations/mcp.py__init__zMCPIntegration.__init__-s /clttttdSdS)zM Patches MCP server classes to instrument handler execution. N)_patch_lowlevel_server_patch_handle_requestr_patch_fastmcpr#r! setup_oncezMCPIntegration.setup_once7s<              r#)TrN) __name__ __module__ __qualname__ identifieroriginboolr" staticmethodr)r(r#r!rr)seJ F///////\r#rrGOptional[Tuple[Optional[sentry_sdk.Scope], Optional[sentry_sdk.Scope]]]c4 tj}n#t$rYdSwxYw|%t|dr|jd|jjvrdS|jjdd|jjddfS)Nrequeststatesentry_sdk.isolation_scopesentry_sdk.current_scope)rget LookupErrorhasattrr4scope)ctxs r!_get_active_http_scopesr=Cso tt sI&&  ;  #++ + +t '"&&'CDD '"&&'ABB s  $$(tuple[Optional[str], Optional[str], str]cd}d}d} tj}||j}t|dr|j|j}t|dr7|jdrd}|jd}nFt|dr6|jdrd }|jd}n#t$rYnwxYw|||fS) a8 Extract request ID, session ID, and MCP transport type from the request context. Returns: Tuple of (request_id, session_id, mcp_transport). - request_id: May be None if not available - session_id: May be None if not available - mcp_transport: "http", "sse", "stdio" Nstdior4 query_params session_idsseheaderszmcp-session-idhttp)rr8 request_idr:r4rArDr9)rFrB mcp_transportr<r4s r!_get_request_context_datarHYs#'J"&J M o ?JsI&& G3;+B+7N33 G8L8P8P 99 G%*M!(!5!9!9,!G!GJJWi00GW_5H5H$66G%+M!(!4!45E!F!FJ       z= 00sCC CC handler_type item_namez#tuple[str, str, str, Optional[str]]c|dkrtj}d}tj}n1|dkrtj}d}tj}ntj}d}d}|d|}||||fS)z Get span configuration based on handler type. Returns: Tuple of (span_data_key, span_name, mcp_method_name, result_data_key) Note: result_data_key is None for resources toolz tools/callpromptz prompts/getzresources/readN )r MCP_TOOL_NAMEMCP_TOOL_RESULT_CONTENTMCP_PROMPT_NAME!MCP_PROMPT_RESULT_MESSAGE_CONTENTMCP_RESOURCE_URI)rIrJ span_data_keymcp_method_nameresult_data_key span_names r!_get_span_configrXs|v . &":  ! ! 0 '"D 1 *"00Y00I )_o EEr#spanr handler_namerTrU argumentszdict[str, Any]rFz Optional[str]rBrGc ||||tj||tj|dkrdnd|tj||r |tj||r |tj||D]+\}} |d|t| ,dS)z%Set input span data for MCP handlers.r@pipetcpzmcp.request.argument.N) set_datarMCP_METHOD_NAMENETWORK_TRANSPORT MCP_TRANSPORTMCP_REQUEST_IDMCP_SESSION_IDitemsr ) rYrZrTrUr[rFrBrGkvs r!_set_span_input_datarhs  MM-...MM(*O<<< MM"mw.F.FFFE MM((-888; h-z:::; h-z:::!!FF1 1a11>!3D3DEEEEFFr#resultc&|dSt|trt|dkr|dSt|tr|St |drt|t t tfsg} |D]a}t |dr||j-t|trd|vr||dbn#t$r|cYSwxYw|rd |n|S|S)a( Extract meaningful content from MCP tool result. Tool handlers can return: - tuple (UnstructuredContent, StructuredContent): Return the structured content (dict) - dict (StructuredContent): Return as-is - Iterable (UnstructuredContent): Extract text from content blocks N__iter__textrN) isinstancetuplelendictr:strbytesappendrn Exceptionjoin)ritextsitems r!_extract_tool_result_contentrzs?~t&%  S[[A%5%5ay&$ vz"" 4:fsE4>P+Q+Q 4  / /4((/LL++++d++/$LLf...  /    MMM #(3sxxV3 MsA$C&& C54C5rVc|dStjt}|dSt o|j}|dkrt |}|i|ri||t|t|tr3|tj t|dSdSdSdS|dkr d}d}t|dr|jr|j}t|}nAt|tr,|dr|d}t|}|dkr |tj||dkrJ|rI|rH|d} d} t| dr| j} n!t| tr d| vr| d} | r |tj| d} t| drY| j} t| d r| j} nt| tr d | vr | d } nrt| t,r| } nZt| trEd| vrA| d} t| tr d | vr | d } nt| t,r| } | r||| dSdSdSdSdS#t.$rYdSwxYwdS) z&Set output span data for MCP handlers.NrLrMrmessagesrlrolecontentrn) sentry_sdk get_clientget_integrationrr rrzr_r rorrrMCP_TOOL_RESULT_CONTENT_COUNTrqr:r|r8MCP_PROMPT_RESULT_MESSAGE_COUNTr}MCP_PROMPT_RESULT_MESSAGE_ROLEr~rnrsrv) rYrirVrI integrationshould_include_data extractedr| message_count first_messager} content_text msg_contents r!_set_span_output_datars`~'))99.IIK233S 8Sv088  %8 MM/>)+D+D E E E)T** V hDc)nnUUUUU !  V V  ! !4 .2HMvz** .v .!? #H FD)) .fjj.D.D .!*- #H q   hF VVV!!&9!h! ( =&111(-DD t441=9P9P(0DQMM("I4PPP $ =)443"/"7K{F333'2'7 #K6636[;P;P'26': #K553'2  t443m9S9S"/ ":K!+t443;9N9N'26': #K553'2 AMM/<@@@@@?"!!!!!<AA    DD i " !sG(K KK original_argsztuple[Any, ...]original_kwargszOptional[dict[str, Any]]z8tuple[str, dict[str, Any], str, str, str, Optional[str]]c|pi}|dkrd|r |d}n|dr|d}i}t|dkr |d}n|dr|d}n|dkrl|r |d}n|dr|d}i}t|dkr |d}n|dr|d}d|i|pi}nFd}|rt|d}n*|drt|d}i}t||\}}}}||||||fS) z Prepare common handler data for both async and sync wrappers. Returns: Tuple of (handler_name, arguments, span_data_key, span_name, mcp_method_name, result_data_key) rLrnamerlr[rMunknownuri)r8rqrsrX) rIrrrZr[rTrWrUrVs r!_prepare_handler_datar=s&+Ov  3(+LL   ( ( 3*62L }   ! !%a(II   - - 5' 4I  ! !  3(+LL   ( ( 3*62L }   ! !%a(II   - - 5' 4I\?io2? !  7}Q/00LL   ' ' 7u566L BRlBB>M9o   r#TfuncCallable[..., Any]r z Optional[Any] force_awaitc K|i}t|||\}}}} } } t} | t} t}nc| \}}|tntj|} |tntj|}t\}}}| 5|5ttj | tj 5}t|||| |||||dkr}|r |d}n|d}d}t|dr|j}n!|rd|vr|dd}|r |t&j| ||g|R}||i|}|st+j|r|d{V}nM#t.$r@}|dkr |t&jd tj|d}~wwxYwt5||| |dddn #1swxYwYdddn #1swxYwYdddn #1swxYwY|S) aK Wrapper for MCP handlers. Args: handler_type: "tool", "prompt", or "resource" func: The handler function to wrap original_args: Original arguments passed to the handler original_kwargs: Original keyword arguments passed to the handler self: Optional instance for bound methods N)oprr/resourcerrschemez://rLT)rr=r rr;use_isolation_scope use_scoperHrr MCP_SERVERrr/rhr8r:rsplitr_rMCP_RESOURCE_PROTOCOLinspect isawaitablervMCP_TOOL_RESULT_IS_ERRORcapture_exceptionr)rIrrrr rrZr[rTrWrUrVscopesisolation_scope_contextcurrent_scope_contextisolation_scope current_scoperFrBrGrYrprotocolries r!_handler_wrapperr~s$ lM?KK% & &F ~"--- + )/&& MMM!55oFF $ MMM!++M:: -F,G,G)J M !2S2S "1 S1 S*(**=%,0 S$ !#!    :--$9+A.-11%88#HsH--@#&:%@%<*?*?#/#5#5e#<# (>(> !T=DODDF".g&9&&A&A.'- #v-- h&GNNN0333 &dFO\RRRa0 S0 S0 S0 S0 S0 S0 S0 S0 S0 S0 S0 S0 S0 S0 S1 S1 S1 S1 S1 S1 S1 S1 S1 S1 S1 S1 S1 S1 S1 S2S2S2S2S2S2S2S2S2S2S2S2S2S2S2Sh MsI/I5BH*.F?>H*? H ;H H H* I*H. .I1H. 2I5 II II IIIoriginal_decoratordecorator_argsdecorator_kwargsc dfd }|S)a Create an instrumented version of an MCP decorator. This function intercepts MCP decorators (like @server.call_tool()) and injects Sentry instrumentation into the handler registration flow. The returned decorator will: 1. Receive the user's handler function 2. Pass the instrumented version to the original MCP decorator This ensures that when the handler is called at runtime, it's already wrapped with Sentry spans and metrics collection. Args: original_decorator: The original MCP decorator method (e.g., Server.call_tool) handler_type: "tool", "prompt", or "resource" - determines span configuration decorator_args: Positional arguments to pass to the original decorator (e.g., self) decorator_kwargs: Keyword arguments to pass to the original decorator Returns: A decorator function that instruments handlers before registering them rrrc`tdfd }i|S)Nargsrrc:Kt|dd{VS)NF)rr)rrrIs r!wrapperzO_create_instrumented_decorator..instrumented_decorator..wrapper s1),dPUVVVVVVVVV Vr#)rrrrr)rrrrrIrs` r!instrumented_decoratorz>_create_instrumented_decorator..instrumented_decorator sc t W W W W W W  WG!!>F5EFFwOOOr#)rrrrr()rrIrrrs```` r!_create_instrumented_decoratorrsD8PPPPPPPPP "!r#ctj d fd }|t_tj d fd }|t_tj d fd }|t_d S) zW Patches the mcp.server.lowlevel.Server class to instrument handler execution. r rkwargsrr2Callable[[Callable[..., Any]], Callable[..., Any]]cfdS)zEPatched version of Server.call_tool that adds Sentry instrumentation.c4tdfi|S)NrLr)rroriginal_call_toolr s r!zC_patch_lowlevel_server..patched_call_tool..s8 :   06  r#r()r rrs``r!patched_call_toolz1_patch_lowlevel_server..patched_call_tools( r#cfdS)zFPatched version of Server.get_prompt that adds Sentry instrumentation.c8td|SNrMr)roriginal_get_promptr s r!rzD_patch_lowlevel_server..patched_get_prompt..,s+ : 4  r#r()r rs`r!patched_get_promptz2_patch_lowlevel_server..patched_get_prompt(" r#cfdS)zIPatched version of Server.read_resource that adds Sentry instrumentation.c8td|SNrr)roriginal_read_resourcer s r!rzG_patch_lowlevel_server..patched_read_resource..9s+ : "J  r#r()r rs`r!patched_read_resourcez5_patch_lowlevel_server..patched_read_resource5rr#N)r rrrrr)r rrr)r call_tool get_prompt read_resource)rrrrrrs @@@r!r%r%s  )"' =)F!+ =+F$1 =1Fr#ctjt d fd }|t_dS) Nr rr;rreceiversendrrcKtj|did<tj|dd<||||d{VdS)Nr5r6r7)rget_isolation_scope setdefaultget_current_scope)r r;rroriginal_handle_requests r!patched_handle_requestz5_patch_handle_request..patched_handle_requestCsz  * , , "%%&BC6@5Q5S5Sg12%%dE7DAAAAAAAAAAAr#) r rr;rrrrrrN)rhandle_requestr)rrs @r!r&r&@s;J "## B- B B B B  B B B B B$# B4J!000r#cNttdr;tjt d fd }|t_ttdr=tjt d fd }|t_d Sd S) aY Patches the standalone fastmcp package's FastMCP class. The standalone fastmcp package (v2.14.0+) registers its own handlers for prompts and resources directly, bypassing the Server decorators we patch. This function patches the _get_prompt_mcp and _read_resource_mcp methods to add instrumentation for those handlers. _get_prompt_mcpr rrrrc:Ktd|||d{VSrr)r rroriginal_get_prompt_mcps r!patched_get_prompt_mcpz._patch_fastmcp..patched_get_prompt_mcp_sH*'  r#_read_resource_mcpc:Ktd|||d{VSrr)r rroriginal_read_resource_mcps r!patched_read_resource_mcpz1_patch_fastmcp..patched_read_resource_mcppsH**  r#N)r rrrrrrr)r:rrrr)rrrrs @@r!r'r'Ssw)**9")"9 & ' '   % 16        ( ' #9w,--?%,%?" ) * *   % 16        + * &?"""??r#)rr2)rr>)rirrr)N)NNTr*)6__doc__r functoolsrtypingrrsentry_sdk.ai.utilsrsentry_sdk.constsrrsentry_sdk.integrationsr r sentry_sdk.utilsr sentry_sdk.scoper $sentry_sdk.integrations._wsgi_commonr mcp.server.lowlevelrmcp.server.lowlevel.serverrmcp.server.streamable_httpr ImportErrorfastmcprrrrrrstarlette.typesrrrrr=rHrsrXrhrzrrr0rrr%r&r'r(r#r!rsb 777777********========++++++444444<<<<<<0******666666HHHHHHH000 ,. / //0GGG5EEEEEEEEEEEEEE4444444444[4M,'1'1'1'1TFF"%F*FFFF6 F  F F F F  F  F  F F  F F F FF%%%%PM MM1@MPSM MMMMn37>>>$>0>@ >>>>J37  lll l%l0 l  l  l llll^$",$"$"$" $"  $"$"$"$"N)1)1)1)1XJJJJ&)?)?)?)?)?)?sAA$(A//A98A9