KiV)dZddlmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZerdd lmZd Zd Zd Zd ZdZd Z d#dedzdedzdeedzdeeeefzdzdeedzdeegefdzdeeefdzdedzdeedefgedefffdZddddddddddededzdedzdedzdedzdeedzdedzdeeefdzdedzdeedefgedefffdZ d$dedzdedzdedzdeedzdeeefdzdedzdeedefgedefffd Z Gd!d"Z!y)%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 ldtdtfdtdtfffd }|S)zADecorator to mark a method as an MCP tool for later registration.func.rc  xs t|  d}|jDcic] \}}| || }}}t|t||Scc}}w)N)rrrrrrrr)r itemssetattr_MCP_REGISTRATION_TOOL_ATTR) r call_argskvrrrrrrrrs o/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/contrib/mcp_mixin/mcp_mixin.py decoratorzmcp_tool..decorator%sl-K-&&($  '0oo&7Ida1=QTI I19= Js AArr) rrrrrrrrr%s ```````` r$mcp_toolr's5 c* xS/A   )rtitler mime_typerrrrurir)r*c pdtdtfdtdtfff d } | S)zEDecorator to mark a method as an MCP resource for later registration.r.rc  xs t|  d }|jDcic] \}}| || }}}t|t||Scc}}w)N) r+rr)rr*rrrr)r rr_MCP_REGISTRATION_RESOURCE_ATTR) rr!r"r#rrrrr*rrr)r+s r$r%zmcp_resource..decoratorEso-K-&"&  '0oo&7Ida1=QTI I5yA Js AAr&) r+rr)rr*rrrrr%s ````````` r$ mcp_resourcer/7s6c*xS/A$ r(cddtdtfdtdtfffd }|S)zCDecorator to mark a method as an MCP prompt for later registration.r.rcxs t| d}|jDcic] \}}| || }}}t|t||Scc}}w)N)rr)rrrr)r rr_MCP_REGISTRATION_PROMPT_ATTR) rr!r"r#rrrrrr)s r$r%zmcp_prompt..decoratordsf-K-&  '0oo&7Ida1=QTI I3Y? Js AAr&)rr)rrrrr%s`````` r$ mcp_promptr3Zs5 c* xS/A   r(c eZdZdZdefdZdefdddedzded dfd Zdefdddedzded dfd Z de fdddedzded dfd Z deee fdddedzd ededed df dZ y)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_typec t|Dcgc]]}tt||rFtt|||r0t||tt|||j f_c}Scc}w)z>Retrieves all methods marked for a specific registration type.)dircallablegetattrhasattrcopy)selfr6 method_names r$_get_methods_to_registerz!MCPMixin._get_methods_to_registersr #4y  k23k24EF k*k24EFKKM    sA"A3N mcp_serverr prefix separatorrc|jtD]\}}|r|||d|d<tj||j d|j d|j d|j d|j d|j d|j d|j d|j d  }|j |y ) 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) fnrrrrrrrDrrN)r?r r from_functiongetadd_tool)r=r@rArBmethodregistration_infotools r$register_toolszMCPMixin.register_toolss*.)F)F '*  & %F%hyk*;F*C)DE"&)%%&**62-11-@&**62-11-@.22>B,00>/33OD&**62)--i8 D    %+ &r(c|jtD]\}}|r|||d|d<|||d|d<tj||d|j d|j d|j d|j d|j d|j d|j d|j d  }|j |y ) 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 '+'. rr+r)rr*rrrr) rEr+rr)rr*rrrrN)r?r.r rFrG add_resource)r=r@rArBrIrJresources r$register_resourceszMCPMixin.register_resourcess  *.)F)F +*  . %F%hyk*;F*C)DE"&)hyk*;E*B)CD"%( --%e,&**62'++G4-11-@+// <&**62)--i8-11-@&**62 H  # #H -1 .r(c l|jtD]\}}|r|||d|d<tj||j d|j d|j d|j d|j d|j d}|j |y) 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)rErr)rrrrN)r?r2rrFrG add_prompt)r=r@rArBrIrJprompts r$register_promptszMCPMixin.register_promptss*.)F)F )*  * %F%hyk*;F*C)DE"&)))&**62'++G4-11-@&**62)--i8&**62F  ! !& )! *r(tool_separatorresource_separatorprompt_separatorc||j||||j||||j|||y)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 '_'). )rArBN)rLrPrT)r=r@rArUrVrWs r$ register_allzMCPMixin.register_allsE* JvP  6EWX jCSTr() __name__ __module__ __qualname____doc__strr?_DEFAULT_SEPARATOR_TOOLrL_DEFAULT_SEPARATOR_RESOURCErP_DEFAULT_SEPARATOR_PROMPTrTrYr(r$r5r5vs  #  "0 $&$&d $& $&  $&R"4 (.(.d (. (.  (.Z"2 **d * *  *H"5"= 9 UUd U U  U  U Ur(r5)NNNNNNNN)NNNNNN)"r]collections.abcrtypingrr mcp.typesrrfastmcp.prompts.promptrfastmcp.resources.resourcer fastmcp.tools.toolr fastmcp.utilities.typesr fastmcp.serverr r r.r2r_r`rar^setdictlistboolr'r/r3r5rbr(r$rose$%2)/#/&6"> :!" ;?%).2"& *t c(T/!4S>1D8  s)d"  #$t+  sCx.4 D[xS!"HS#X$667B"  &*"&    *  :  t  Tz  c(T/ t#  sCx.4  D[ xS!"HS#X$667 H" "&  * :t c(T/  sCx.4   D[ xS!"HS#X$6678^U^Ur(