Ki2dZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZmZdd lmZdd lmZmZdd lmZdd lmZeeZGd dZy)zResource manager functionality.) annotationsN)Callable)Any)AnyUrl)settings) FastMCPError NotFoundError ResourceError)Resource)ResourceTemplatematch_uri_template)DuplicateBehavior) get_loggerceZdZdZ d ddZddZddZ d ddZ d ddZddZ d dd Z dd Z dd Z dd Z dd Zy)ResourceManagerzManages FastMCP resources.Nci|_i|_|xstj|_|d}|tj vr.t d|ddjtj ||_y)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 _templatesrmask_error_detailsr__args__ ValueErrorjoinduplicate_behavior)selfrrs n/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/resources/resource_manager.py__init__zResourceManager.__init__s0279"4"S8S8S  %!'  %6%?%? ?./A.BC##'99->-G-G#H"IK #5c4Kt|jSw)z+Get all registered resources, keyed by URI.)dictrrs r get_resourceszResourceManager.get_resources8DOO$$c4Kt|jSw)z4Get all registered templates, keyed by URI template.)r rr!s rget_resource_templatesz&ResourceManager.get_resource_templates<r#r$c  ddlm d|vxrd|v}t fdtj|j j D}|s|r|j||||||S|s|s|j||||||Std)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. r)Context{}c3>K|]}|jur|yw)N) annotation).0pr(s r zCResourceManager.add_resource_or_template_from_fn..\s% ||7*  sziInvalid resource or template definition due to a mismatch between URI parameters and function parameters.) fastmcp.server.contextr(anyinspect signature parametersvaluesadd_template_from_fnadd_resource_from_fnr) rfnuriname description mime_typetagshas_uri_paramshas_func_paramsr(s @r add_resource_or_template_from_fnz0ResourceManager.add_resource_or_template_from_fn@s. 32s  &&r*55<<>   _,,C{It  ,,C{It K rctjrtjdtdt j ||||||}|j|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)r8r9r:r;r<r=)rdeprecation_warningswarningsrDeprecationWarningr from_function add_resource)rr8r9r:r;r<r=resources rr7z$ResourceManager.add_resource_from_fnpsY0  ( ( MMs"  ))#    **rc|jj|j}|r|jdk(rKt|}tjdd|i|jd{}|j |x}r|S|j d{}|j D]1\}}t||x} |j||d{cStd|77X7#t$rt$rD} tjd| |jr td| td| | d} ~ wwxYww) 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 resourcer9)extraNparamsz'Error creating resource from template: z%Error creating resource from templatezUnknown resource: )rZrQdebugr"rOr&itemsr create_resourcer ExceptionrMrrr ) rr9r[r\rJr] storage_keyrWrces r get_resourcezResourceManager.get_resourcesEc( 'w/? @,,..  }}W- -8 -O5577 %.__%6 ! !K,WkBBO!!)!9!9%":"  !.0 :;;;/ 8  $ !LL#J1#!NO..()PQWXX)EaSI ! !sW8DB>,D'C((DC(C)C,DDCD?DDDcKt|}||jvr2|j|d{} |jd{S|jjD]I\}}t||x} |j||d{}|jd{cStd|d77{#t$rt $rJ}t jd||jrtd||td|d||d}~wwxYw77{#t$rt $rJ}t jd||jrtd||td|d||d}~wwxYww)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.)rZrrjreadrrgrQ exceptionrr rrer rfr )rr9r[rJrirPrWrcs r read_resourcezResourceManager.read_resource s c( doo %!..w77H %]]_,,&"__224 !MC,Wc::G!%-%=%=gf%=%UUH!)00  !(i{+>??S8,     #:7+!FG**'*A'(MNTUU(1'BqcB  V0# !$$?{K..+CG;O !,CG;bQRPST ! !s.F CF C C C /F >D*D&D*,D(-D*0F CD#ADD##F &D*(D**FrsK%"$II//0 H y@y@r