o inF@s(dZddlmZddlZddlmZddlmZddl m Z m Z m Z ddl Z ddlmZddlZddlmZddlmZeGd d d ZeGd d d ZeGd ddZeGdddZeGdddZd+ddZd,ddZd-ddZGdddeeZd.d"d#Zd/d$d%Z d0d1d)d*Z dS)2z+Utilities for inspecting FastMCP instances.) annotationsN) dataclass)Enum)AnyLiteralcast)FastMCP)Clientc@seZdZUdZded<ded<ded<ded<d Zd ed <d Zd ed <d Zd ed<d Zded<d Z ded<d Z ded<d Z d ed<d S)ToolInfozInformation about a tool.strkeyname str | None descriptiondict[str, Any] input_schemaNdict[str, Any] | None output_schemarlist[str] | Nonetags bool | Noneenabledtitlelist[dict[str, Any]] | Noneiconsmeta) __name__ __module__ __qualname____doc____annotations__rrrrrrrr!r!_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/utilities/inspect.pyr        r c@sreZdZUdZded<ded<ded<dZded <dZd ed <dZd ed <dZded<dZ ded<dZ ded<dS) PromptInfozInformation about a prompt.r r r rrNr argumentsrrrrrrrr) rrrrr r%rrrrrr!r!r!r"r$#s      r$c@seZdZUdZded<ded<ded<ded<dZded <dZd ed <dZd ed <dZded<dZ ded<dZ ded<dZ d ed<dS) ResourceInfozInformation about a resource.r r urirr rN mime_typerrrrrrrrrr) rrrrr r(rrrrrrr!r!r!r"r&2r#r&c@seZdZUdZded<ded<ded<ded<dZded <dZd ed <dZd ed <dZd ed<dZ ded<dZ ded<dZ ded<dZ d ed<dS) TemplateInfoz&Information about a resource template.r r uri_templaterr rNr(r parametersrrrrrrrrr) rrrrr r(r+rrrrrrr!r!r!r"r)Cs        r)c@szeZdZUdZded<ded<ded<ded<ded <ded <ded <d ed <ded<ded<ded<ded<ded<dS) FastMCPInfoz.Information extracted from a FastMCP instance.r r r instructionsversion website_urlrrfastmcp_version mcp_versionintserver_generationzlist[ToolInfo]toolszlist[PromptInfo]promptszlist[ResourceInfo] resourceszlist[TemplateInfo] templatesr capabilitiesN)rrrrr r!r!r!r"r,Us  r,mcp FastMCP[Any]returncs|IdH}|IdH}|IdH}|IdH}g}|D]F}|j|jd}|t|j|jp4|j|j |j r<|j ni|j |j rG|j nd|jrPt|jnd|j|j|jr`dd|jDnd|jd q!g}|D]9} |t| j| jpx| j| j | jrdd| jDnd| jrt| jnd| j| j| jrdd| jDnd| jd qlg} |D]7} | t| j| j| j| j | j| j r| j nd| jrt| jnd| j| j| jrd d| jDnd| jd qg} |D];} | t| j| j| j| j | j| j| j r| j nd| jr t| jnd| j| j| jrd d| jDnd| jd qd didddd diid}t|jdrE|jjrEdd|jjDnd}t|jdrR|jjnd}t|j|jt|drb|jn|jj||t j!t"j#dd||| | |d S)zExtract information from a FastMCP v2.x instance. Args: mcp: The FastMCP v2.x instance to inspect Returns: FastMCPInfo dataclass containing the extracted information N)r cSg|]}|qSr! model_dump.0iconr!r!r" z&inspect_fastmcp_v2.. r r rrrrrrrrrcSr<r!r=r@argr!r!r"rBrCcSr<r!r=r?r!r!r"rBrC r r rr%rrrrrcSr<r!r=r?r!r!r"rBrC r r'r rr(rrrrrrcSr<r!r=r?r!r!r"rBrC r r*r rr(r+rrrrrr listChangedTF subscriberJr4r6r5loggingrcSr<r!r=r?r!r!r"rBrCr/r.r9 r r-r.r/rr0r1r3r4r5r6r7r8)$_list_tools_middleware_list_prompts_middleware_list_resources_middleware#_list_resource_templates_middleware to_mcp_toolr appendr r r inputSchemarrr>rlistrrrrr$r%r&r(r)r+hasattr _mcp_serverr/r,r-r.fastmcp __version__ importlibmetadata)r9 tools_list prompts_listresources_listtemplates_list tool_infostoolmcp_tool prompt_infospromptresource_infosresourcetemplate_infostemplater8 server_iconsserver_website_urlr!r!r"inspect_fastmcp_v2hs       rn FastMCP1xcst|4IdHW}|IdH}|IdH}|IdH}z |IdH}Wn ty4g}Ynwg}|D]-}|t|j|j|j |j rJ|j nidddddt |dr`|j r`dd|j Dnddd q9g}|D]7} d} t | dr| j rdd| j D} |t| j| j| j | dddt | dr| j rdd| j Dnddd qkg} |D]-} | tt| jt| j| j| j | jddddt | dr| j rd d| j Dnddd qg} |D]0}| tt|jt|j|j|j |jdddddt |dr|j rd d|j Dnddd qddidddddiid}|jj}t |dr.|j r.dd|j Dnd}t |dr9|jnd}t|jj|jj|jj||tjtjdd||| | |d WdIdHS1IdHshwYdS)zExtract information from a FastMCP v1.x instance using a Client. Args: mcp: The FastMCP v1.x instance to inspect Returns: FastMCPInfo dataclass containing the extracted information NrcSr<r!r=r?r!r!r"rBrCz&inspect_fastmcp_v1..rDr%cSr<r!r=rEr!r!r"rB%rCcSr<r!r=r?r!r!r"rB0rCrGcSr<r!r=r?r!r!r"rBErCrHcSr<r!r=r?r!r!r"rB[rCrIrJTFrKrMcSr<r!r=r?r!r!r"rBmrC websiteUrlr9rP) r list_tools list_promptslist_resourceslist_resource_templates ExceptionrVr r rrWrYrr%r$r&r r'mimeTyper) uriTemplateinitialize_result serverInforpr,rZr-r.r[r\r]r^)r9client mcp_tools mcp_prompts mcp_resources mcp_templatesrcrerf mcp_promptr%rh mcp_resourcerj mcp_templater8 server_inforlrmr!r!r"inspect_fastmcp_v1s   2rFastMCP[Any] | FastMCP1xcs2t|tr t|IdHStttt|IdHS)a`Extract information from a FastMCP instance into a dataclass. This function automatically detects whether the instance is FastMCP v1.x or v2.x and uses the appropriate extraction method. Args: mcp: The FastMCP instance to inspect (v1.x or v2.x) Returns: FastMCPInfo dataclass containing the extracted information N) isinstancerorrnrrr)r9r!r!r"inspect_fastmcps rc@seZdZdZdZdZdS) InspectFormatz"Output format for inspect command.r[r9N)rrrrFASTMCPMCPr!r!r!r"rsrinfobytescCsP|j|j|j|j|j|j|jd|j|jd|j |j |j |j d}t j|ddS)zFormat FastMCPInfo as FastMCP-specific JSON. This includes FastMCP-specific fields like tags, enabled, annotations, etc. )r r-r.r/r generationr8r[r9)server environmentr4r5r6r7rOindent)r r-r.r/rr3r8r0r1r4r5r6r7 pydantic_coreto_json)rresultr!r!r"format_fastmcp_infos" rc st|4IdHH}|IdH}|IdH}|IdH}|IdH}|jj}tjt j dd|i|j |j |j|jd}tj|ddWdIdHS1IdHsWwYdS)zFormat server info as standard MCP protocol JSON. Uses Client to get the standard MCP protocol format with camelCase fields. Includes version metadata at the top level. Nr9r)rrzr8r4r5r6resourceTemplatesrOr)r list_tools_mcplist_prompts_mcplist_resources_mcplist_resource_templates_mcpryrzr[r\r]r^r.r4r5r6rrr)r9r{ tools_resultprompts_resultresources_resulttemplates_resultrrr!r!r"format_mcp_infos&  0rformat)InspectFormat | Literal['fastmcp', 'mcp']FastMCPInfo | Nonecsbt|tr t|}|tjkrt|IdHS|tjkr*|dur&t|IdH}t|Std|)a0Format server information according to the specified format. Args: mcp: The FastMCP instance format: Output format ("fastmcp" or "mcp") info: Pre-extracted FastMCPInfo (optional, will be extracted if not provided) Returns: JSON bytes in the requested format NzUnknown format: ) rr rrrrrr ValueError)r9rrr!r!r" format_infos   r)r9r:r;r,)r9ror;r,)r9rr;r,)rr,r;r)r9rr;r)N)r9rrrrrr;r)!r __future__rimportlib.metadatar] dataclassesrenumrtypingrrrrmcp.server.fastmcprror[r fastmcp.server.serverr r$r&r)r,rnrrr rrrrr!r!r!r"s>           $