~h3.dZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZdd lmZdd lmZeeZGd d Zy)zResource manager functionality.N)Callable)Any)AnyUrl)settings) NotFoundError ResourceError)Resource)ResourceTemplatematch_uri_template)DuplicateBehavior) get_loggerceZdZdZ ddedzdedzfdZ ddedefde d e dzd e dzd e dzd e e dzd e e zfdZ ddedefde d e dzd e dzd e dzd e e dzd e fdZdde de dzd e fdZ ddedefde d e dzd e dzd e dzd e e dzd e fdZ dde de dzd e fdZdee zd efdZdee zd e fdZdee zd e ezfdZd ee e ffdZd ee e ffdZy)ResourceManagerzManages FastMCP resources.Nduplicate_behaviormask_error_detailsci|_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 _templatesrrr __args__ ValueErrorjoinr)selfrrs ^/opt/mcp/mcp-sentiment/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 #5fn.uriname description mime_typetagsreturnc  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|ywN) annotation).0pr%s r zCResourceManager.add_resource_or_template_from_fn..Rs% ||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) rrrrr r!r"has_uri_paramshas_func_paramsr%s @r add_resource_or_template_from_fnz0ResourceManager.add_resource_or_template_from_fn6s. 32s  &&r*55<<>   _,,C{It  ,,C{It K rctjdtdtj||||||}|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)rrrr r!r")warningsrDeprecationWarningr from_function add_resource)rrrrr r!r"resources rr5z$ResourceManager.add_resource_from_fnfsP0  o  ))#    **rrAkeyc|xst|j}tjd|j|t |j |j d|jj|}|r|jdk(r(tjd|||j|<nM|jdk(r||j|<n.|jdk(rtd||jdk(r|S||j|<|S) zAdd a resource to the manager. Args: resource: A Resource instance to add key: Optional URI to use as the storage key (if different from resource.uri) zAdding resource)r storage_keytype resource_nameextrarzResource already exists: replaceerrorignore) strrloggerdebugrE__name__rrgetrwarningr)rrArBrDexistings rr@zResourceManager.add_resources.S.  ||*X//!)   ??&&{3 &&&0!:;-HI/7 ,((I5/7 ,((G3 #<[M!JKK((H4'/ $r uri_templatectjdtdtj||||||}|j |S)z"Create a template from a function.ziadd_template_from_fn is deprecated. Use ResourceTemplate.from_function() and call add_template() instead.r:r;)rSrr r!r")r=rr>r r? add_template)rrrSrr r!r"templates rr4z$ResourceManager.add_template_from_fnsP  w  $11 %#    **rrVct|j}|xs|}tjd||t |j |j d|jj|}|r|jdk(r(tjd|||j|<nM|jdk(r||j|<n.|jdk(rtd||jdk(r|S||j|<|S) agAdd a template to the manager. Args: template: A ResourceTemplate instance to add key: Optional URI template to use as the storage key (if different from template.uri_template) Returns: The added template. If a template with the same URI already exists, returns the existing template. zAdding template)rSrDrE template_namerGrzTemplate already exists: rIrJrK) rLrSrMrNrErOrrrPrrQr)rrVrBuri_template_strrDrRs rrUzResourceManager.add_templatesx445--   0*X//!)   ??&&{3 &&&0!:;-HI/7 ,((I5/7 ,((G3 #<[M!JKK((H4'/ $rct|}||jvry|jjD]}t ||syy)zCheck if a resource exists.TF)rLrrkeysr )rruri_str template_keys r has_resourcezResourceManager.has_resourcesHc( doo % OO002 L!'<8 rc8Kt|}tjdd|i|jj |x}r|S|j j D]1\}}t||x}s |j||d{cStd|7#t$r}tjd||d}~wt$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 resourcerrG)paramsNz'Error creating resource from template: z%Error creating resource from templatezUnknown resource: )rLrMrNrrPritemsr create_resourcerrJ Exceptionrrr)rrr\rArDrVr`es r get_resourcezResourceManager.get_resources:c( 'w/? @**73 38 3O&*__%:%:%< ! !K+G[AAvA!!)!9!9%":"  !00 :;;) %LL#J1#!NOG !LL#J1#!NO..()PQWXX)EaSI ! !sHA2D6B$ B" B$D"B$$ D-C D?DDDczK|j|d{} |jd{S77#t$r"}tj d|d||d}~wt $rM}tj d|d||j rtd||td|d||d}~wwxYww)zRead a resource contents.NzError reading resource z: )rereadrrMrJrcr)rrrArds r read_resourcezResourceManager.read_resource!s**3// S!( (0)  LL23'A3? @G S LL23'A3? @&&#&=cW$EFAM$&=cWBqc$JKQRR SsGB;5B;979B;9 B8A B8+AB33B88B;c|jS)z+Get all registered resources, keyed by URI.)rrs r get_resourceszResourceManager.get_resources7 rc|jS)z4Get all registered templates, keyed by URI template.)rrjs r get_templateszResourceManager.get_templates;rlr)NN)NNNNr))rO __module__ __qualname____doc__r boolrrrrLsetr r r8r5r@r4rUrr^rebytesrhdictrkrnrrrrsR$8<*.5-45!4K5@ "& $ $. S#X ..Dj . 4Z . : .#ho. $ $.h "& $ $%+ S#X %+%+Dj %+ 4Z %+ : %+#ho%+ %+NXC$J(F "& $ $+ S#X ++Dj + 4Z + : +#ho+ +6=A$($/2Tz$ $L )rsC%$;//0 H ffr