~htdZddlmZddlZddlZddlmZddlmZm Z m Z ddl Z ddl m ZddlmZmZmZmZmZmZddlmZdd lmZdd lmZdd lmZer Gd d eej<Z Gdde Zy)z2Base classes and interfaces for FastMCP resources.) annotationsN)Callable) TYPE_CHECKING AnnotatedAny)Resource)AnyUrl ConfigDictFieldUrlConstraintsfield_validatormodel_validator)Self) get_context)FastMCPComponent)find_kwarg_by_typec>eZdZUdZedZeddZded<ed d Z d ed <ed ddZ d ed<e d ddZ e ddeddZedddZej&d dZd!dZd"dZy)#rzBase 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_typeNc :tj|||||||S)Nfnrrrrtagsenabled)FunctionResource from_functionrs V/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/resources/resource.pyr$zResource.from_function/s1 --#.  before)modec |r|Sy)z&Set default MIME type if not provided.r)clsrs r%set_default_mime_typezResource.set_default_mime_typeCs  r&afterc|jr |S|jrt|j|_|Std)z*Set default name from URI if not provided.z#Either name or uri must be provided)rrr ValueErrorselfs r%set_default_namezResource.set_default_nameKsB 99   XXDHH DI BC Cr&c Kyw)zRead the resource content.Nr*r0s r%readz Resource.readVs  sc ||j|j|j|jd}t di||zS)z'Convert the resource to an MCPResource.)rrrmimeTyper*)rrrr MCPResource)r1 overrideskwargss r%to_mcp_resourcezResource.to_mcp_resource[s>88II++   0Vi/00r&c |jjd|jd|jd|jd|j d S)Nz(uri=z, name=z, description=z, tags=)) __class____name__rrrr!r0s r%__repr__zResource.__repr__esb..))*%|7499-~^b^n^n]qqxy}zCzCyDDEF Fr&NNNNNr Callable[[], Any]rz str | AnyUrlr str | NonerrCrrCr!zset[str] | Noner"z bool | Nonereturnr#)rrCrDr)rDrrDz str | bytes)r8rrDr7)rDr)r> __module__ __qualname____doc__r model_configr r__annotations__rr staticmethodr$r classmethodr,rr2abcabstractmethodr4r:r?r*r&r%rr s('t4LBG!6CC ?b.DED#E72Is  "& $ $#              &[x01 '"#   1Fr&rc`eZdZUdZded<e d ddZd dZy) 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 rBr Nc t|tr t|}||||xs |j|xs |j|xsd|xs t ||SdS)z*Create a FunctionResource from a function.rTr) isinstancerr r>rHset)r+r rrrrr!r"s r%r$zFunctionResource.from_functionxsj c3 +C$#1rzz/<&2G  9=  r&cKddlm}i}t|j|}| t ||<|jdi|}t j |jr |d{}t|tr|jd{St|tr|St|tr|Stj|tdjS7z7Tw)z2Read the resource by calling the wrapped function.r)Context) kwarg_typeN)fallbackindentr*)fastmcp.server.contextrTrr rinspectiscoroutinefunctionrQrr4bytesr pydantic_coreto_jsondecode)r1rTr9 context_kwargresults r%r4zFunctionResource.reads2*477wG  $$/MF= !"6"  & &tww /!\F fh '& &  &M  $M ((#aHOOQ Q"'s%A%C&'C"('C&C$AC&$C&r@rArE)r>rFrGrHrJrLr$r4r*r&r%r#r#is    "& $ $#              .Rr&r#) rH __future__rrMrZcollections.abcrtypingrrrr] mcp.typesrr7pydanticr r r r r rtyping_extensionsrfastmcp.server.dependenciesrfastmcp.utilities.componentsrfastmcp.utilities.typesrABCr#r*r&r%rlsh8" $00-#39FFFFR;Rx;Rr&