Ki<dZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z ddl mZmZddl mZdd lmZmZmZdd lmZdd lmZmZdd lmZdd lmZddlm Z ddl!m"Z"ddZ#ddZ$ddZ%GddeZGddeZ&y)z Resource template functionality.) annotationsN)Callable) AnnotatedAny)parse_qsunquote) AnnotationsIcon)ResourceTemplate)Fieldfield_validator validate_call)Resource) get_contextwithout_injected_parameters TaskConfig)FastMCPComponent)compress_schema)get_cached_typeadapterctjd|}|r<|jdjdDchc]}|j c}St Scc}w)zFExtract query parameter names from RFC 6570 `{?param1,param2}` syntax.z \{\?([^}]+)\},)researchgroupsplitstripset) uri_templatematchps f/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/resources/template.pyextract_query_paramsr$sL II& 5E #(;;q>#7#7#<=a == 5L>sAcrtjdd|}tjd|}d}|D]k}|jdr@|j dr/|dd}|j dr|d d}|d |d z }J|d |d z }T|tj |z }mtj d |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>[^/]+)^$)rsubr startswithendswithescapecompile)templatetemplate_without_querypartspatternpartnames r# build_regexr8#s VVNBA HH^%;  - % 5D|#+D1!4t  5 MCsB ceZdZUdZedZded<eddZded <ed Zd ed <ed dZ ded<ddZ d fd Z d fd Z e d! d"dZed ded#dZd$dZd%dZd&dZd d d'dZed(dZeddZxZS))r .A template for dynamically creating resources.zSubclasses must implement read() or override create_resource()NotImplementedError)rZ argumentss r#readzResourceTemplate.reads! L   c Ktdw)zCreate a resource from the template with the given parameters. The base implementation does not support background tasks. Use FunctionResourceTemplate for task support. z[Subclasses must implement create_resource(). Use FunctionResourceTemplate for task support.r|)rZr?rFs r#create_resourcez ResourceTemplate.create_resources " =  rinclude_fastmcp_metac t|jd|j|jd|j|jd|j|jd|j |jd|j |jd|j|jd|j|jd|j|  S) z8Convert the resource template to an SDKResourceTemplate.r7 uriTemplaterLmimeTyperjrkr_metar)r7rrLrrjrkrr) SDKResourceTemplategetr7r rLrPrjrkrget_meta)rZr overridess r#to_mcp_templatez ResourceTemplate.to_mcp_templates#vtyy1! mT5F5FG! mT5E5EF]]:t~~>--4--4! mT5E5EF--z!FunctionResourceTemplate.s jk&Br9)default_factoryz_Annotated[TaskConfig, Field(description='Background task execution configuration (SEP-1686).')] task_configc Kdfd }tj||jjjj j jSw)z>Create a resource from the template with the given parameters.cHKjd{}|S7w)N)r~)r)resultrFrZs r#resource_read_fnzBFunctionResourceTemplate.create_resource..resource_read_fns$99v966FM7s " ")rir?r7rLrPrXrlrn)rr)rrpr7rLrPrXrlr)rZr?rFrs` ` r#rz(FunctionResourceTemplate.create_resourcesU  %%((nnLL!!  sA!A%cK|j}tj|j}t |j D]\}}||j vst|ts&|j |}|j}|tjjus|turf |turt|||<n4|turt|||<n|tur|jdv||<|jdi|}tj$|r |d{}|S#t t"f$rYwxYw7w)r{)true1yesNru)copyinspect signaturerilistr>rR isinstancerM annotation Parameteremptyintfloatboollower ValueErrorAttributeError isawaitable) rZr~kwargssig param_name param_valueparamrrs r#rzFunctionResourceTemplate.reads5!('+FLLN';  #J S^^+ ;0Lz2"-- !2!2!8!88J#->-@DX-Xz* ("6"   v &!\F #N3"sDAE  E 1AE 2A D2=.E +E,E 2EE EE Nc B|xs%t|ddxs|jj} | dk(r tdt j |}|j jD]4}|jtjjk(s+tdttjd|}t|}||z}|s tdt|}t j |}t|j j!}|Dchc]k}|j |j"tjj$ur6|j |jtjj&k7r|m}}|Dchc]k}|j |j"tjj$ur6|j |jtjj&k7r|m}}|r||z }|rtd|d |j)|std |d |t+d |j jDs"|j)|std |d||xst j,|}| t/d}n(t1| t2rt/j4| }n| }|j7|| t j8|s |j:}t1|t<r |j>}t|}tA|}|jC}tE|d}tG|}||| ||||xsd|||xs t| | nd| | | Scc}wcc}w)z"Create a template from a function.rWNzz,You must provide a name for lambda functionszz9FunctionResourceTemplate.from_function..ls-  JJ'++77 7 s13zURI parameters z- must be a subset of the function arguments: rrrT) prune_titlesrN) r r7rjrLrkrPrirRrXrlrrmr)$getattrrVrWrrrrRvaluesrrVAR_POSITIONALrrfindallr$rkeysrOrrissubsetanygetdocrrr from_boolvalidate_function isroutine__call__r__func__r json_schemarr)rvrir r7rjrLrkrPrXrlrrmrn func_namerr path_params query_paramsall_uri_params wrapper_fnuser_sig func_paramsr"required_paramsoptional_paramsinvalid_query_paramsr type_adapterrRs r#rpz&FunctionResourceTemplate.from_function"s$RGB D9RR\\=R=R  "KL L#^^**, EzzW..=== R "**%6 EF +L9 $|3OP P14 $$Z0(--2245 ! ""1%--1B1B1H1HH##A&++w/@/@/L/LL   ! ""1%--W5F5F5L5LL##A&++w/@/@/L/LL    #//#A # '(<'==vw '' 4..??lmxlyz   ..0  "**;7 %n%55bcnbop"7W^^B%7  <$+6K d #$..t4KK%%b)4  $B b, 'B04 -j9 !--/ $ZdC : &%#/<!&2G##  A  s#A0NA0Nrrrr) rWrrrrr rrrrrprur9r#roros8 BCD  &>   "&#' $ $#*.&*)-z z z  z  z  z !z z z z (z $z 'z  "z z r9ro)r rMrzset[str])r2rMrz re.Pattern)r?rMr rMrzdict[str, str] | None)'r __future__rrrcollections.abcrtypingrr urllib.parserr mcp.typesr r r rpydanticr r rfastmcp.resources.resourcerfastmcp.server.dependenciesrrfastmcp.server.tasks.configrfastmcp.utilities.componentsrfastmcp.utilities.json_schemarfastmcp.utilities.typesrr$r8rIrorur9r#rsl&" $!*'= 0P299:&4>H.'H.Vv /v r9