$j]dZddlmZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddl mZmZmZddlZddlmZmZddlmZe r dd lmZdd lmZdd lmZdd lmZmZm Z dd l!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3d'dZ4d(dZ5d)dZ6d*d"Z7Gd#d$e-ZGd%d&eZ8dS)+z Resource template functionality.) annotationsN)Callable) TYPE_CHECKINGAnyClassVaroverload)parse_qsquoteunquote) AnnotationsIcon)SkipJsonSchema)Docket) Execution)ResourceTemplate)Fieldfield_validator validate_call)ResourceResourceResult) AuthCheck)transform_context_annotationswithout_injected_parameters) TaskConfigTaskMeta)FastMCPComponent)compress_schema)resolve_ui_mime_type)get_cached_typeadapter uri_templatestrreturnset[str]ctjd|}|r2d|ddDSt S)zFExtract query parameter names from RFC 6570 `{?param1,param2}` syntax. \{\?([^}]+)\}c6h|]}|Sstrip).0ps b/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/resources/template.py z'extract_query_params..+s ===a ===,)researchgroupsplitset)r matchs r,extract_query_paramsr7'sQ I& 5 5E >==5;;q>>#7#7#<#<==== 55Lr.templatere.Pattern[str] | Nonectjdd|}tjd|}d}|D]}|drR|dr=|dd}|dr|d d}|d |d z }_|d |d z }i|tj|z } tjd |dS#tj$rYd SwxYw)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) Returns None if the template produces an invalid regex (e.g. parameter names with hyphens, leading digits, or duplicates from a remote server). z \{\?[^}]+\}z (\{[^}]+\}){}r/*Nz(?P.+)z>[^/]+)^$)r1subr4 startswithendswithescapecompileerror)r8template_without_querypartspatternpartnames r, build_regexrM/s VNBAA H^%; < z&match_uri_template..as$ B B B1a B B Br.r) partitionrMr6 groupdictitemsr7r ) rNr uri_path_ query_stringregexr6paramsquery_param_names parsed_queryrLs r,match_uri_templater`Os!$ c 2 2Ha  % %E }t KK ! !E t B B(9(9(?(?(A(A B B BF50>> -- % 5 5D|##+D1!4t Mr.r]dict[str, Any]c|}D]H\}}t|}|d|d|}|d|d|}Id fd }tjd ||}|S) uExpand a URI template with parameters — inverse of `match_uri_template`. Supports the same RFC 6570 subset: - Path params: `{var}`, `{var*}` - Query params: `{?var1,var2}` r<r=z*}r6 re.Match[str]r"r!cd|ddD}fd|D}|rdd|zSdS)Nc6g|]}|Sr'r()r*ns r, zDexpand_uri_template.._expand_query_block..s >>>q>>>r.r/r0c g|]:}|vt|dtt|;S)=)r r!)r*rLr]s r,rgzDexpand_uri_template.._expand_query_block..sR   v~~T{{ 7 7U3vd|#4#455 7 7~~r.rQ&r;)r3r4join)r6namesrIr]s r,_expand_query_blockz0expand_uri_template.._expand_query_blocksz>>EKKNN$8$8$=$=>>>         )%( (rr.r%)r6rcr"r!)rXr!replacer1rB)r r]resultkeyvalue value_strrms ` r,expand_uri_templaterspsFllnn:: UJJ  S i88 S y99      V$&96 B BF Mr.c eZdZUdZdZded<edZded<ed d Zded <ed Z ded<edd Z ded<edddZ ded<dVdZ e dWdXd-Zed d./edYd0ZdZd2Zd[d5Zd\d9Ze d]d^d=Zed_d@Z d]d`dCZdadEZdbdHZedcdJZedVdKZdddNZdddOdedTZdffdU ZxZS)gr.A template for dynamically creating resources.r8z ClassVar[str] KEY_PREFIXzSubclasses must implement read() or override create_resource()NotImplementedError)rrs r,readzResourceTemplate.reads! L   r. raw_valuerrcNt|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)rrs r,convert_resultzResourceTemplate.convert_results, i 0 0  i(((r.r] task_metaNonec KdSNr'rrNr]rs r,_readzResourceTemplate._read r.rmcp.types.CreateTaskResultc KdSrr'rs r,rzResourceTemplate._read &)Sr.TaskMeta | None+ResourceResult | mcp.types.CreateTaskResultcKddlm}||d||d{V}|r|S|||d{V}|d{V}||S)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_taskr8 component task_typerrN)fastmcp.server.tasks.routingrcreate_resourcerr)rrNr]rr task_resultresourceros r,rzResourceTemplate._reads0 GFFFFF11jFi            --c6::::::::}}&&&&&&""6***r.rc$Ktd)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)rrNr]s r,rz ResourceTemplate.create_resources " =   r. overridesSDKResourceTemplatec t|d|j|d|j|d|j|d|j|d|j|d|j|d|j|d|  S) z8Convert the resource template to an SDKResourceTemplate.rL uriTemplaterwmimeTyperrr_meta)rLrrwrrrrr) rgetrLr rwrzrrrget_meta)rrs r,to_mcp_templatez ResourceTemplate.to_mcp_template's #vty11! mT5FGG! mT5EFF]]:t~>>--44--44! mT5EFF--    r. mcp_templatecN||j|j|j|jpdiS)zJCreates a FastMCP ResourceTemplate from a raw MCP ResourceTemplate object.rx)r rLrwrzr{)rrLrwr)rrs r,from_mcp_templatez"ResourceTemplate.from_mcp_template:s= s%1"$0"+;|     r.cR||j}|d|jpdS)z1The globally unique lookup key for this template.@r;)make_keyr r)rbase_keys r,rpzResourceTemplate.keyGs3==!23311T\/R111r.docketrc|jsdS||j|jgdSz_[x000[10 ::::    )))) BFX)))X) NR$+$+$+$+$+L         &    [  222X2 5555"# >>>>>>.          r.rceZdZUdZded<e d9d:d Zed;dZ d9ddZd?dZ dddd@d#Z e dAdBd8Z dS)Crruz"SkipJsonSchema[Callable[..., Any]]rNrNr!r]rarrr"rc KdSrr'rs r,rzFunctionResourceTemplate._readvrr.rrc KdSrr'rs r,rzFunctionResourceTemplate._read{rr.rrcKddlm}||d||d{V}|r|S||d{V}||S)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. rrr8rNr)rrrr)rrNr]rrrros r,rzFunctionResourceTemplate._reads& GFFFFF11jFi            yy6y22222222""6***r.rc Kdfd }tj||jjjjjjS)z>Create a resource from the template with the given parameters.r"rcDKd{V}|S)Nr)r)ror]rs r,resource_read_fnzBFunctionResourceTemplate.create_resource..resource_read_fns199v966666666FMr.)rrNrLrwrzrrr~)r"r)rrrLrwrzrrr~)rrNr]rs` ` r,rz(FunctionResourceTemplate.create_resourcesm        %(n!    r.rrcK|}tj|j}t |D]\}}||jvrt|tr|j|}|j }|tj j us |turT |turt|||<nb|turt|||<nF|tur=|}|dvrd||<n|dvrd||<nt!d|d|#t t"f$rwxYw|jdi|} tj| r| d{V} | S) r)true1yesT)false0noFzInvalid boolean value for z: Nr')copyinspect signaturerlistrXr{rr! annotation Parameteremptyintfloatboollower ValueErrorAttributeError isawaitable) rrrsig param_name param_valueparamr rros r,rzFunctionResourceTemplate.reads!!(('+FLLNN';';   #J S^++ ;0L0L+z2"- !2!888J#D%%D8rrc|jsdS||j|jgdSr)rrrrrprs r,rz-FunctionResourceTemplate.register_with_dockets@..00  Fz22222r.rrrrrrrcZK|p|j}|r||d<|j|fi|di|d{VS)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() rpNr'rrs r,rz&FunctionResourceTemplate.add_to_docketsa('tx  %$F5M5ZVZ 55f55???????????r.rr rLrrrrwrrrzrrrr|rrrrr~rc|pt|ddp |jj}|dkrtdt j|}|jD]+}|jtj j krtd,ttj d|}t|}||z}|stdt|}t j|tj}fd|D}fd |D}|r||z }|rtd |d ||std |d |t%d|jDs*||std|d|||nt j|}| t)d}n,t+| t,rt)j| }n| }|||t j|s!t+|t4js|j}t+|t:r|j}t?|}t|}tA|}|!}tE|d}tG|}tI||}||||tK|nd||||pd||| p t| | || S)z"Create a template from a function.rNzz,You must provide a name for lambda functionsz.&s^   "1%-1B1HHH#A&+w/@/LLL MLLr.ch|]L}j|jtjjur'j|jtjjkJ|MSr'rr"s r,r-z9FunctionResourceTemplate.from_function..,s^   "1%-W5F5LLL#A&+w/@/LLL MLLr.zQuery parameters z9 must be optional function parameters with default valueszRequired function arguments z- must be a subset of the URI path parameters c3JK|]}|jtjjkVdSr)r r rr!)r*rs r, z9FunctionResourceTemplate.from_function..CsC   J'+7 7      r.zURI parameters z- must be a subset of the function arguments: forbiddenrT) prune_titlesrx)r rLrrrwrrzrr{rrrrr~)&getattrrrrr r r{valuesr rVAR_POSITIONALr5r1findallr7rkeysissubsetanygetdocrrr from_boolvalidate_function isroutine functoolspartial__call__r__func__rr json_schemarrrr!)rrr rLrrrwrrzrrrrr~ func_namerr path_params query_paramsall_uri_params wrapper_fn func_paramsrequired_paramsoptional_paramsinvalid_query_paramsr type_adapterr{ resolved_mimer#s @r,rz&FunctionResourceTemplate.from_functions&RGB D99RR\=R  " "KLL L##^**,,  EzW.=== R> "*%6 EEFF +L99 $|3 QOPP P144 $Z00(-224455                    #//#A #  w(<www '' 44 zzzmxzz    ..00      "**;77  pnppcnpp&1%    &%%%%N3333"# @@@@@@2  $( "&#' $ $*.&*)-37B B B B [B B B r.r)r r!r"r#)r8r!r"r9)rNr!r r!r"rO)r r!r]rar"r!)9r __future__rr4r r1collections.abcrtypingrrrr urllib.parser r r mcp.typesmcpr r pydantic.json_schemarrrdocket.executionrrrpydanticrrrfastmcp.resources.baserr!fastmcp.server.auth.authorizationrfastmcp.server.dependenciesrrfastmcp.server.tasks.configrrfastmcp.utilities.componentsrfastmcp.utilities.json_schemarfastmcp.utilities.mimerfastmcp.utilities.typesrr7rMr`rsrr'r.r,rUs&&"""""" $$$$$$9999999999991111111111''''''''//////+******====== <;;;;;;;777777=<<<<<<<999999999999777777::::::@B@^ ^ ^ ^ ^ '^ ^ ^ BJ J J J J /J J J J J r.