o i3@sdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZdd lmZdd lmZmZdd lmZdd lmZeeZGd ddZdS)zResource manager functionality.) annotationsN)Callable)Any)AnyUrl)settings) NotFoundError ResourceError)Resource)ResourceTemplatematch_uri_template)DuplicateBehavior) get_loggerc@seZdZdZ  d3d4ddZd5d d Zd6ddZ    d7d8ddZ    d7d9ddZd:d!d"Z    d7d;d%d&Z dd.d/Z d?d1d2ZdS)@ResourceManagerzManages FastMCP resources.Nduplicate_behaviorDuplicateBehavior | Nonemask_error_details bool | NonecCsTi|_i|_|p tj|_|durd}|tjvr%td|ddtj||_dS)aInitialize the ResourceManager. Args: duplicate_behavior: How to handle duplicate resources (warn, error, replace, ignore) mask_error_details: Whether to mask error details from exceptions other than ResourceError NwarnzInvalid duplicate_behavior: z. Must be one of: z, ) _resources _templatesrrr __args__ ValueErrorjoinr)selfrrrh/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/resources/resource_manager.py__init__s     zResourceManager.__init__returndict[str, Resource]c t|jS)z+Get all registered resources, keyed by URI.)dictrrrrr get_resources8 zResourceManager.get_resourcesdict[str, ResourceTemplate]cr)z4Get all registered templates, keyed by URI template.)r rr!rrrget_resource_templates<r#z&ResourceManager.get_resource_templatesfnCallable[..., Any]uristrname str | None description mime_typetagsset[str] | NoneResource | ResourceTemplatec s~ddlmd|vo d|v}tfddt|jD}|s#|r-|||||||S|s;|s;|||||||St d)aAdd a resource or template to the manager from a function. Args: fn: The function to register as a resource or template uri: The URI for the resource or template name: Optional name for the resource or template description: Optional description of the resource or template mime_type: Optional MIME type for the resource or template tags: Optional set of tags for categorizing the resource or template Returns: The added resource or template. If a resource or template with the same URI already exists, returns the existing resource or template. rContext{}c3s|] }|jur|VqdS)N) annotation).0pr1rr \s zCResourceManager.add_resource_or_template_from_fn..ziInvalid resource or template definition due to a mismatch between URI parameters and function parameters.) fastmcp.server.contextr2anyinspect signature parametersvaluesadd_template_from_fnadd_resource_from_fnr) rr&r(r*r,r-r.has_uri_paramshas_func_paramsrr1r add_resource_or_template_from_fn@s    z0ResourceManager.add_resource_or_template_from_fnr cC6tjr tjdtddtj||||||d}||S)a#Add a resource to the manager from a function. Args: fn: The function to register as a resource uri: The URI for the resource name: Optional name for the resource description: Optional description of the resource mime_type: Optional MIME type for the resource tags: Optional set of tags for categorizing the resource Returns: The added resource. If a resource with the same URI already exists, returns the existing resource. zaadd_resource_from_fn is deprecated. Use Resource.from_function() and call add_resource() instead. stacklevel)r&r(r*r,r-r.)rdeprecation_warningswarningsrDeprecationWarningr from_function add_resource)rr&r(r*r,r-r.resourcerrrr@ps z$ResourceManager.add_resource_from_fnrMcC|j|j}|r>|jdkrtd|j||j|j<n |jdkr*||j|j<n|jdkr7td|j|jdkr>|S||j|j<|S)aAdd a resource to the manager. Args: resource: A Resource instance to add. The resource's .key attribute will be used as the storage key. To overwrite it, call Resource.model_copy(key=new_key) before calling this method. rzResource already exists: replaceerrorignore)rgetkeyrloggerwarningr)rrMexistingrrrrLs     zResourceManager.add_resource uri_templater cCrD)z"Create a template from a function.ziadd_template_from_fn is deprecated. Use ResourceTemplate.from_function() and call add_template() instead.rErF)rWr*r,r-r.)rrHrIrrJr rK add_template)rr&rWr*r,r-r.templaterrrr?s  z$ResourceManager.add_template_from_fnrYcCrN)aAdd a template to the manager. Args: template: A ResourceTemplate instance to add. The template's .key attribute will be used as the storage key. To overwrite it, call ResourceTemplate.model_copy(key=new_key) before calling this method. Returns: The added template. If a template with the same URI already exists, returns the existing template. rzTemplate already exists: rOrPrQ)rrRrSrrTrUr)rrYrVrrrrXs      zResourceManager.add_template AnyUrl | strboolcsTt|}|IdH}||vrdS|IdH}|D] }t||dur'dSqdS)zCheck if a resource exists.NTF)r)r"r%r )rr(uri_str resources templates template_keyrrr has_resourceszResourceManager.has_resourcec s t|}tjdd|id|IdH}||}r|S|IdH}|D]T\}}t||}dur}z |j||dIdHWSt yY} z t d| | d} ~ wt y|} zt d| |j rpt d| t d| | d} ~ wwq)td|) zGet resource by URI, checking concrete resources first, then templates. Args: uri: The URI of the resource to get Raises: NotFoundError: If no resource or template matching the URI is found. zGetting resourcer()extraNparamsz'Error creating resource from template: z%Error creating resource from templatezUnknown resource: )r)rTdebugr"rRr%itemsr create_resourcerrP Exceptionrrr) rr(r\r]rMr^ storage_keyrYrcerrr get_resources>  zResourceManager.get_resource str | bytesc st|}||jvrZ||IdH}z|IdHWSty0}z td||d}~wtyY}ztd||jrJtd||td|d||d}~ww|j D]a\}}t ||}durz|j ||dIdH}|IdHWSty}z td||d}~wty}ztd||jrtd||td|d||d}~wwq_t d|d)zv Internal API for servers: Finds and reads a resource, respecting the filtered protocol path. NzError reading resource z: rbz%Error reading resource from template z Resource z not found.)r)rrjreadrrT exceptionrgrrrer rfr)rr(r\rMrirSrYrcrrr read_resource!sf  zResourceManager.read_resource)NN)rrrr)rr)rr$)NNNN)r&r'r(r)r*r+r,r+r-r+r.r/rr0)r&r'r(r)r*r+r,r+r-r+r.r/rr )rMr rr )r&r'rWr)r*r+r,r+r-r+r.r/rr )rYr rr )r(rZrr[)r(rZrr )r(rZrrk)__name__ __module__ __qualname____doc__rr"r%rCr@rLr?rXr`rjrnrrrrrs6   4 (   -r)rr __future__rr;rIcollections.abcrtypingrpydanticrfastmcprfastmcp.exceptionsrrfastmcp.resources.resourcer fastmcp.resources.templater r fastmcp.settingsr fastmcp.utilities.loggingr rorTrrrrrs