iADdZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZddlZddlZddlmZddlmZmZmZmZdd lmZdd lmZddl Z dd l!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5m6Z6m7Z7e3e8Z9e r ddl:m;Z;ddleGddeZ?eddGddZ@Gdd e-ZAed'd!ZBeddddde5dddddddd" d(d#ZBe d)dddddde5dddddddd$ d*d%ZB d)dddddde5dddddddd$ d+d&ZBy),z'Standalone @tool decorator for FastMCP.) annotationsN)Callable) dataclassfield) TYPE_CHECKING AnnotatedAnyLiteralProtocolTypeVaroverloadruntime_checkable)McpError) ErrorDataIconToolAnnotations ToolExecution)Field)SkipJsonSchema)resolve_task_config) AuthCheck)without_injected_parameters) TaskConfig)ParsedFunction_is_object_schema)Tool ToolResultToolResultSerializerType)call_sync_fn_in_threadpool) get_logger)NotSetNotSetTget_cached_typeadapter)Docket) ExecutionF.)boundc$eZdZUdZded<ddZy) DecoratedToolz,Protocol for functions decorated with @tool.ToolMeta __fastmcp__cyN)selfargskwargss k/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/tools/function_tool.py__call__zDecoratedTool.__call__=s#N)r0r r1r returnr )__name__ __module__ __qualname____doc____annotations__r3r.r4r2r)r)7s6=r4r)T)frozenkw_onlyceZdZUdZeddZded<dZded <dZd ed <dZ ded <dZ ded <dZ ded<dZ ded<e Zded<dZded<dZded<dZded<dZded<dZded<dZded<dZd ed!<dZd"ed#<d$Zd%ed&<y)'r*z6Metadata attached to functions by the @tool decorator.toolF)defaultinitzLiteral['tool']typeN str | Nonenamestr | int | Noneversiontitle descriptionlist[Icon] | Noneiconsset[str] | Nonetagsdict[str, Any] | NotSetT | None output_schemaToolAnnotations | Nonerdict[str, Any] | Nonemetar appbool | TaskConfig | Nonetasklist[str] | None exclude_args Any | None serializer float | Nonetimeout"AuthCheck | list[AuthCheck] | NoneauthTboolenabled)r6r7r8r9rrAr:rCrErFrGrIrKr!rMrrPrQrSrUrWrYr[r]r.r4r2r*r*@s@!&u=D/=D* $G $E:"K"#E # D/ 5;M2;*.K'."&D &CO%)D ")%)L")!J ! G\ /3D ,3GTr4r*ceZdZUded<dZded< d fd Zedddddddddedddddd ddZdd Z dd Z ddd  dd Z xZ S) FunctionToolz"SkipJsonSchema[Callable[..., Any]]fnNz3Annotated[SkipJsonSchema[Any], Field(exclude=True)] return_typec t|di|}|jjr)d|vr%t |jj |_|S)zConvert the FastMCP tool to an MCP tool. Extends the base implementation to add task execution mode if enabled. execution) taskSupportr.)super to_mcp_tool task_configsupports_tasksrmoderc)r/ overridesmcp_tool __class__s r2rfzFunctionTool.to_mcp_tool[sP7&33    * * ,I1M!.4;K;K;P;P!QH r4)metadatarCrErFrGrIrKrrUrMrWrPrSrYr[ctd||||||| | || ||f Dxs| tuxs| du}| |r td|t||||||| | | || | ||}|j6t j jrtjdtd|jr6t j jrtjdtdtj||j }|jxs |j}|d k(r td |j }| t#d }n(t%|t&rt#j(|}n|}|j+||t%|j,t.r |j,}n |j,}|)t%|t0rt3|std||did|j4d|j6d|jxs |jd|j8t;|j8ndd|j<d|j>xs |j>d|j@d|jBd|d|jDd|jFxs tId|jd|jJd|d|jLd|jNS) a{Create a FunctionTool from a function. Args: fn: The function to wrap metadata: ToolMeta object with all configuration. If provided, individual parameters must not be passed. name, title, etc.: Individual parameters for backwards compatibility. Cannot be used together with metadata parameter. c38K|]}|duxr|tuywr-)r!).0xs r2 z-FunctionTool.from_function..s, A 1!6/1 sNz|Cannot pass both 'metadata' and individual parameters to from_function(). Use metadata alone or individual parameters alone.rCrErFrGrIrKrMrrPrSrUrWrYr[zThe `serializer` parameter is deprecated. Return ToolResult from your tools for full control over serialization. See https://gofastmcp.com/servers/tools#custom-serialization for migration examples. stacklevelzThe `exclude_args` parameter is deprecated as of FastMCP 2.14. Use dependency injection with `Depends()` instead for better lifecycle management. See https://gofastmcp.com/servers/dependency-injection#using-depends for examples.)rUzz,You must provide a name for lambda functions forbidden)rizROutput schemas must represent object types due to MCP spec limitations. Received: r`rarCrErFrGrI parametersrMrrKrWrPrgrYr[r.)(anyr! TypeErrorr*rWfastmcpsettingsdeprecation_warningswarningswarnDeprecationWarningrUr from_functionrC ValueErrorrSr isinstancer\ from_boolvalidate_functionrMr"dictrr`rarEstrrFrGrI input_schemarrKsetrPrYr[)clsr`rmrCrErFrGrIrKrrUrMrWrPrSrYr[individual_params_provided parsed_fn func_name task_valuergfinal_output_schemas r2rzFunctionTool.from_functionms4B    ("F*# ($4'' #,  $>E   '+')%H"    *w/?/?/T/T MMg#    W%5%5%J%J MMe#  #00(BWBWX MM3Y^^  "KL L]]  $+6K  D )$..z:K$K%%b)4 h,,g 6"+"9"9 "*"8"8   *z:Mt/T$%89 !!4 79  || !-- 0).. .6-=-=-IC(()t  ..  !,,E 0E0E  .. !-- . !,, '#%  **  $ $$ !  r4c Kt|j}t|}|j t j |j5t j|r|j|d{}n=t|j|d{}t j|r |d{}dddnlt j|r|j|d{}n=t|j|d{}t j|r |d{}|j#S777#1swYxYw#t$rctjd|jd|jdtt!dd|jd|jddwxYw777w) zRun the tool with arguments.NzTool 'z' timed out after z`s. Consider using task=True for long-running operations. See https://gofastmcp.com/servers/tasksiz' execution timed out after s)codemessage)rr`r#rYanyio fail_afterinspectiscoroutinefunctionvalidate_pythonr isawaitable TimeoutErrorloggerwarningrCrrconvert_result)r/ arguments wrapper_fn type_adapterresults r2runzFunctionTool.runs09 -j9  << # %%dll322:>'3'C'CI'N!N(B(88)("#..v6+1\F44**:6+;;IFF9 00) &&v.#)\F""6**A"O" &243 TYYK'9$,,H>? #"( 3OPTP\P\~]^ _   G*s-G E*E9E:EEE7E8E<E*G .G/G  GG ,G-G EEEE EA,GG G G c|jjsy|j|j|jgy)zRegister this tool with docket for background execution. FunctionTool registers the underlying function, which has the user's Depends parameters for docket to resolve. N)names)rgrhregisterr`key)r/dockets r2register_with_docketz!FunctionTool.register_with_docket*s3 ..0 z2r4)fn_keytask_keyc~K|xs |j}|r||d<|j|fi|di|d{S7w)aSchedule this tool for background execution via docket. FunctionTool splats the arguments dict since .fn expects **kwargs. 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() rNr.)radd)r/rrrrr1 lookup_keys r2 add_to_docketzFunctionTool.add_to_docket4sH('txx $F5M5ZVZZ 5f5B BBBBs 4=;=)rjr r5zmcp.types.Tool)"r`Callable[..., Any]rmzToolMeta | NonerCrBrErDrFrBrGrBrIrHrKrJrrNrUrTrMrLrWzToolResultSerializerType | NonerPrOrSrRrYrXr[rZr5r_)rdict[str, Any]r5r)rr$r5None) rr$rrrrBrrBr1r r5r%) r6r7r8r:rarf classmethodr!rrrr __classcell__)rls@r2r_r_Ws**GKKDK $ %)$( "&#' $.2)-9?6:&*)- $37'M M " M  M "M M  M !M M ,M 'M 7M 4M $!M "'#M $%M &1'M ( )M M ^++Z3"# CC"C  C  CC Cr4r_cyr-r.)r`s r2r>r>Nsr4) rErFrGrIrKrMrrPrSrUrWrYr[c yr-r.) name_or_fnrErFrGrIrKrMrrPrSrUrWrYr[s r2r>r>Ps"r4rscyr-r.)rrCrErFrGrIrKrMrrPrSrUrWrYr[s r2r>r>bs$r4c ttr tdit|tr t dd     f d d     f d d fd t j |r ||St|tr| t d|n||nt dt|d fd }|S) zStandalone decorator to mark a function as an MCP tool. Returns the original function with metadata attached. Register with a server using mcp.add_tool(). zsTo decorate a classmethod, use @classmethod above @tool. See https://gofastmcp.com/servers/tools#using-with-methodscv t| t  }tj||S)Nrs)rm)r*rr_r)r` tool_name tool_metarr[rGrUrIrPrMrWrKrSrYrFrEs r2 create_toolztool..create_toolsT#'#$T*%! ))"y)AAr4c| t|   }t|dr |jn|}||_|S)Nrs__func__)r*hasattrrr+)r`rrmtargetrr[rGrUrIrPrMrWrKrSrYrFrEs r2attach_metadataztool..attach_metadatasZ#'#%!  !(J 7R% r4ctjjdk(r%tjdt d||S||S)Nobjectzdecorator_mode='object' is deprecated and will be removed in a future version. Decorators now return the original function with metadata attached.ru)r{r|decorator_moder~rr)r`rrrs r2 decoratorztool..decoratorsK    * *h 6 MMV"   r9- -r9--r4z6Cannot specify name both as first argument and keywordzInvalid first argument: c|Sr-r.)r`rrs r2wrapperztool..wrappersY''r4r.)r`rrrBr5r_)r`r&rrBr5r&r`r&r5r&) rrrrrzr isroutinerrA)rrCrErFrGrIrKrMrrPrSrUrWrYr[rrrrrs ````````````` @@@@r2r>r>ws.+t$%4 4 *k* I  BBB(* .$T** J $  TU U   24 3C2DEFF( Nr4r)rrrErDrFrBrGrBrIrHrKrJrMrLr'ToolAnnotations | dict[str, Any] | NonerPrOrSrRrUrTrWrVrYrXr[rZr5Callable[[F], F]r-) rrrCrBrErDrFrBrGrBrIrHrKrJrMrLrrrPrOrSrRrUrTrWrVrYrXr[rZr5r) rzstr | Callable[..., Any] | NonerCrBrErDrFrBrGrBrIrHrKrJrMrLrrrPrOrSrRrUrTrWrVrYrXr[rZr5r )Cr9 __future__rrr~collections.abcr dataclassesrrtypingrrr r r r r rr mcp.typesmcpmcp.shared.exceptionsrrrrrpydanticrpydantic.json_schemarr{fastmcp.decoratorsr!fastmcp.server.auth.authorizationrfastmcp.server.dependenciesrfastmcp.server.tasks.configrfastmcp.tools.function_parsingrrfastmcp.tools.toolrrrfastmcp.utilities.async_utilsrfastmcp.utilities.loggingr fastmcp.utilities.typesr!r"r#r6rrr$docket.executionr%r&r)r*r_r>r.r4r2rs-"$(    *EE/27C2L E0 H * CxS)*>H>> $%&,tC4tCn   !%"# 5;;?"&%)%)! /3       39  ## - ! "  $"# 5;;?"&%)%)! /3!        39  ## -!"# *37b $"# 5;;?"&%)%)! /3!b/b b b  b  b b b3b9b b #b#bbb -!b" #br4