$jQRUddlmZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z ddlZddlmZddl mZmZmZmZmZmZddl mZddlmZmZmZdd lmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/ ddl0m1Z2ddl3m4Z5dZ6n #e7$rdZ6YnwxYwerddl8m9Z9ddl:m;Z;ddlm?Z?m@Z@e)eAZBeegeCfZDdeEd<d:dZFGdd eZGGd!d"e'Z d;dd.ZKd;d?d2ZLd@d4ZMd;dAd5ZN d;dBd7ZOd"d gZPdCd9ZQdS)D) annotationsN)Callable) TYPE_CHECKING AnnotatedAnyClassVar TypeAliasoverload)validate_and_warn_tool_name)CallToolResult ContentBlockIcon TextContentToolAnnotations ToolExecution)Tool) BaseModelFieldmodel_validator)SkipJsonSchema)FastMCPDeprecationWarning) AuthCheck) TaskConfigTaskMeta)FastMCPComponent) get_logger)AudioFileImageNotSetNotSetT) PrefabApp) ComponentTF)Docket) Execution FunctionTool) ArgTransformTransformedToolr ToolResultSerializerTypedatarreturnstrc\tj|tS)N)fallback) pydantic_coreto_jsonr-decode)r+s Z/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/tools/base.pydefault_serializerr4As$   4 4 4 ; ; = ==ceZdZUedZded<eddZded <edd Zded < ddfd ZddZ xZ S) ToolResultz*List of content blocks for the tool result) descriptionlist[ContentBlock]contentNz4Structured content matching the tool's output schema)defaultr8dict[str, Any] | Nonestructured_contentz)Runtime metadata about the tool executionmetalist[ContentBlock] | Any | Nonedict[str, Any] | Any | Nonecp||td||}t|}|trWt|trt |}n2t|t rt t |} tj|}n5#tj $r#}t d|d}~wwxYwt|ts(tdt|jd|dt||| dS) 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.r:r=r>) ValueError_convert_to_content _HAS_PREFAB isinstance _PrefabApp_prefab_to_json_PrefabComponentr0to_jsonable_pythonPydanticSerializationErrorloggererrordicttype__name__super__init__)selfr:r=r>converted_contente __class__s r3rVzToolResult.__init__Ps ?19TUU U _(G0C70S0S0S  ) 0*==)89K)L)L&& 24DEE)8"(:;;;**& %2%E,&&&""!;    dabdd   0$77  V 233<VV@RVVV %:LSW      s B C/C  Cr,Olist[ContentBlock] | tuple[list[ContentBlock], dict[str, Any]] | CallToolResultc|j!t|j|j|jS|j|jS|j|jfS)N)structuredContentr:_meta)r>r r=r:rWs r3 to_mcp_resultzToolResult.to_mcp_result|sV 9 !"&"9 i   " *< |T444r5)NNN)r:r?r=r@r>r<)r,r[) rT __module__ __qualname__rr:__annotations__r=r>rVr` __classcell__rZs@r3r7r7Es"'%@###G16"X111#(%"M###D 48:>&* * * * * * * * X 5 5 5 5 5 5 5 5r5r7ceZdZUdZdZded<ded<dZded <dZd ed <dZd ed <dZ ded<dZ ded<dZ ded<e ddZdZ d[dZeddddddddeddddddd\d5Zd]d9Zd^d;Ze d_d`d>ZedadAZ d_dbdDZdcdGZdddHdddMZedeedeededddN dedWZedfdXZdgfdY ZxZS)hrz 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)moder,c.t|j|S)ztaskrnrmfnCallable[..., Any]rr str | Nonerstr | int | Nonerwr8rzlist[Icon] | Nonerset[str] | NoneToolAnnotations | Nonerlist[str] | Nonedict[str, Any] | NotSetT | NoneToolResultSerializerType | Noner>r<rbool | TaskConfig | None float | None"AuthCheck | list[AuthCheck] | Noner'cLddlm}|j||||||||| | | | | ||S)zCreate a Tool from a function.rr&)rrrrrwr8rzrrrrjrlr>rrnrm)fastmcp.tools.function_toolr' from_function)clsrrrrrwr8rzrrrrjrlr>rrnrmr's r3rzTool.from_functions_* =<<<<<)|)##%'!    r5 argumentsdict[str, Any]r7c$Ktd)a" Run the tool with arguments. This method is not implemented in the base Tool class and must be implemented by subclasses. `run()` can EITHER return a list of ContentBlocks, or a tuple of (list of ContentBlocks, dict of structured output). zSubclasses must implement run())NotImplementedError)rWrs r3runzTool.runs""CDDDr5 raw_valuect|tr|Strtt|trt |t |St|t r,t t|t |St||j}t|trt|S|j st|ttztztzs6t|tt zr)t#d|Drt|S t%j|}n,#t$jt*f$rt|cYSwxYw|j 6t|t,rt||St|S|j d}t||rd |in||rd d d iind S)zConvert a raw result to ToolResult. Handles ToolResult passthrough and converts raw values using the tool's attributes (serializer, output_schema) for proper conversion. fastmcp_app_namerC)rl)r:Nc3@K|]}t|tVdSNrJr .0items r3 z&Tool.convert_result../s,MM4 466MMMMMMr5r:r=zx-fastmcp-wrap-resultrBfastmcp wrap_resultTrF)rJr7rIrK_prefab_to_tool_result_get_fastmcp_app_namerMrHrlbytesrjr rrrlisttupleanyr0rNrOUnicodeDecodeErrorrRr|)rWrr: structuredrs r3convert_resultzTool.convert_resultsC i , ,    )Z00 -%:4%@%@)%566 -I...%:4%@%@ &iDOLLL i ' ' /g... .   % y,"6">"E F F &9dUl33 &MM9MMMMM &g... . /&9)DDJJ8:LM / / /g... . . . /   %*d++ R!'jQQQQg... .(,,-DEE 9DT*55*7BL)mT233    s E&FF task_metaNonec KdSrrWrrs r3_runz Tool._runGs Sr5rmcp.types.CreateTaskResultc KdSrrrs r3rz Tool._runNs &)Sr5TaskMeta | None'ToolResult | mcp.types.CreateTaskResultczKddlm}||d||d{V}|r|S||d{VS)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_taskrg) component task_typerrN)fastmcp.server.tasks.routingrr)rWrrr task_results r3rz Tool._runUs2 GFFFFF11            XXi(((((((((r5docketr$c|jsdS||j|jgdS)z8Register this tool with docket for background execution.N)names)r~rregisterrkey)rWrs r3register_with_docketzTool.register_with_docket{s@..00  F 33333r5)fn_keytask_keyrrkwargsr%c`K|p|j}|r||d<|j|fi||d{VS)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)rWrrrrr lookup_keys r3 add_to_docketzTool.add_to_dockets[$'tx  %$F5M5ZVZ 55f55i@@@@@@@@@r5) rrrwr8rrrjrlr>transform_args transform_fnTool | Callable[..., Any]str | NotSetT | None ToolAnnotations | NotSetT | Nonerdict[str, ArgTransform] | NonerCallable[..., Any] | Noner)c nddlm} ||}| j|| ||| ||||||  S)Nr)r)) rgrrrrwrr8rrrjrlr>)fastmcp.tools.tool_transformr) _ensure_tool from_tool) rrgrrrwr8rrrjrlr>rrr)s r3rzTool.from_toolsg A@@@@@%%((%)##'!    r5ct|tr|Sddlm}ddlm}m}||}t||r|j||S||S)zCCoerce a callable into a Tool, respecting @tool decorator metadata.r)get_fastmcp_meta)r'ToolMeta)metadata)rJrfastmcp.decoratorsrrr'rr)rrgrr'rfmetas r3rzTool._ensure_tools dD ! ! K777777FFFFFFFF  && eX & & D-<-dUCCC C  &&&r5cPtdddzS)Nrg LocalProvider)zfastmcp.component.typezfastmcp.provider.type)rUget_span_attributes)rWrZs r3rzTool.get_span_attributess0ww**,,&,%40 0   r5)r,r)rtrr,ru) rrrrrrrrwrr8rrzrrrrrrrrjrrlrr>r<rrrnrrmrr,r')rrr,r7)rrr,r7r)rrrrr,r7)rrrrr,r)rrrrr,r)rr$r,r) rr$rrrrrrrrr,r%)rgrrrrrwrr8rrrrrrjrrlrr>rrrrrr,r))rgrr,r)r,r)rTrarb__doc__rhrcrjrrkrlrmrnrrsr classmethodr rrrr rrrrrrrdres@r3rrs** &J&&&&                               _'"""#" !!!!F  $( "&#' $.2)-9?6:&*)- $37#& & & & & [& P E E E E7 7 7 7 rX )))X)&*$)$)$)$)$)L4444"# AAAAAA.  &,,2 $8>9?6:069=26     [ B ' ' '[ '          r5rrBrlrc|@ ||S#t$r'}td|dYd}~nd}~wwxYwt|S)Nz!Error serializing tool result: %sT)exc_info) ExceptionrPwarningr4)rBrlrYs r3_serialize_with_fallbackrs :f%% %    NN3           f % %%s  A;Arr ct|tr|St|tr|St|tr|St|t r|St|trtd|St|trp td| dS#t$r>ddl }td|| dcYSwxYwtdt||S)NtextrSrzutf-8rascii)rJr rto_image_contentrto_audio_contentrto_resource_contentr-rrr2rbase64 b64encoder)rrlrs r3 _convert_to_single_content_blockrsa$ %% $'$$&&&$'$$&&&$*'')))$3T2222$Y YFW1E1EFFF F! Y Y Y MMMF1A1A$1G1G1N1Nw1W1WXXX X X X Y F)A$ )S)S T T TTs#C33AD;:D;z[Rendered Prefab UI]app_namerCallable[..., str] | NonecH ddlm}||S#t$rYdSwxYw)zr|rJrRr-)rgr> fastmcp_metars r3rrsh 9D t88I&&L,%%u%% c3   J 4r5clttdtgt||S)z,Convert a PrefabApp to a FastMCP ToolResult.rrrr)r7r_PREFAB_TEXT_FALLBACKrL)rrs r3rr$s> &/DEEEF*3AQRRR   r5r9c,|gSt|ttzst|gSt d|Dr|St d|Drfd|DSt dt|gS)z2Convert a result to a sequence of content objects.Nc3@K|]}t|tVdSrrrs r3rz&_convert_to_content..9s, = =d:dL ) ) = = = = = =r5c3pK|]1}t|ttztztzV2dSr)rJr rrrrs r3rz&_convert_to_content..>s; T TT:dL50584? @ @ T T T T T Tr5c^g|])}t|tst|n|*Sr)rJr r)rrrls r3 z'_convert_to_content..?sM   dL11  ,T: > > >   r5rr)rJrrrallrrr)rBrls `r3rHrH,s ~ fte| - -F0DDEE = =f = = ===  T TV T T TTT           V*B6:*V*V W W W XXr5rrcdddd}||vrFddl}|jjr tjd|dt d dd lm}t||Std td |)z2Deprecated re-exports for backwards compatibility.r'ParsedFunctionrg)r'r rgrNz Importing zX from fastmcp.tools.tool is deprecated. Import from fastmcp.tools.function_tool instead.) stacklevel) function_toolzmodule z has no attribute ) rsettingsdeprecation_warningswarningswarnr fastmcp.toolsr getattrAttributeErrorrT)rrdeprecated_exportsrr s r3 __getattr__rLs'*  !!!   0  MDTDDD)      0/////}d+++ I8IIII J JJr5)r+rr,r-r)rBrrlrr,r-)rrrlrr,r )rrr,r)rrrrr,r)rgrr,r)rrrrr,r7)rBrrlrr,r9)rrr-r,r)R __future__rrcollections.abcrtypingrrrrr r mcp.typesmcpr0mcp.shared.tool_name_validationr r r rrrrrrupydanticrrrpydantic.json_schemarfastmcp.exceptionsr!fastmcp.server.auth.authorizationrfastmcp.server.tasks.configrrfastmcp.utilities.componentsrfastmcp.utilities.loggingrfastmcp.utilities.typesrrrr r! prefab_ui.appr"rKprefab_ui.components.baser#rMrIrrr$docket.executionr%rr'rr(r)rTrPr-r*rcr4r7rrrrrLrrrH__all__rrr5r3r)s@"""""""$$$$$$GGGGGG&%%%%%6666666666//////888888777777<<<<<<<<999999000000555555GGGGGGKKKKKK******888888JJJJJJJJ H  '/ucz&:::::>>>>D5D5D5D5D5D5D5D5NA A A A A A A A J @D & & & & &$37UUUUU</         37YYYYY: < KKKKKKsBBB