o iV)@s8dZddlmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZer8dd lmZd Zd Zd Zd ZdZd Z        d&dedBdedBdeedBdeeeefBdBdeedBdeegefdBdeeefdBdedBdeedefgedefffddZddddddddddededBdedBdedBdedBdeedBdedBdeeefdBdedBdeedefgedefffd d!Z      d'dedBdedBdedBdeedBdeeefdBdedBdeedefgedefffd"d#Z Gd$d%d%Z!dS)(z_Provides a base mixin class and decorators for easy registration of class methods with FastMCP.)Callable) TYPE_CHECKINGAny) AnnotationsToolAnnotations)Prompt)Resource)Tool) get_fn_name)FastMCP_mcp_tool_registration_mcp_resource_registration_mcp_prompt_registration_+Nname descriptiontags annotations exclude_args serializermetaenabledreturn.c s8dtdtfdtdtfffdd }|S)zADecorator to mark a method as an MCP tool for later registration.func.rc s@pt|d}dd|D}t|t||S)N)rrrrrrrrcSi|] \}}|dur||qSN.0kvrri/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/contrib/mcp_mixin/mcp_mixin.py 0z/mcp_tool..decorator..)r itemssetattr_MCP_REGISTRATION_TOOL_ATTRr call_argsrrrrrrrrrr" decorator%s  zmcp_tool..decoratorrr) rrrrrrrrr+rr*r"mcp_tools4 r-)rtitler mime_typerrrrurir.r/c s:dtdtfdtdtfff dd } | S)zEDecorator to mark a method as an MCP resource for later registration.r.rc sBpt|d }dd|D}t|t||S)N) r0rr.rr/rrrrcSrrrrrrr"r#Qr$z3mcp_resource..decorator..)r r%r&_MCP_REGISTRATION_RESOURCE_ATTRr( rrrrr/rrr.r0rr"r+Es  zmcp_resource..decoratorr,) r0rr.rr/rrrrr+rr2r" mcp_resource7s6r3cs4dtdtfdtdtfffdd }|S)zCDecorator to mark a method as an MCP prompt for later registration.r.rcs<pt|d}dd|D}t|t||S)N)rr.rrrrcSrrrrrrr"r#nr$z1mcp_prompt..decorator..)r r%r&_MCP_REGISTRATION_PROMPT_ATTRr(rrrrrr.rr"r+ds  zmcp_prompt..decoratorr,)rr.rrrrr+rr5r" mcp_promptZs0 r6c @seZdZdZdefddZdefdddedBd ed dfd d ZdefdddedBd ed dfd dZ de fdddedBd ed dfddZ deee fdddedBdededed df ddZ dS)MCPMixinaBase mixin class for objects that can register tools, resources, and prompts with a FastMCP server instance using decorators. This mixin provides methods like `register_all`, `register_tools`, etc., which iterate over the methods of the inheriting class, find methods decorated with `@mcp_tool`, `@mcp_resource`, or `@mcp_prompt`, and register them with the provided FastMCP server instance. registration_typecsfddtDS)z>Retrieves all methods marked for a specific registration type.csHg|] }tt|rtt|rt|tt|fqSr)callablegetattrhasattrcopy)r method_namer8selfrr" s z5MCPMixin._get_methods_to_register..)dir)r?r8rr>r"_get_methods_to_registers z!MCPMixin._get_methods_to_registerN mcp_serverr prefix separatorrcCs|tD]A\}}|r|||d|d<tj||d|d|d|d|d|d|d|d|d d }||qd S) aRegisters all methods marked with @mcp_tool with the FastMCP server. Args: mcp_server: The FastMCP server instance to register tools with. prefix: Optional prefix to prepend to tool names. If provided, the final name will be f"{prefix}{separator}{original_name}". separator: The separator string used between prefix and original name. Defaults to '_'. rrrrrr output_schemarr) fnrrrrrrrFrrN)rBr'r from_functiongetadd_tool)r?rCrDrEmethodregistration_infotoolrrr"register_toolss(  zMCPMixin.register_toolscCs|tD]L\}}|r#|||d|d<|||d|d<tj||d|d|d|d|d|d|d|d|d d }||qd S) aRegisters all methods marked with @mcp_resource with the FastMCP server. Args: mcp_server: The FastMCP server instance to register resources with. prefix: Optional prefix to prepend to resource names and URIs. If provided, the final name will be f"{prefix}{separator}{original_name}" and the final URI will be f"{prefix}{separator}{original_uri}". separator: The separator string used between prefix and original name/URI. Defaults to '+'. rr0r.rr/rrrr) rGr0rr.rr/rrrrN)rBr1rrHrI add_resource)r?rCrDrErKrLresourcerrr"register_resourcess,  zMCPMixin.register_resourcesc Csz|tD]5\}}|r|||d|d<tj||d|d|d|d|d|dd}||qdS) aRegisters all methods marked with @mcp_prompt with the FastMCP server. Args: mcp_server: The FastMCP server instance to register prompts with. prefix: Optional prefix to prepend to prompt names. If provided, the final name will be f"{prefix}{separator}{original_name}". separator: The separator string used between prefix and original name. Defaults to '_'. rr.rrrr)rGrr.rrrrN)rBr4rrHrI add_prompt)r?rCrDrErKrLpromptrrr"register_promptss"  zMCPMixin.register_promptstool_separatorresource_separatorprompt_separatorcCs4|j|||d|j|||d|j|||ddS)aRegisters all marked tools, resources, and prompts with the server. This method calls `register_tools`, `register_resources`, and `register_prompts` internally, passing the provided prefix and separators. Args: mcp_server: The FastMCP server instance to register with. prefix: Optional prefix applied to all registered items unless overridden by a specific separator argument. tool_separator: Separator for tool names (defaults to '_'). resource_separator: Separator for resource names/URIs (defaults to '+'). prompt_separator: Separator for prompt names (defaults to '_'). )rDrEN)rNrQrT)r?rCrDrUrVrWrrr" register_allszMCPMixin.register_all) __name__ __module__ __qualname____doc__strrB_DEFAULT_SEPARATOR_TOOLrN_DEFAULT_SEPARATOR_RESOURCErQ_DEFAULT_SEPARATOR_PROMPTrTrXrrrr"r7vsj  ) - $r7)NNNNNNNN)NNNNNN)"r\collections.abcrtypingrr mcp.typesrrfastmcp.prompts.promptrfastmcp.resources.resourcerfastmcp.tools.toolr fastmcp.utilities.typesr fastmcp.serverr r'r1r4r^r_r`r]setdictlistboolr-r3r6r7rrrr"s         !    $