i"ddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZmZddlmZmZddlmZddlmZer dd lmZdd lmZed e ZGd de dZddZddZddZGddeZ y)) annotations)Sequence) TYPE_CHECKING AnnotatedAnyClassVar TypedDictcast)Icon)BeforeValidatorField)SelfTypeVar TaskConfig)FastMCPBaseModel)Docket) ExecutionT)defaultc,eZdZUded<ded<ded<y) FastMCPMetaz list[str]tagsstrversionversionsN)__name__ __module__ __qualname____annotations__l/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/utilities/components.pyrrs O Lr"rF)totalcv|siStt|jdxs|jdxsiS)zExtract FastMCP metadata from a component's meta dict. Handles both the current `fastmcp` namespace and the legacy `_fastmcp` namespace for compatibility with older FastMCP servers. fastmcp_fastmcp)r rget)metas r#get_fastmcp_metadatar*s5   TXXi0NDHHZ4HNB OOr"cT| tSt|tr|St|S)z@Convert a sequence to a set, defaulting to an empty set if None.)set isinstance) maybe_sets r#_convert_set_default_noner/%s)u )S! y>r"cF|yt|}d|vrtd||S)z|Coerce version to string, accepting int or str. Raises ValueError if version contains '@' (used as key delimiter). N@z;Version string cannot contain '@' (used as key delimiter): )r ValueError)vrs r#_coerce_versionr4.s:  y!fG g~I' U   Nr"ceZdZUdZdZded<d-fd ZedZded <ed d Z d ed<ed d Z ded<ed d Z ded<ed d Z ded<ee dZded<ed d Zded<edZd ed!<ed.d"Zed/d#Zd0d$Zd1d%Zd/d&Zd2d'Zd2d(Zd3d)Zd4d*Z d5d+Zd0d,ZxZS)6FastMCPComponentzIBase class for FastMCP tools, prompts, resources, and resource templates.z ClassVar[str] KEY_PREFIXc t|di||js*ddl}|j |j dt dyy)Nrzb does not define KEY_PREFIX. Component keys will not be type-prefixed, which may cause collisions.) stacklevelr!)super__init_subclass__r8warningswarnr UserWarning)clskwargsr> __class__s r#r=z"FastMCPComponent.__init_subclass__BsN !+F+~~  MM<<.!XY  r"zThe name of the component.) descriptionrnameNzpOptional version identifier for this component. Multiple versions of the same component (same name) can coexist.)rrDz7Annotated[str | None, BeforeValidator(_coerce_version)]rz0The title of the component for display purposes. str | Nonetitlez!The description of the component.rDzHOptional list of icons for this component to display in user interfaces.zlist[Icon] | NoneiconszTags for the component.)default_factoryrDz?Annotated[set[str], BeforeValidator(_convert_set_default_none)]rz$Meta information about the componentdict[str, Any] | Noner)ctdS)N forbidden)moderr!r"r#zFastMCPComponent.ms jk&Br")rIz_Annotated[TaskConfig, Field(description='Background task execution configuration (SEP-1686).')] task_configc@|jr|jd|S|S)zConstruct the lookup key for this component type. Args: identifier: The raw identifier (name for tools/prompts, uri for resources) Returns: A prefixed key like "tool:name" or "resource:uri" :)r8)rA identifiers r#make_keyzFastMCPComponent.make_keyos' >>nn%Qzl3 3r"cb|j|j}|d|jxsdS)aThe globally unique lookup key for this component. Format: "{key_prefix}:{identifier}@{version}" or "{key_prefix}:{identifier}@" e.g. "tool:my_tool@v2", "tool:my_tool@", "resource:file://x.txt@" The @ suffix is ALWAYS present to enable unambiguous parsing of keys (URIs may contain @ characters, so we always include the delimiter). Subclasses should override this to use their specific identifier. Base implementation uses name. r1r7)rSrEr)selfbase_keys r#keyzFastMCPComponent.key}s1==+1T\\/R011r"c|jrt|jni}dt|ji}|j|j|d<|j dx}ct |ts td|jDcic]\}}|jdr||}}}tt||z}||d<|Scc}}w)a8Get the meta information about the component. Returns a dict that always includes a `fastmcp` key containing: - `tags`: sorted list of component tags - `version`: component version (only if set) Internal keys (prefixed with `_`) are stripped from the fastmcp namespace. rrr&zmeta['fastmcp'] must be a dict_) r)dictsortedrrr(r- TypeErroritems startswithr r)rUr) fastmcp_meta upstream_metakr3public_upstreams r#get_metazFastMCPComponent.get_metas#'))tDII%+VDII->$? << #&*llL #"XXi0 0M =mT2 @AA"/!4!4!6!6Aall3>O1!6  _|-KLL&Y s C1Cct|t|uryt|t|sy|j|jk(S)NF)typer- model_dump)rUothers r#__eq__zFastMCPComponent.__eq__sA :T%[ (%d, E$4$4$666r"cBd|jg}|jr|jd|j|jd|jd|j d|j g|jjddj|dS) Nzname=zversion=ztitle=z description=ztags=(z, )) rErappendextendrGrDrrCrjoin)rUpartss r#__repr__zFastMCPComponent.__repr__s &' << LL8DLL#34 5 't//23 {#  ..))*!DIIe,<+=Q??r"c4td|jd)z6Removed in 3.0. Use server.enable(keys=[...]) instead.zHComponent.enable() was removed in FastMCP 3.0. Use server.enable(keys=[' ']) instead.NotImplementedErrorrWrUs r#enablezFastMCPComponent.enables$!((,z ?  r"c4td|jd)z7Removed in 3.0. Use server.disable(keys=[...]) instead.zJComponent.disable() was removed in FastMCP 3.0. Use server.disable(keys=['rrrsrus r#disablezFastMCPComponent.disables%!))- , @  r"c"|jS)zCreate a copy of the component.) model_copyrus r#copyzFastMCPComponent.copys  r"cy)zRegister this component with docket for background execution. No-ops if task_config.mode is "forbidden". Subclasses override to register their callable (self.run, self.read, self.render, or self.fn). Nr!)rUdockets r#register_with_docketz%FastMCPComponent.register_with_docketsr"cK|jjs0td|jjd|j dt |jjdw)aSchedule this component for background execution via docket. Subclasses override this to handle their specific calling conventions: - Tool: add_to_docket(docket, arguments: dict, **kwargs) - Resource: add_to_docket(docket, **kwargs) - ResourceTemplate: add_to_docket(docket, params: dict, **kwargs) - Prompt: add_to_docket(docket, arguments: dict | None, **kwargs) The **kwargs are passed through to docket.add() (e.g., key=task_key). z Cannot add z 'z)' to docket: task execution not supportedz# does not implement add_to_docket())rOsupports_tasks RuntimeErrorrCrrErt)rUr}argsrBs r# add_to_docketzFastMCPComponent.add_to_docketss..0dnn556b D/0 "~~&&''J K  sA-A/cd|jiS)z{Return span attributes for telemetry. Subclasses should call super() and merge their specific attributes. zfastmcp.component.key)rWrus r#get_span_attributesz$FastMCPComponent.get_span_attributess (22r")rBrreturnNone)rRrrr)rr)rzdict[str, Any])rgobjectrbool)rr)rr)r}rrr)r}rrrrBrrr)rrr__doc__r8r r=r rErrGrDrHr,rr)rO classmethodrSpropertyrWrcrhrprvrxr{r~rr __classcell__)rCs@r#r6r6=s_S "J " 0D#HMKHG D FE:$7K %^ E MR-MD I#("H#D  BCD    2 287 @  !   %( 47  ,3r"r6N)r)rJrr)r.zset[T] | Sequence[T] | Nonerzset[T])r3zstr | int | NonerrF)! __future__rcollections.abcrtypingrrrrr r mcp.typesr pydanticr r typing_extensionsrrfastmcp.server.tasks.configrfastmcp.utilities.typesrr}rdocket.executionrrrr*r/r4r6r!r"r#rsb"$KK++24* C)5 P u3'u3r"