$jXCdZddlmZddlZddlZddlmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZddlZddlmZddlmZmZmZdd lmZdd lmZddlZdd lm Z dd l!m"Z"dd l#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,ddl-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4ddl5m6Z6m7Z7m8Z8e4e9Z:e r ddl;mZ>edede fZ?eGddeZ@eddGddZAGd d!e*ZBedJd$ZCeddddde6dddddddd% dKdBZCe dLdddddde6ddddddddCdMdFZC dLdddddde6ddddddddCdNdIZCdS)Oz'Standalone @tool decorator for FastMCP.) annotationsN)Callable) dataclassfield) TYPE_CHECKING AnnotatedAnyLiteralProtocolTypeVaroverloadruntime_checkable)McpError) ErrorDataIconToolAnnotations)Field)SkipJsonSchema)resolve_task_config)FastMCPDeprecationWarning) AuthCheck)without_injected_parameters) TaskConfig)Tool ToolResultToolResultSerializerType)ParsedFunction_is_object_schema)call_sync_fn_in_threadpoolis_coroutine_function) get_logger)NotSetNotSetTget_cached_typeadapter)Docket) ExecutionF.)boundc&eZdZUdZded<d dZd S) DecoratedToolz,Protocol for functions decorated with @tool.ToolMeta __fastmcp__argsr kwargsreturncdSN)selfr-r.s c/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/tools/function_tool.py__call__zDecoratedTool.__call__@sN)r-r r.r r/r )__name__ __module__ __qualname____doc____annotations__r5r2r6r4r*r*:s066======r6r*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&<dS)'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)r7r8r9r:rrBr;rDrFrGrHrJrLr"rNrrQrRrTrVrXrZr\r^r2r6r4r+r+CsD@@!E&u===D====D $G$$$$E"K""""#E#### D    5;M;;;;*.K...."&D&&&&COOOO%)D))))%)L))))!J!!!! G    /3D3333Gr6r+ceZdZUded<dZded<edddddddddeddddddd7d%Zd8d)Ze d9d,Z d:d0Z ddd1d;d6Z dS)< FunctionToolz"SkipJsonSchema[Callable[..., Any]]fnNz3Annotated[SkipJsonSchema[Any], Field(exclude=True)] return_type)metadatarDrFrGrHrJrLrrVrNrXrQrTrZr\Callable[..., Any]rcToolMeta | NonerDrCrFrErGrHrJrIrLrKrrOrVrUrNrMrXToolResultSerializerType | NonerQrPrTrSrZrYr\r[r/ctd||||||| | || ||f Dp | tup| du}||rtd|t||||||| | | || | ||}|j-t jjrtj dtd|j r-t jjrtj dtdtj ||j }|jp|j}|d krtd |j}|t#d }n,t%|t&rt#j|}n|}|||t%|jt.r|j}n|j}|6t%|t0r!t3|std||did|jd|jd|jp|jd|jt;|jndd|jd|j|jn|jd|j d|j!d|d|j"d|j#p tId|jd|j%d|d|j&d|j'S) 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. c30K|]}|duo|tuVdSr1)r").0xs r4 z-FunctionTool.from_function..sC 1!6/r6Nz|Cannot pass both 'metadata' and individual parameters to from_function(). Use metadata alone or individual parameters alone.rDrFrGrHrJrLrNrrQrTrVrXrZr\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.)rVzz,You must provide a name for lambda functions forbidden)modezROutput schemas must represent object types due to MCP spec limitations. Received: rarbrDrFrGrHrJ parametersrNrrLrXrQ task_configrZr\r2)(anyr" TypeErrorr+rXfastmcpsettingsdeprecation_warningswarningswarnrrVr from_functionrD ValueErrorrTr isinstancer] from_boolvalidate_functionrNr#dictrrarbrFstrrGrHrJ input_schemarrLsetrQrZr\)clsrarcrDrFrGrHrJrLrrVrNrXrQrTrZr\individual_params_provided parsed_fn func_name task_valuersfinal_output_schemas r4r{zFunctionTool.from_function^sB     ("F*# ($4'' #,  $> E   '+')%H"   *w/?/T * Mg*        W%5%J  Me*     #0(BWXXX M3Y^  " "KLL L]  $+666KK  D ) ) %$.z::KK$K%%b)444 h,g 6 6 9"+"9  "*"8   *z:Mt/T/T *$%899  9!499 s   || !-- 0). .6-=-IC()))t  ..  #/!,,& .. !-- .- !,, '#%%  **  $ ! "$$# $%  r6 argumentsdict[str, Any]rc Kt|j}t|}|j t j|j5t |r||d{V}n7t|j|d{V}tj |r|d{V}| |d{V}dddn #1swxYwYn#t$r\td|jd|jdt!t#dd|jd|jddwxYwt |r||d{V}n7t|j|d{V}tj |r|d{V}| |d{V}||S) 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)rrar$rZanyio fail_afterr validate_pythonrinspect isawaitable_materialize_generator TimeoutErrorloggerwarningrDrrconvert_result)r3r wrapper_fn type_adapterresults r4runzFunctionTool.runs099 -j99  < # %dl33GG,Z88 2'3'C'CI'N'N!N!N!N!N!N!N(B(8)((""""""#.v662+1\\\\\\F$(#>#>v#F#FFFFFFFFGGGGGGGGGGGGGGG   ?TY??$,??? # _ _ _PTP\ _ _ _   %Z00 *+;;IFFFFFFFF9 0)  &v..*#)\\\\\\F66v>>>>>>>>F""6***s1CA>C CCCCCA&Err cKtj|rd|2d{VStj|rt|S|S)a Consume generators/async generators into lists. Without this, async generators pass through as objects (repr string), and sync generators get consumed during text serialization but are exhausted by the time structured content is built. c"Kg|3d{V}| 6Sr1r2)riitems r4 z7FunctionTool._materialize_generator..*s.222222224D2222sN)r isasyncgen isgeneratorlist)rs r4rz#FunctionTool._materialize_generator!se  f % % 3226222222222 2  v & & <<  r6docketr%Nonec|jsdS||j|jgdS)u Register this tool with docket for background execution. Registers the raw function so Docket sees and resolves ALL dependencies — both FastMCP's (CurrentContext, Progress) and Docket-native ones (Retry, Timeout, ConcurrencyLimit). N)names)rssupports_tasksregisterrakey)r3rs r4register_with_docketz!FunctionTool.register_with_docket/sB..00  Fz22222r6)fn_keytask_keyrrr.r&cZK|p|j}|r||d<|j|fi|di|d{VS)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() rNr2)radd)r3rrrrr. lookup_keys r4 add_to_docketzFunctionTool.add_to_docket:sa('tx  %$F5M5ZVZ 55f55BB BBBBBBBBBr6)"rardrcrerDrCrFrErGrCrHrCrJrIrLrKrrOrVrUrNrMrXrfrQrPrTrSrZrYr\r[r/r`)rrr/r)rr r/r )rr%r/r) rr%rrrrCrrCr.r r/r&) r7r8r9r;rb classmethodr"r{r staticmethodrrrr2r6r4r`r`Zs****GKKKKKK %)$( "&#' $.2)-9?6:&*)- $37'O O O O O [O b/+/+/+/+b   \  3 3 3 3 "# CCCCCCCCr6r`rar/cdSr1r2)ras r4r?r?Tscr6) rFrGrHrJrLrNrrQrTrVrXrZr\ name_or_fnrrFrErGrCrHrJrIrLrKrNrMr'ToolAnnotations | dict[str, Any] | NonerQrPrTrSrVrUrXrWrZrYr\r[Callable[[F], F]c dSr1r2)rrFrGrHrJrLrNrrQrTrVrXrZr\s r4r?r?Vs "sr6rlrrDcdSr1r2)rrDrFrGrHrJrLrNrrQrTrVrXrZr\s r4r?r?hs $sr6str | Callable[..., Any] | Noner c ttr tdit|trt dd    f d d    f d dfd t j|r ||St|tr|t d |n$||nt dt|dfd }|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-methodsrard tool_namerCr/r`c t| t  }t||S)Nrl)rc)r+rr`r{)rar tool_metarr\rHrVrJrQrNrXrLrTrZrGrFs r4 create_toolztool..create_toolsd#'#$T**%!   ))"y)AAAr6r'c t|   }t|dr|jn|}||_|S)Nrl__func__)r+hasattrrr,)rarrctargetrr\rHrVrJrQrNrXrLrTrZrGrFs r4attach_metadataztool..attach_metadatasj#'#%!    !(J 7 7?R% r6ctjjdkr(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.rn)rvrwdecorator_moderyrzr)rarrrs r4 decoratorztool..decorators_   *h 6 6 MV)      ;r9-- -r9---r6Nz6Cannot specify name both as first argument and keywordzInvalid first argument: c|Sr1r2)rarrs r4wrapperztool..wrappersyY'''r6r2)rardrrCr/r`)rar'rrCr/r'rar'r/r') r}rrrrur isroutinerrB)rrDrFrGrHrJrLrNrrQrTrVrXrZr\rrrrrs ````````````` @@@@r4r?r?}s.+t$$5%44 44 *k**  I   BBBBBBBBBBBBBBBBBB(* . . . . . . .$$ GyT*** J $ $G  TUU U   E4 3C3CEEFFF((((((( Nr6r)rrrFrErGrCrHrCrJrIrLrKrNrMrrrQrPrTrSrVrUrXrWrZrYr\r[r/rr1) rrrDrCrFrErGrCrHrCrJrIrLrKrNrMrrrQrPrTrSrVrUrXrWrZrYr\r[r/r) rrrDrCrFrErGrCrHrCrJrIrLrKrNrMrrrQrPrTrSrVrUrXrWrZrYr\r[r/r )Dr: __future__rrrycollections.abcr dataclassesrrtypingrrr r r r r rrmcp.shared.exceptionsr mcp.typesrrrpydanticrpydantic.json_schemarrvfastmcp.decoratorsrfastmcp.exceptionsr!fastmcp.server.auth.authorizationrfastmcp.server.dependenciesrfastmcp.server.tasks.configrfastmcp.tools.baserrrfastmcp.tools.function_parsingrrfastmcp.utilities.async_utilsrr fastmcp.utilities.loggingr!fastmcp.utilities.typesr"r#r$r7rrr%docket.executionr&r'r*r+r`r?r2r6r4rsx--""""""$$$$$$((((((((                     ******6666666666//////222222888888777777CCCCCC222222 MLLLLLLL100000 H  +****** GCxS)***>>>>>H>>> $%%%&%,wCwCwCwCwC4wCwCwCt   !%"# 5;;?"&%)%)! /3 "  $"# 5;;?"&%)%)! /3! *37b $"# 5;;?"&%)%)! /3!bbbbbbbbr6