o i  @sUdZddlmZddlmZddlmZmZddlZ ddl m Z ddlm Z ddl mZdd lmZdd lmZdd lmZmZmZdd lmZmZdd lmZeedZeed<GdddeZdede e fddZ!ej"e!dZ# d'dedee$dfdee%e$efdBdfde j&j'fddZ(ej"e(dZ)GdddeZ*dede e j&j+fdd Z,ej"e,dZ-ded!eee$Bd"fde e fd#d$Z.ej"e.dZ/Gd%d&d&eZ0dS)(z=A middleware for injecting tools into the MCP server context.)Sequence)Logger) AnnotatedAnyN)ReadResourceContents)Prompt)AnyUrl)override)Context)CallNext MiddlewareMiddlewareContext)Tool ToolResult) get_loggernameloggerc@seZdZdZdeefddZedee j j de e j j eefdeefdd Z edee j jde e j jefdefd d Zd S) ToolInjectionMiddlewarez2A middleware for injecting tools into the context.toolscCs||_dd|D|_dS)z)Initialize the tool injection middleware.cSsi|]}|j|qSr).0toolrrn/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/server/middleware/tool_injection.py sz4ToolInjectionMiddleware.__init__..N)_tools_to_inject_tools_to_inject_by_nameselfrrrr__init__s z ToolInjectionMiddleware.__init__context call_nextreturncsg|j||IdHS)zInject tools into the response.N)r)rr r!rrr on_list_toolssz%ToolInjectionMiddleware.on_list_toolscsF|jj|jvr|j|jj}|j|jjpidIdHS||IdHS)z'Intercept tool calls to injected tools.) argumentsN)messagerrrunr$)rr r!rrrr on_call_tool(s z$ToolInjectionMiddleware.on_call_toolN)__name__ __module__ __qualname____doc__rrrr r mcptypesListToolsRequestr r#CallToolRequestParamsrr'rrrrrs&   rr r"c|IdHS)z%List prompts available on the server.N) list_promptsr rrrr16r1)fnrz!The name of the prompt to render.r$z$The arguments to pass to the prompt.cs|j||dIdHS)z(Render a prompt available on the server.)rr$N) get_prompt)r rr$rrrr5@sr5c"eZdZdZdfdd ZZS)PromptToolMiddlewarez=A middleware for injecting prompts as tools into the context.r"Ncttg}tj|ddSN)r)list_prompts_toolget_prompt_toolsuperrr __class__rrrSzPromptToolMiddleware.__init__r"Nr(r)r*r+r __classcell__rrr=rr7Pr7cr0)z'List resources available on the server.N)list_resourcesr2rrrrDXr3rDuriz The URI of the resource to read.cs|j|dIdHS)z(Read a resource available on the server.)rEN) read_resource)r rErrrrFbsrFcr6)ResourceToolMiddlewarez?A middleware for injecting resources as tools into the context.r"Ncr8r9)list_resources_toolread_resource_toolr<rrr=rrrrr?zResourceToolMiddleware.__init__r@rArrr=rrGorCrG)N)1r+collections.abcrloggingrtypingrr mcp.typesr, mcp.server.lowlevel.helper_typesrrpydanticrtyping_extensionsr fastmcp.server.contextr $fastmcp.server.middleware.middlewarer r r fastmcp.tools.toolrrfastmcp.utilities.loggingrr(r__annotations__rlistr1 from_functionr:strdictr-GetPromptResultr5r;r7ResourcerDrHrFrIrGrrrrsd        !