j iJdZddlmZddlmZmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZmZ ddlZd Zn #e$rd ZYnwxYwerdd lmZmZerdd lmZee zezZnee zZGd ded ZeGddZe Gdde ZdS)a[Interceptor interfaces and types for MCP client tool call lifecycle management. This module provides an interceptor interface for wrapping and controlling MCP tool call execution with a handler callback pattern. In the future, we might add more interceptors for other parts of the request / result lifecycle, for example to support elicitation. ) annotations) dataclassreplace) TYPE_CHECKINGAnyProtocolruntime_checkable) ToolMessage)CallToolResult) NotRequired TypedDictUnpackNTF) AwaitableCallable)Commandc2eZdZUdZded<ded<ded<dS) _MCPToolCallRequestOverrideszPossible overrides for MCPToolCallRequest.override() method. Only includes modifiable request fields, not context fields like server_name and runtime which are read-only. zNotRequired[str]namezNotRequired[dict[str, Any]]argsz"NotRequired[dict[str, Any] | None]headersN)__name__ __module__ __qualname____doc____annotations__C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain_mcp_adapters/interceptors.pyrr'sB %%%%//////rr)totalcVeZdZUdZded<ded<ded<dZded <dZd ed <ddZdS)MCPToolCallRequestaTool execution request passed to MCP tool call interceptors. This tool call request follows a similar pattern to LangChain's ToolCallRequest (flat namespace) rather than separating the call data and context into nested objects. Modifiable fields (override these to change behavior): name: Tool name to invoke. args: Tool arguments as key-value pairs. headers: HTTP headers for applicable transports (SSE, HTTP). Context fields (read-only, use for routing/logging): server_name: Name of the MCP server handling the tool. runtime: LangGraph runtime context (optional, None if outside graph). strrzdict[str, Any]r server_nameNzdict[str, Any] | Nonerz object | Noneruntime overrides$Unpack[_MCPToolCallRequestOverrides]returnc t|fi|S)aReplace the request with a new request with the given overrides. Returns a new `MCPToolCallRequest` instance with the specified attributes replaced. This follows an immutable pattern, leaving the original request unchanged. Args: **overrides: Keyword arguments for attributes to override. Supported keys: - name: Tool name - args: Tool arguments - headers: HTTP headers Returns: New MCPToolCallRequest instance with specified overrides applied. Note: Context fields (server_name, runtime) cannot be overridden as they are read-only. Examples: ```python # Modify tool arguments new_request = request.override(args={"value": 10}) # Change tool name new_request = request.override(name="different_tool") ``` )r)selfr%s roverridezMCPToolCallRequest.overrideKsBt))y)))r)r%r&r'r!)rrrrrrr$r*rrrr!r!3sy III%)G))))!G!!!!!*!*!*!*!*!*rr!ceZdZdZd dZd S) ToolCallInterceptoraProtocol for tool call interceptors using handler callback pattern. Interceptors wrap tool execution to enable request/response modification, retry logic, caching, rate limiting, and other cross-cutting concerns. Multiple interceptors compose in "onion" pattern (first is outermost). The handler can be called multiple times (retry), skipped (caching/short-circuit), or wrapped with error handling. Each handler call is independent. Similar to LangChain's middleware pattern but adapted for MCP remote tools. requestr!handlerr>s#"""""********BBBBBBBBBBBB//////$$$$$$<<<<<<<<<<4333333335''''''&4w>&4 0 0 0 0 09E 0 0 0 0 8*8*8*8*8*8*8* 8*v     (     s;AA