kKjdZddlmZddlZddlmZddlmZddl Z ddlm Z ddl m Z ddl mZmZmZd gZej$eZGd d eZy) zAResponse limiting middleware for controlling tool response sizes.) annotationsN)Any) TextContent) ToolResult)CallNext MiddlewareMiddlewareContextResponseLimitingMiddlewarecZeZdZdZdddd d dZ d d dZ d dZy) r a#Middleware that limits the response size of tool calls. Intercepts tool call responses and enforces size limits. If a response exceeds the limit, it extracts text content, truncates it, and returns a single TextContent block. Example: ```python from fastmcp import FastMCP from fastmcp.server.middleware.response_limiting import ( ResponseLimitingMiddleware, ) mcp = FastMCP("MyServer") # Limit all tool responses to 500KB mcp.add_middleware(ResponseLimitingMiddleware(max_size=500_000)) # Limit only specific tools mcp.add_middleware( ResponseLimitingMiddleware( max_size=100_000, tools=["search", "fetch_data"], ) ) ``` i@Bz( [Response truncated due to size limit]N)max_sizetruncation_suffixtoolscz|dkrtd|||_||_|t||_yd|_y)a}Initialize response limiting middleware. Args: max_size: Maximum response size in bytes. Defaults to 1MB (1,000,000). truncation_suffix: Suffix to append when truncating responses. Defaults to "\n\n[Response truncated due to size limit]". tools: List of tool names to apply limiting to. If None, applies to all. rzmax_size must be positive, got N) ValueErrorr rsetr)selfr rrs |/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/fastmcp/server/middleware/response_limiting.py__init__z#ResponseLimitingMiddleware.__init__2sC q=>xjIJ J  !2#(#4SZ $ c~t|jjd}d}|j|z |z }|dkr |j}nR|jd}t||kr||jz}n#|d|j dd|jz}t t d|g|| Si S) z)  /1  rcpK||d{}|j$|jj|jvr|Stj|t }t ||jkr|Stjd|jjt ||j|jDcgc]}t|ts|j!}}|rdj|n|jdd}|j!||j"S7 cc}ww) z-Intercept tool calls and limit response size.N)fallbackzDTool %r response exceeds size limit: %d bytes > %d bytes, truncatingz rreplacer)r )rmessagename pydantic_coreto_jsonstrr!r loggerwarningr isinstancerrjoinr#r)r )rcontext call_nextresult serializedbtextsrs r on_call_toolz'ResponseLimitingMiddleware.on_call_toolis  !)) :: !goo&:&:$**&LM#**6C@ z?dmm +M  R OO  O MM  "(NA:a3MN KK ""79"= ''6;;'??5*&Os) D6D.B6D6D1D1)AD61D6)r intrr1rzlist[str] | NonereturnNone)N)rr1r zdict[str, Any] | Noner>r)r6z+MiddlewareContext[mt.CallToolRequestParams]r7z.CallNext[mt.CallToolRequestParams, ToolResult]r>r)__name__ __module__ __qualname____doc__rr)r<rrr r s>"!M"& ?? ?  ?  ?0'+    $     D @< @B @  @r)rC __future__rloggingtypingr mcp.typestypesmtr/rfastmcp.tools.baser middlewarerr r __all__ getLoggerr@r2r rDrrrOsMG"!)?? ' (   8 $t@t@r