fiTN0dZddlZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZ dd lmZdd lmZn#e$r ed wxYwer dd lmZmZmZGdde ZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dS)a 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. N)wraps) TYPE_CHECKING)get_start_span_function)OPSPANDATA) Integration DidNotEnable)safe_serialize)should_send_default_pii)Server) request_ctxzMCP SDK not installed)AnyCallableOptionalc4eZdZdZdZddZedZdS)MCPIntegrationmcpz auto.ai.mcpTc||_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)include_prompts)selfrs k/var/www/html/Pagina-Ingenieria-Requisitos/venv/lib/python3.11/site-packages/sentry_sdk/integrations/mcp.py__init__zMCPIntegration.__init__$s /c"tdS)zM Patches MCP server classes to instrument handler execution. N)_patch_lowlevel_serverrr setup_oncezMCPIntegration.setup_once/s      rN)T)__name__ __module__ __qualname__ identifieroriginr staticmethodrrrrrr sMJ F / / / /!!\!!!rrcd}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" Nstdiorequest query_params session_idsseheaderszmcp-session-idhttp)r get request_idhasattrr&r'r* LookupError)r-r( mcp_transportctxr&s r_get_request_context_datar28sJJM o ?JsI&& G3;+B+7N33 G8L8P8P 99 G%*M!(!5!9!9,!G!GJJWi00GW_5H5H$66G%+M!(!4!45E!F!FJ       z= 00sCC CCc|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) handler_type item_name span_data_keymcp_method_nameresult_data_key span_names r_get_span_configrBcs|v . &":  ! ! 0 '"D 1 *"00Y00I )_o EErc ||||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 ) span handler_namer>r? argumentsr-r(r0kvs r_set_span_input_datarR}s  MM-...MM(*O<<< MM"mw.F.FFFE MM((-888; h-z:::; h-z:::!!FF1 1a11>!3D3DEEEEFFrc&|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__textr6) isinstancetuplelendictr.strbytesappendrW Exceptionjoin)resulttextsitems r_extract_tool_result_contentrds?~t&%  S[[A%5%5ay&$ vz"" 4:fsE4>P+Q+Q 4  / /4((/LL++++d++/$LLf...  /    MMM #(3sxxV3 MsA$C&& C54C5c|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.Nr4r5rmessagesrUrolecontentrW) sentry_sdk get_clientget_integrationrr rrdrFr rXr[rMCP_TOOL_RESULT_CONTENT_COUNTrZr.rfr,MCP_PROMPT_RESULT_MESSAGE_COUNTrgMCP_PROMPT_RESULT_MESSAGE_ROLErhrWr\r_) rMrar@r< integrationshould_include_data extractedrf message_count first_messagerg content_text msg_contents r_set_span_output_datarvs`~'))99.IIK233S 8Sv088  %8 MM/>)+D+D E E E)T** V hDc)nnUUUUU !  V V  ! !4 HMvz** .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 KKc6|dkr&|d}t|dkr|dni}nQ|dkr.|d}t|dkr|dni}d|i|pi}n|d}|rt|nd}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) r4rrUr5nameunknown)rZr\rB) r< original_argsrNrOurir>rAr?r@s r_prepare_handler_datar|sv$Q' (+M(:(:Q(>(>M!$$B  ! !$Q' (+M(:(:Q(>(>M!$$B \?io2? A#&5s3xxxI  BRlBB>M9o   rc Kt||\}}}}}}ttj|tj5} t \} } } t| ||||| | | |dkre|d} d}t| dr| j }n!|rd|vr| dd}|r | tj | ||d{V}nM#t$r@}|dkr | tjdt!j|d}~wwxYwt%| ||||cdddS#1swxYwYdS) z Async wrapper for MCP handlers. Args: handler_type: "tool", "prompt", or "resource" func: The async handler function to wrap original_args: Original arguments passed to the handler oprxr"resourcerNscheme://r4Tr|rr MCP_SERVERrr"r2rRr.rsplitrFrMCP_RESOURCE_PROTOCOLr_MCP_TOOL_RESULT_IS_ERRORricapture_exceptionrvr<funcrzrNrOr>rAr?r@rMr-r(r0r{protocolraes r_async_handler_wrapperrBs" lM:: # " " = $   * 0I0K0K- J            : % %"CHsH%% 8: 8%<"7"7'--e44Q7 H h.async_wrappers-0 dDIIIIIIII Irc&t|Sr)rrs r sync_wrapperz2_create_instrumented_handler..sync_wrappers)tTBB Br)inspectiscoroutinefunctionr)r<rrrs`` r_create_instrumented_handlerrs$"4(( t J J J J J  J t C C C C C  Crcfd}|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. Wrap it with instrumentation via _create_instrumented_handler 3. 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 cFt|}i|Sr)r)rinstrumented_funcdecorator_argsdecorator_kwargsr<original_decorators rinstrumented_decoratorz>_create_instrumented_decorator..instrumented_decorators>9tLLF!!>F5EFF    rr)rr<rrrs```` r_create_instrumented_decoratorrs66         "!rctjfd}|t_tjfd}|t_tjfd}|t_dS)zW Patches the mcp.server.lowlevel.Server class to instrument handler execution. cfdS)zEPatched version of Server.call_tool that adds Sentry instrumentation.c4tdfi|S)Nr4r)rkwargsoriginal_call_toolrs rzC_patch_lowlevel_server..patched_call_tool..s8 :   06  rr)rrrs``rpatched_call_toolz1_patch_lowlevel_server..patched_call_tools( rcfdS)zFPatched version of Server.get_prompt that adds Sentry instrumentation.c8td|S)Nr5r)roriginal_get_promptrs rrzD_patch_lowlevel_server..patched_get_prompt..&s+ : 4  rr)rrs`rpatched_get_promptz2_patch_lowlevel_server..patched_get_prompt#" rcfdS)zIPatched version of Server.read_resource that adds Sentry instrumentation.c8td|S)Nrr)roriginal_read_resourcers rrzG_patch_lowlevel_server..patched_read_resource..2s+ : "J  rr)rrs`rpatched_read_resourcez5_patch_lowlevel_server..patched_read_resource/rrN)r call_tool get_prompt read_resource)rrrrrrs @@@rrrs  ))F!++F$11Fr)'__doc__r functoolsrtypingrrisentry_sdk.ai.utilsrsentry_sdk.constsrrsentry_sdk.integrationsrr sentry_sdk.utilsr sentry_sdk.scoper mcp.server.lowlevelr mcp.server.lowlevel.serverr ImportErrorrrrrr2rBrRrdrvr|rrrrrrrrrs 777777********========++++++4444440******6666666000 ,. / //0/..........!!!!![!!!0(1(1(1VFFF4!F!F!FH&&&RLLLf"""J>>>B>>>B!!!H$"$"$"N'1'1'1'1'1s AA