o i9@sdZddlmZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZddl mZdd lmZmZmZdd lmZdd lmZdd lmZdd lmZddlmZmZd!ddZ d"ddZ!d#ddZ"GdddeZGdd d eZ#dS)$z Resource template functionality.) annotationsN)Callable)Any)parse_qsunquote) AnnotationsIcon)ResourceTemplate)Fieldfield_validator validate_call)Resource) get_context)FastMCPComponent)compress_schema)find_kwarg_by_typeget_cached_typeadapter uri_templatestrreturnset[str]cCs0td|}|rdd|ddDStS)zFExtract query parameter names from RFC 6570 `{?param1,param2}` syntax.z \{\?([^}]+)\}cSsh|]}|qS)strip.0prr`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/resources/template.py !sz'extract_query_params..,)researchgroupsplitset)rmatchrrrextract_query_paramss r&template re.PatterncCstdd|}td|}d}|D]6}|dr@|dr@|dd}|dr7|d d}|d |d 7}q|d |d 7}q|t|7}qtd |dS)aBuild regex pattern for URI template, handling RFC 6570 syntax. Supports: - `{var}` - simple path parameter - `{var*}` - wildcard path parameter (captures multiple segments) - `{?var1,var2}` - query parameters (ignored in path matching) z \{\?[^}]+\}z (\{[^}]+\}){}r*Nz(?P.+)z>[^/]+)^$)r subr# startswithendswithescapecompile)r'template_without_querypartspatternpartnamerrr build_regex%s    r:uridict[str, str] | Nonec Csz|d\}}}t|}||}|sdSdd|D}|r;t|}t|} |D]} | | vr:| | d|| <q,|S)zMatch URI against template and extract both path and query parameters. Supports RFC 6570 URI templates: - Path params: `{var}`, `{var*}` - Query params: `{?var1,var2}` ?NcSsi|] \}}|t|qSr)r)rkvrrr Osz&match_uri_template..r) partitionr:r% groupdictitemsr&r) r;ruri_path_ query_stringregexr%paramsquery_param_names parsed_queryr9rrrmatch_uri_template?s rKcseZdZUdZeddZded<edddZded <ed dZd ed <ed ddZ ded<dEddZ dFfdd Z dFfdd Z e dGdHd(d)Zed d*d+edId,d-ZdJd/d0ZdKd3d4ZdLd7d8Zd d9dMd>d?ZedNdAdBZedEdCdDZZS)Or .A template for dynamically creating resources.zSubclasses must implement read() or override create_resource())NotImplementedError)rYrzrrrreadszResourceTemplate.readrHr c s4dfdd }tj||jjjjjdS)z>Create a resource from the template with the given parameters.rr{csjdIdH}|S)N)rz)r~)resultrHrYrrresource_read_fnsz:ResourceTemplate.create_resource..resource_read_fn)rgr;r9rMrPrWrnN)rr{)r rtr9rMrPrWrn)rYr;rHrrrrcreate_resourcesz ResourceTemplate.create_resourceinclude_fastmcp_metar overridesrMCPResourceTemplatecKsnt|d|j|d|j|d|j|d|j|d|j|d|j|d|j|d|j |d d S) z8Convert the resource template to an MCPResourceTemplate.r9 uriTemplaterMmimeTyperjrkr_metar)r9rrMrrjrkrr) rgetr9rrMrPrjrkrget_meta)rYrrrrrto_mcp_templates        z ResourceTemplate.to_mcp_template mcp_templatecCs||j|j|j|jp didS)zJCreates a FastMCP ResourceTemplate from a raw MCP ResourceTemplate object.rN)rr9rMrPrR)rr9rMr)rwrrrrfrom_mcp_templatesz"ResourceTemplate.from_mcp_templatecCs |jp|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. )_keyrrXrrrkeys zResourceTemplate.key)rr)rr[ NNNNNNNNNrgrhrrr9rirjrirMrirkrlrPrirWrmrnrorrSrprqrrr)rPrirr)r;rrrqrzrQrr{)r;rrHrQrr )rrorrrr)rrrr )rV __module__ __qualname____doc__r r__annotations__rPrRrrZr_rf staticmethodrtr classmethodrxryr~rrrpropertyr __classcell__rrrdrr ^sP        r c@sFeZdZUdZded<ddd Ze dd ddZd S)!rrrLrhrgrzrQrr{c sddlm}|}t|j|d}|r||vrt||<t|j}t| D]Q\}}||j vrzt |t rz|j |}|j } | tjjusI| t urJq)z$| turVt|||<n| turat|||<n | turm|dv||<Wq)ttfyyYq)wq)|jdi|} t| r| IdH} | S)r|rContext kwarg_type)true1yesNr)fastmcp.server.contextrcopyrrgrinspect signaturelistrCrR isinstancer annotation Parameteremptyintfloatboollower ValueErrorAttributeError isawaitable) rYrzrkwargs context_kwargsig param_name param_valueparamrrrrrr~s:       zFunctionResourceTemplate.readNrrr9rirjrMrkrlrPrWrmrnrorrSrprqc sddlm} |pt|ddp|jj} | dkrtdt|j D] }|j tj j kr1tdq$t || dttd |}t|}||B}|sNtd tj}r\|fd d |D}fd d |D}|r||}|rtd|d||std|d|tddj Ds||std|d||pt|}t|s|j}t|tr|j}t|}|}rΈgnd}t||d}t|}||| ||||pd|||pt| dur| nd| | d S)z"Create a template from a function.rrrVNzz,You must provide a name for lambda functionsz.cs@h|]}j|jtjjurj|jtjjkr|kr|qSrrrrrrrRrzQuery parameters z9 must be optional function parameters with default valueszRequired function arguments z- must be a subset of the URI path parameters css|] }|jtjjkVqdSr])rrrr)rrrrr js  z9FunctionResourceTemplate.from_function..zURI parameters z- must be a subset of the function arguments: ) prune_paramsrNT) rr9rjrMrkrPrgrRrWrnrrp) rrgetattrrUrVrrrrRvaluesrrVAR_POSITIONALrr$r findallr&keysdiscardissubsetanygetdoc isroutine__call__rr__func__r json_schemarr )rwrgrr9rjrMrkrPrWrnrrpr func_namer path_params query_paramsall_uri_params func_paramsrequired_paramsoptional_paramsinvalid_query_params type_adapterrRrrrrrts             z&FunctionResourceTemplate.from_functionrrr)rVrrrrr~rrtrrrrrrs  (rr)rrrr)r'rrr()r;rrrrr<)$r __future__rrr collections.abcrtypingr urllib.parserr mcp.typesrrr rpydanticr r r fastmcp.resources.resourcer fastmcp.server.dependenciesrfastmcp.utilities.componentsrfastmcp.utilities.json_schemarfastmcp.utilities.typesrrr&r:rKrrrrrrs*