o i"@szdZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZeeZGd d d Zd S) z ComponentService: Provides async management of tools, resources, and prompts for FastMCP servers. Handles enabling/disabling components both locally and across mounted servers. ) NotFoundError)Prompt)Resource)ResourceTemplate)FastMCPhas_resource_prefixremove_resource_prefix)Tool) get_loggerc@seZdZdZdefddZdedefddZdedefd d Z dede e Bfd d Z dede e Bfd dZ dedefddZdedefddZdS)ComponentServicezCService for managing components like tools, resources, and prompts.servercCs"||_|j|_|j|_|j|_dS)N)_server _tool_manager_resource_manager_prompt_manager)selfr ry/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/contrib/component_manager/component_service.py__init__s zComponentService.__init__keyreturnctd|||jjjvr|j|IdH}||St|jjD])}|j rL| |j drK| |j d}t |j }||IdH}|Sq#q#td|)zHandle 'enableTool' requests. Args: key: The key of the tool to enable Returns: The tool that was enabled zEnabling tool: %sN_Unknown tool: )loggerdebugr r_toolsget_toolenablereversed_mounted_serversprefix startswith removeprefixr r _enable_toolrrrtoolmountedtool_keymounted_servicerrrr$   zComponentService._enable_toolcr)zHandle 'disableTool' requests. Args: key: The key of the tool to disable Returns: The tool that was disabled zDisable tool: %sNrr)rrr rrrdisablerr r!r"r#r r _disable_toolrr%rrrr,7r*zComponentService._disable_toolctd|||jjvr|j|IdH}||S||jjvr1|j|IdH}||St |jj D]'}|j r^t ||j |j r]t||j |j }t|j}||IdH}|Sq7q7td|)zHandle 'enableResource' requests. Args: key: The key of the resource to enable Returns: The resource that was enabled zEnabling resource: %sNUnknown resource: )rrr _resourcesr get_resourcer _templatesget_resource_templaterr r!rresource_prefix_formatrr r _enable_resourcerrrresourcetemplater'r)mounted_resourcerrrr4T:   z!ComponentService._enable_resourcecr-)zHandle 'disableResource' requests. Args: key: The key of the resource to disable Returns: The resource that was disabled zDisable resource: %sNr.)rrrr/r r0r+r1r2rr r!rr3rr r _disable_resourcerr5rrrr:r9z"ComponentService._disable_resourcecr)zHandle 'enablePrompt' requests. Args: key: The key of the prompt to enable Returns: The prompt that was enabled zEnabling prompt: %sNrUnknown prompt: )rrr r_prompts get_promptrrr r!r"r#r r _enable_promptrrrpromptr' prompt_keyr)rrrr>r*zComponentService._enable_promptcs||jjjvr|j|IdH}||St|jjD])}|jrF||jdrE| |jd}t |j }| |IdH}|Sqqt d|)zHandle 'disablePrompt' requests. Args: key: The key of the prompt to disable Returns: The prompt that was disabled Nrr;)r rr<r=r+rr r!r"r#r r _disable_promptrr?rrrrBs  z ComponentService._disable_promptN)__name__ __module__ __qualname____doc__rrstrr r$r,rrr4r:rr>rBrrrrr s++r N)rFfastmcp.exceptionsrfastmcp.prompts.promptrfastmcp.resources.resourcerfastmcp.resources.templaterfastmcp.server.serverrrrfastmcp.tools.toolr fastmcp.utilities.loggingr rCrr rrrrs