i%WhdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z mZddlZddlmZmZddlmZer dd lmZdd lmZdd lmZdd lmZmZmZdd lm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddZ2ddZ3ddZ4Gdde-ZGddeZ5y)z Resource template functionality.) annotationsN)Callable) TYPE_CHECKINGAnyClassVaroverload)parse_qsunquote) AnnotationsIcon)SkipJsonSchema)Docket) Execution)ResourceTemplate)Fieldfield_validator validate_call)ResourceResourceResult)resolve_ui_mime_type) AuthCheck)transform_context_annotationswithout_injected_parameters) TaskConfigTaskMeta)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 j/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/resources/template.pyextract_query_paramsr,&sR II& 5E #(;;q>#7#7#<=#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>[^/]+)^$)r"subr% startswithendswithescapecompile)templatetemplate_without_querypartspatternpartnames r+ build_regexr@.s VVNBA HH^%;  - %D|#+D1!4t & MCsB c>eZdZUdZdZded<edZded<ed d Zded <ed Z ded<edd Z ded<edddZ ded<d,dZ e d- d.dZed ded/dZd0dZd1dZd2d Ze d3 d4d!Ze d5d"Z d3 d6d#Zd7d$Z d8d%Zed9d&Zed,d'Zd:d(Zddd) d;d*Zd<fd+ ZxZS)=r.A template for dynamically creating resources.r:z ClassVar[str] KEY_PREFIXzSubclasses must implement read() or override create_resource()NotImplementedError)rd argumentss r+readzResourceTemplate.reads! L   c<t|tr|St|S)anConvert a raw result to ResourceResult. This is used in two contexts: 1. In _read() to convert user function return values to ResourceResult 2. In tasks_result_handler() to convert Docket task results to ResourceResult Handles ResourceResult passthrough and converts raw values using ResourceResult's normalization. ) isinstancer)rd raw_values r+convert_resultzResourceTemplate.convert_results! i 0 i((rAc KywNrtrdrGrN task_metas r+_readzResourceTemplate._read c Kywrrtrs r+rzResourceTemplate._read &)rcKddlm}||d||d{}|r|S|j||d{}|jd{}|j |S7K7/7w)aServer entry point that handles task routing. This allows ANY ResourceTemplate subclass to support background execution by setting task_config.mode to "supported" or "required". The server calls this method instead of create_resource()/read() directly. Args: uri: The concrete URI being read params: Template parameters extracted from the URI task_meta: If provided, execute as a background task and return CreateTaskResult. If None (default), execute synchronously and return ResourceResult. Returns: ResourceResult when task_meta is None. CreateTaskResult when task_meta is provided. Subclasses can override this to customize task routing behavior. For example, FastMCPProviderResourceTemplate overrides to delegate to child middleware without submitting to Docket. rcheck_background_taskr: component task_typer}rN)fastmcp.server.tasks.routingrcreate_resourcer~r)rdrGrNrr task_resultresourceresults r+rzResourceTemplate._readst0 G1jFi    --c6::}}&""6** ;&s1A*A$A*A&A*A(A*&A*(A*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{)rdrGrNs r+rz ResourceTemplate.create_resources " =  rc 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.r? uriTemplaterUmimeTyperjrkr_meta)r?rrUrrjrkrr) SDKResourceTemplategetr?r(rUrYrjrkrget_meta)rd overridess r+to_mcp_templatez ResourceTemplate.to_mcp_templates #vtyy1! mT5F5FG! mT5E5EF]]:t~~>--4--4! mT5E5EF--  rAct||j|j|j|jxsdiS)zJCreates a FastMCP ResourceTemplate from a raw MCP ResourceTemplate object.rW)r(r?rUrYr[)rr?rUr)ru mcp_templates r+from_mcp_templatez"ResourceTemplate.from_mcp_templates= %11""$00"++;|   rAcb|j|j}|d|jxsdS)z1The globally unique lookup key for this template.@r.)make_keyr(ri)rdbase_keys r+keyzResourceTemplate.keys3==!2!231T\\/R011rAc|jjsy|j|j|jgy)z[x01 : ) BF .;? )) .);C) #)) NR$+$+ .$+;J$+ 4$+L      &    22 5"# >>>  >  >> >.  rArceZdZUdZded<e d ddZe ddZ d ddZddZdd Zdd Z ddd  dd Z e d dd Z y)rnrSz"SkipJsonSchema[Callable[..., Any]]rhNc Kywrrtrs r+rzFunctionResourceTemplate._readMrrc Kywrrtrs r+rzFunctionResourceTemplate._readRrrcKddlm}||d||d{}|r|S|j|d{}|j|S737w)aOptimized server entry point that skips ephemeral resource creation. For FunctionResourceTemplate, we can call read() directly instead of creating a temporary resource, which is more efficient. Args: uri: The concrete URI being read params: Template parameters extracted from the URI task_meta: If provided, execute as a background task and return CreateTaskResult. If None (default), execute synchronously and return ResourceResult. Returns: ResourceResult when task_meta is None. CreateTaskResult when task_meta is provided. rrr:rNr})rrr~r)rdrGrNrrrrs r+rzFunctionResourceTemplate._readWs`& G1jFi    yy6y22""6** 3sAA AAAAc Kdfd }tj||jjjj j jSw)z>Create a resource from the template with the given parameters.cHKjd{}|S7w)Nr)r~)rrNrds r+resource_read_fnzBFunctionResourceTemplate.create_resource..resource_read_fnys$99v966FM7s " ")rhrGr?rUrYrcrmr^)rr)rror?rUrYrcrr^)rdrGrNrs` ` r+rz(FunctionResourceTemplate.create_resourcevsU  %%((nn!!  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)rz)true1yesNrt)copyinspect signaturerhlistrFr[rrV annotation Parameteremptyintfloatboollower ValueErrorAttributeError isawaitable) rdr}rsig param_name param_valueparamrrs r+r~zFunctionResourceTemplate.reads0!('+FLLN'; #J S^^+ ;0Lz2"-- !2!2!8!88J#->-@DX-Xz*(<("6"   v &!\F #N3"sDAE  E 1AE 2A D2=.E +E,E 2EE EE c|jjsy|j|j|jgy)zRegister this template with docket for background execution. FunctionResourceTemplate registers the underlying function, which has the user's Depends parameters for docket to resolve. Nr)rrrrhrrs r+rz-FunctionResourceTemplate.register_with_dockets3 ..0 z2rArc~K|xs |j}|r||d<|j|fi|di|d{S7w)aSchedule this template for background execution via docket. FunctionResourceTemplate splats the params dict since .fn expects **kwargs. Args: docket: The Docket instance params: Template parameters fn_key: Function lookup key in Docket registry (defaults to self.key) task_key: Redis storage key for the result **kwargs: Additional kwargs passed to docket.add() rNrtrrs r+rz&FunctionResourceTemplate.add_to_docketsH('txx $F5M5ZVZZ 5f5?????rc|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>}tA|}t|}tC|}|jE}tG|d}tI|}tK||}|||| tM|nd||||xsd||| xs t| | || Scc}wcc}w)z"Create a template from a function.rbNzz,You must provide a name for lambda functionszz9FunctionResourceTemplate.from_function..s- 0 JJ'++77 70s13zURI parameters z- must be a subset of the function arguments: forbiddenrqT) prune_titlesrW)r(r?rirjrUrkrYrhr[rcrrlrr^)'getattrrarbrrrr[valuesrrVAR_POSITIONALr'r"findallr,rkeysrXrrissubsetanygetdocrrr from_boolvalidate_function isroutine__call__r__func__rr json_schemarrrrV)rurhr(r?rirjrUrkrYrcrrlrmr^ func_namerr path_params query_paramsall_uri_params wrapper_fnuser_sig func_paramsr*required_paramsoptional_paramsinvalid_query_paramsr type_adapterr[ resolved_mimes r+roz&FunctionResourceTemplate.from_functions&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, 'B+2 .04 -j9 !--/ $ZdC : &-\9E %$+$7CLT##3|!##  M  s#A0N8A0N=rrrrrrrrrr) rbrrrrrrrr~rrrrortrAr+rnrnHs8** BF .;? )) .);C) #)) NR++ .+;J+ 4+> &>3"# @@@  @  @@ @2  $( "&#' $ $*.&*)-37B B B  B " B  B  B !B B B (B $B 'B 1B  "B B rArn)r(rVrzset[str])r:rVrz re.Pattern)rGrVr(rVrzdict[str, str] | None)6r __future__rrr"collections.abcrtypingrrrr urllib.parser r mcp.typesmcpr r pydantic.json_schemar rrdocket.executionrrrpydanticrrrfastmcp.resources.resourcerrfastmcp.server.appsr!fastmcp.server.auth.authorizationrfastmcp.server.dependenciesrrfastmcp.server.tasks.configrrfastmcp.utilities.componentsrfastmcp.utilities.json_schemarfastmcp.utilities.typesrr,r@rQrnrtrAr+r,s&" $99*'/*= @47=99:&4>^ '^ BF /F rA