iG*UddlmZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z ddlZddlmZddl mZmZmZmZmZmZddl mZddlmZmZmZdd lmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%dd l&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,m-Z- ddl.m/Z0ddl1m2Z3dZ4erddl6m7Z7ddl8m9Z9ddl:m;Z;ddlZ>e'e?Z@eegeAfZBdeCd<d$dZDGddeZEGdde%Z d% d&dZF d% d'dZGd ZHd(d!ZI d% d)d"ZJddgZKd*d#ZLy#e5$rdZ4YwxYw)+) annotationsN)Callable) TYPE_CHECKING AnnotatedAnyClassVar TypeAliasoverload)validate_and_warn_tool_name)CallToolResult ContentBlockIcon TextContentToolAnnotations ToolExecution)Tool) BaseModelFieldmodel_validator)SkipJsonSchema) AuthCheck) TaskConfigTaskMeta)FastMCPComponent) get_logger)AudioFileImageNotSetNotSetT) PrefabApp) ComponentTF)Docket) Execution FunctionTool) ArgTransformTransformedToolr ToolResultSerializerTypecTtj|tjS)N)fallback) pydantic_coreto_jsonstrdecode)datas b/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/tools/tool.pydefault_serializerr2@s   4 ; ; ==ceZdZUedZded<eddZded <edd Zded < d dfd Z dd Z xZ S) ToolResultz*List of content blocks for the tool result) descriptionlist[ContentBlock]contentNz4Structured content matching the tool's output schema)defaultr6dict[str, Any] | Nonestructured_contentz)Runtime metadata about the tool executionmetac| | td||}t|}|trKt|tr|j }n*t|t rt |j } tj|}t|ts%tdt|jd|dt|A||| y#tj$r}tjd|d}~wwxYw) Nz5Either content or structured_content must be provided)resultview)valuezCould not serialize structured content. If this is unexpected, set your tool's output_schema to None to disable automatic serialization: z/structured_content must be a dict or None. Got z: zA. Tools should wrap non-dict values based on their output_schema.)r8r;r<) ValueError_convert_to_content _HAS_PREFAB isinstance _PrefabAppr-_PrefabComponentr,to_jsonable_pythonPydanticSerializationErrorloggererrordicttype__name__super__init__)selfr8r;r<converted_contente __class__s r1rPzToolResult.__init__Os6 ?19TU U _(G0C70S  )0*=);)C)C)E& 24DE)39K)L)T)T)V& %2%E%E,&"0$7  23<<=R@R?UVVV %:LSW  !;;  `ab`cd  s5CD&C??Dc|j,t|j|j|jS|j |jS|j|jfS)N)structuredContentr8_meta)r<r r;r8rQs r1 to_mcp_resultzToolResult.to_mcp_resultys_ 99 !"&"9"9 ii   " " *<< ||T4444r3)NNN)r8zlist[ContentBlock] | Any | Noner;zdict[str, Any] | Any | Noner<r:)returnzOlist[ContentBlock] | tuple[list[ContentBlock], dict[str, Any]] | CallToolResult) rN __module__ __qualname__rr8__annotations__r;r<rPrY __classcell__rTs@r1r5r5Ds"'@#G 16"X1-#("M#D  48:>&* ( 0( 8( $ ( T 5 X 5r3r5c.eZdZUdZdZded<ded<dZded <dZd ed <dZd ed <dZ ded<dZ ded<dZ ded<e dd&dZ d'dZeddddddddedddddd d(dZd)dZd*dZe d+ d,dZe d-dZ d+ d.dZd/dZddd  d0d!Zedeedeededdd" d1d#Zed2d$Zd3fd% ZxZS)4rz Internal tool registration info.toolz ClassVar[str] KEY_PREFIXzOAnnotated[dict[str, Any], Field(description='JSON schema for tool parameters')] parametersNzRAnnotated[dict[str, Any] | None, Field(description='JSON schema for tool output')] output_schemaz]Annotated[ToolAnnotations | None, Field(description='Additional annotations about the tool')]rz]Annotated[ToolExecution | None, Field(description='Task execution configuration (SEP-1686)')] executionzAnnotated[SkipJsonSchema[ToolResultSerializerType | None], Field(description='Deprecated. Return ToolResult from your tools for full control over serialization.')] serializerzAnnotated[SkipJsonSchema[AuthCheck | list[AuthCheck] | None], Field(description='Authorization checks for this tool', exclude=True)]authzkAnnotated[float | None, Field(description='Execution timeout in seconds. If None, no timeout is applied.')]timeoutafter)modec0t|j|S)zz&Tool.convert_result..sM94 469)r8r8r;zx-fastmcp-wrap-resultr>)rEr5rDrF_prefab_to_tool_resultrGrCrfrdr rrrlisttupleanyr,rHrIrLrt)rQ raw_valuer8 structured wrap_results r1convert_resultzTool.convert_resultsN i ,  )Z0-i88)%56-ji.HII%iDOOL    % y,"6">"E F9dUl3M9MMg. . /&99)DJ    %*d+!'jQQg. .((,,-DE 9D*5  JT  77 /g. . /sEE65E6c KywrrQr task_metas r1_runz Tool._run0s c Kywrrrs r1rz Tool._run7s &)rcKddlm}||d||d{}|r|S|j|d{S7!7w)a-Server entry point that handles task routing. This allows ANY Tool subclass to support background execution by setting task_config.mode to "supported" or "required". The server calls this method instead of run() directly. Args: arguments: Tool arguments task_meta: If provided, execute as background task and return CreateTaskResult. If None (default), execute synchronously and return ToolResult. Returns: ToolResult when task_meta is None. CreateTaskResult when task_meta is provided. Subclasses can override this to customize task routing behavior. For example, FastMCPProviderTool overrides to delegate to child middleware without submitting to Docket. r)check_background_taskra) component task_typerrN)fastmcp.server.tasks.routingrr)rQrrr task_results r1rz Tool._run>sO2 G1    XXi((( )s>:><>>c|jjsy|j|j|jgy)z8Register this tool with docket for background execution.N)names) task_configsupports_tasksregisterrkey)rQdockets r1register_with_docketzTool.register_with_docketds1..0  3r3)fn_keytask_keyc~K|xs |j}|r||d<|j|fi||d{S7w)aeSchedule this tool for background execution via docket. Args: docket: The Docket instance arguments: Tool arguments 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() rN)radd)rQrrrrkwargs lookup_keys r1 add_to_docketzTool.add_to_docketjsE$'txx $F5M5ZVZZ 5f5i@@@@s 4=;=) rlror6ryrrdrfr<transform_args transform_fnc jddlm} |j|}| j|| ||| ||||||  S)Nr)r() rarrlrorr6ryrrdrfr<)fastmcp.tools.tool_transformr( _ensure_tool from_tool) rrarlror6ryrrdrfr<rrr(s r1rzTool.from_toolsN A%(((%)##'!  r3ct|tr|Sddlm}ddlm}m}||}t||r|j||S|j|S)zCCoerce a callable into a Tool, respecting @tool decorator metadata.r)get_fastmcp_meta)r&ToolMeta)metadata)rErfastmcp.decoratorsrr~r&rr)rrarr&rfmetas r1rzTool._ensure_toolsS dD !K7F & eX &-<--dUC C  &&r3c,t|dddzS)Nra LocalProvider)zfastmcp.component.typezfastmcp.provider.type)rOget_span_attributes)rQrTs r1rzTool.get_span_attributess#w*,&,%40   r3)rZr)rvrrZrs) r}zCallable[..., Any]rl str | Nonerxzstr | int | Nonerorr6rrrzlist[Icon] | Noneryset[str] | NonerzToolAnnotations | Nonerzzlist[str] | Nonerddict[str, Any] | NotSetT | NonerfToolResultSerializerType | Noner<r:r{zbool | TaskConfig | Nonerhz float | Nonergz"AuthCheck | list[AuthCheck] | NonerZr&)rdict[str, Any]rZr5)rrrZr5r)rrrNonerZr5)rrrrrZzmcp.types.CreateTaskResult)rrrzTaskMeta | NonerZz'ToolResult | mcp.types.CreateTaskResult)rr#rZr) rr#rrrrrrrrrZr$)raTool | Callable[..., Any]rlrrostr | NotSetT | Noner6rryrrz ToolAnnotations | NotSetT | Nonerdrrfrr<rrzdict[str, ArgTransform] | NonerzCallable[..., Any] | NonerZr()rarrZr)rZr)rNr[r\__doc__rbr]rdrrerfrgrhrrmrw classmethodrrrrr rrrrrrr^r_s@r1rrsU* &J &              '"#     4  $( "&#' $.2)-9?6:&*)- $37#& &  & " &  &  & !& & ,& '& 7& 4& $& '& !& "1#& $ %& & P E, \!   )!)) $ ))&*$)!$)#$) 1 $)L4"# AA"A  A  AA A.  &,,2 $8>9?6:069=26 '   $  *   6 7 4 . 7 0    B ' '  r3rc| ||St|S#t$r,}tjd|dYd}~t|Sd}~wwxYw)Nz!Error serializing tool result: %sT)exc_info) ExceptionrJwarningr2)r>rfrSs r1_serialize_with_fallbackrse f% % f %%  NN3    f %% s A AA cNt|tr|St|tr|jSt|tr|j St|t r|jSt|tr td|Stdt||S)NtextrMr) rEr rto_image_contentrto_audio_contentrto_resource_contentr.rr)rrfs r1 _convert_to_single_content_blockrs$ % $$$&&$$$&&$''))$T22 F)A$ )S TTr3z[Rendered Prefab UI]cXttdtg|jS)z,Convert a PrefabApp to a FastMCP ToolResult.rrr)r5r_PREFAB_TEXT_FALLBACKr-)apps r1rrs& &/DEF;;= r3c0|gSt|ttzs t||gSt d|Dr|St d|Dr,|Dcgc] }t|t s t||n|"c}Stdt||gScc}w)z2Convert a result to a sequence of content objects.c3<K|]}t|tywrrrs r1rz&_convert_to_content..s =fd:dL )frc3fK|])}t|ttztztz+ywr)rEr rrrrs r1rz&_convert_to_content..s' TVT:dL50584? @Vs/1rr) rErrrallrr rr)r>rfrs r1rCrCs ~ fte| -0DEE =f ==  TV TT   dL1 -T: >     V*B6:*V W XX s%Bcdddd}||vrLddl}|jjr tjd|dt d dd lm}t||Std td |)z2Deprecated re-exports for backwards compatibility.r&ParsedFunctionra)r&rrarNz Importing zX from fastmcp.tools.tool is deprecated. Import from fastmcp.tools.function_tool instead.) stacklevel) function_toolzmodule z has no attribute ) fastmcpsettingsdeprecation_warningswarningswarnDeprecationWarning fastmcp.toolsrgetattrAttributeErrorrN)rldeprecated_exportsrrs r1 __getattr__r s'*  !!    0 0 MMTF#CD"   0}d++ 78,.@I JJr3)r0rrZr.r)r>rrfrrZr.)rrrfrrZr )rrrZr5)r>rrfrrZr7)rlr.rZr)M __future__rrcollections.abcrtypingrrrrr r mcp.typesmcpr,mcp.shared.tool_name_validationr r r rrrrrrspydanticrrrpydantic.json_schemar!fastmcp.server.auth.authorizationrfastmcp.server.tasks.configrrfastmcp.utilities.componentsrfastmcp.utilities.loggingrfastmcp.utilities.typesrrrrr prefab_ui.appr!rFprefab_ui.components.baser"rGrD ImportErrorrr#docket.executionr$r~r&rr'r(rNrJr.r)r]r2r5rrrrrC__all__rrr3r1rsd"$G&66/7<905GK*8J H '/ucz&:):>B5B5Jm m b @D &  &< & &$37U U/UU,/37Y Y/YY: < KyKs?DDD