o i@sdZddlmZddlZddlmZddlmZmZm Z ddl Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZmZer[ Gd ddeZGdddeZ dS)z2Base classes and interfaces for FastMCP resources.) annotationsN)Callable) TYPE_CHECKING AnnotatedAny) AnnotationsIcon)Resource)AnyUrl ConfigDictFieldUrlConstraintsfield_validatormodel_validator)Self) get_context)FastMCPComponent)find_kwarg_by_type get_fn_namecseZdZUdZeddZedddZded<ed d dZ d ed <ed dddZ d ed<dZ ded<dCfdd Z dCfdd Z e         dDdEd+d,Zedd-d.edFd/d0Zed1d.dGd3d4ZdHd6d7Zdd8dId=d>ZdJd?d@ZedJdAdBZZS)Kr zBase class for all resources.T)validate_default.zURI of the resource)default descriptionz6Annotated[AnyUrl, UrlConstraints(host_required=False)]urizName of the resourcestrname text/plainz!MIME type of the resource contentz^[a-zA-Z0-9]+/[a-zA-Z0-9\-+.]+$)rrpattern mime_typeNzfAnnotated[Annotations | None, Field(description="Optional annotations about the resource's behavior")]rreturnNonec4tz t}|WdStyYdSwN)superenabler_queue_resource_list_changed RuntimeErrorselfcontext __class__`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/resources/resource.pyr$4  zResource.enablecr!r")r#disablerr%r&r'r*r,r-r/<r.zResource.disablefnCallable[..., Any] str | AnyUrl str | Nonetitlericonslist[Icon] | Nonetagsset[str] | Noneenabled bool | NoneAnnotations | Nonemetadict[str, Any] | NoneFunctionResourcec Cs tj|||||||||| | d S)N r0rrr4rr5rr7r9rr<)r> from_functionr?r,r,r-r@DszResource.from_functionbefore)modecCs |r|SdS)z&Set default MIME type if not provided.rr,)clsrr,r,r-set_default_mime_type`szResource.set_default_mime_typeafterrcCs*|jr |S|jrt|j|_|Std)z*Set default name from URI if not provided.z#Either name or uri must be provided)rrr ValueErrorr(r,r,r-set_default_namehs zResource.set_default_name str | bytescs td)zRead the resource content. This method is not implemented in the base Resource class and must be implemented by subclasses. z Subclasses must implement read())NotImplementedErrorrGr,r,r-readssz Resource.readinclude_fastmcp_metarM overridesr MCPResourcecKsnt|d|j|d|j|d|j|d|j|d|j|d|j|d|j|d|j |d d S) z'Convert the resource to an MCPResource.rrrmimeTyper4r5r_metarL)rrrrPr4r5rrQ) rOgetrrrrr4r5rget_meta)r(rMrNr,r,r-to_mcp_resource{s        zResource.to_mcp_resourcec Cs.|jjd|jd|jd|jd|jd S)Nz(uri=z, name=z, description=z, tags=))r+__name__rrrr7rGr,r,r-__repr__s.zResource.__repr__cCs|jpt|jS)a* The key of the component. This is used for internal bookkeeping and may reflect e.g. prefixes or other identifiers. You should not depend on keys having a certain value, as the same tool loaded from different hierarchies of servers may have different keys. )_keyrrrGr,r,r-keysz Resource.key)rr NNNNNNNNNr0r1rr2rr3r4r3rr3r5r6rr3r7r8r9r:rr;r<r=rr>)rr3rr)rrrrI)rMr:rNrrrO)rr)rV __module__ __qualname____doc__r model_configr r__annotations__rrrr$r/ staticmethodr@r classmethodrDrrHrKrTrWpropertyrY __classcell__r,r,r*r-r !sL        r c@sFeZdZUdZded<e         ddddZdddZdS)r>aA resource that defers data loading by wrapping a function. The function is only called when the resource is read, allowing for lazy loading of potentially expensive data. This is particularly useful when listing resources, as the function won't be called until the resource is actually accessed. The function can return: - str for text content (default) - bytes for binary content - other types will be converted to JSON r1r0Nrr2rr3r4rr5r6rr7r8r9r:rr;r<r=rc CsXt|tr t|}||||pt|||pt|||pd|p t| dur&| nd| | d S)z*Create a FunctionResource from a function.rNTr?) isinstancerr rinspectgetdocset) rCr0rrr4rr5rr7r9rr<r,r,r-r@s   zFunctionResource.from_functionrIcsddlm}i}t|j|d}|durt||<|jdi|}t|r+|IdH}t|tr7| IdHSt|t t Br@|St j |t dS)z2Read the resource by calling the wrapped function.r)Context) kwarg_typeN)fallbackr,)fastmcp.server.contextrjrr0rrg isawaitablerfr rKbytesr pydantic_coreto_jsondecode)r(rjkwargs context_kwargresultr,r,r-rKs     zFunctionResource.readrZr[r\)rVr]r^r_rarcr@rKr,r,r,r-r>s  r>)!r_ __future__rrgcollections.abcrtypingrrrrp mcp.typesrrr rOpydanticr r r r rrtyping_extensionsrfastmcp.server.dependenciesrfastmcp.utilities.componentsrfastmcp.utilities.typesrrr>r,r,r,r-s"       }