iFYUdZddlZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z ddlmZmZejeZdaejZejZdZdZdZe  dd lmZen(#e$r Ze d eYdZ[ndZ[wwxYw dd l!m"Z"e"n(#e$r Ze d eYdZ[ndZ[wwxYwe j#Z$ee%e%fe&d <e j'Z(ee%e)fe&d<ga*ee%e&d<ddgdgdgdgdggdgddggdgddgd Z+ d=dee%dee%d e,d!eee%effd"Z-hd#Z.d$d%hZ/d&e%d'ee%efd!ee%effd(Z0d)e%fd*Z1d>d)e%d+e,fd,Z2d)e%fd-Z3 d?d.e%d/ee%efd0e e%d1e e%d2e e%d3e e%d4e ee%d5e,d!e%fd6Z4d!ee%fd7Z5d&e%d!e e%fd8Z6d!ee%e)ffd9Z7d!ee%e,ffd:Z8d>d;e,d!e ee%ee)ffd<Z9dS)@a Model Tools Module Thin orchestration layer over the tool registry. Each tool file in tools/ self-registers its schema, handler, and metadata via tools.registry.register(). This module triggers discovery (by importing all tool modules), then provides the public API that run_agent.py, cli.py, batch_runner.py, and the RL environments consume. Public API (signatures preserved from the original 2,400-line version): get_tool_definitions(enabled_toolsets, disabled_toolsets, quiet_mode) -> list handle_function_call(function_name, function_args, task_id, user_task) -> str TOOL_TO_TOOLSET_MAP: dict (for batch_runner.py) TOOLSET_REQUIREMENTS: dict (for cli.py, doctor.py) get_all_tool_names() -> list get_toolset_for_tool(name) -> str get_available_toolsets() -> dict check_toolset_requirements() -> dict check_tool_availability(quiet) -> tuple N)DictAnyListOptionalTuple)discover_builtin_toolsregistry)resolve_toolsetvalidate_toolsetct5ttrtjatcdddS#1swxYwYdS)a^Return a long-lived event loop for running async tool handlers. Using a persistent loop (instead of asyncio.run() which creates and *closes* a fresh loop every time) prevents "Event loop is closed" errors that occur when cached httpx/AsyncOpenAI clients attempt to close their transport on a dead loop during garbage collection. N)_tool_loop_lock _tool_loop is_closedasyncionew_event_loop3/home/agentuser/.hermes/hermes-agent/model_tools.py_get_tool_loopr,s   !5!5!7!7  /11Js:AAActtdd}||r3tj}tj||t_|S)uReturn a persistent event loop for the current worker thread. Each worker thread (e.g., delegate_task's ThreadPoolExecutor threads) gets its own long-lived loop stored in thread-local storage. This prevents the "Event loop is closed" errors that occurred when asyncio.run() was used per-call: asyncio.run() creates a loop, runs the coroutine, then *closes* the loop — but cached httpx/AsyncOpenAI clients remain bound to that now-dead loop and raise RuntimeError during garbage collection or subsequent use. By keeping the loop alive for the thread's lifetime, cached clients stay valid and their cleanup runs on a live loop. loopN)getattr_worker_thread_localrrrset_event_loopr)rs r_get_worker_loopr;sV ' 6 6D |t~~''|%''t$$$$(! Krc. tj}n#t$rd}YnwxYw|r|rmddl}|jd5}|tj|}| dcdddS#1swxYwYtj tj ur#t}||St}||S)aRun an async coroutine from a sync context. If the current thread already has a running event loop (e.g., inside the gateway's async stack or Atropos's event loop), we spin up a disposable thread so asyncio.run() can create its own loop without conflicting. For the common CLI path (no running loop), we use a persistent event loop so that cached async clients (httpx / AsyncOpenAI) remain bound to a live loop and don't trigger "Event loop is closed" on GC. When called from a worker thread (parallel tool execution), we use a per-thread persistent loop to avoid both contention with the main thread's shared loop AND the "Event loop is closed" errors caused by asyncio.run()'s create-and-destroy lifecycle. This is the single source of truth for sync->async bridging in tool handlers. The RL paths (agent_loop.py, tool_context.py) also provide outer thread-pool wrapping as defense-in-depth, but each handler is self-protecting via this function. Nr) max_workersi,)timeout)rget_running_loop RuntimeError is_runningconcurrent.futuresfuturesThreadPoolExecutorsubmitrunresult threadingcurrent_thread main_threadrrun_until_completer)coror concurrentpoolfuture worker_loop tool_loops r _run_asyncr3QsV,'))  .!!.!!!!   2 2q 2 A A .T[[d33F===-- . . . . . . . . . . . . . . . .!!)>)@)@@@&(( --d333  I  ' ' - --s %%6B  B$'B$)discover_mcp_toolszMCP tool discovery failed: %s)discover_pluginszPlugin discovery failed: %sTOOL_TO_TOOLSET_MAPTOOLSET_REQUIREMENTS_last_resolved_tool_names web_search web_extractterminalvision_analyzemixture_of_agentsimage_generate) skills_list skill_view skill_manage) browser_navigatebrowser_snapshot browser_click browser_typebrowser_scroll browser_back browser_pressbrowser_get_imagesbrowser_visionbrowser_consolecronjob) rl_list_environmentsrl_select_environmentrl_get_current_configrl_edit_configrl_start_trainingrl_check_statusrl_stop_trainingrl_get_results rl_list_runsrl_test_inference) read_file write_filepatch search_filestext_to_speech) web_toolsterminal_tools vision_tools moa_tools image_tools skills_tools browser_tools cronjob_toolsrl_tools file_tools tts_toolsFenabled_toolsetsdisabled_toolsets quiet_modereturnc Nt}||D]}t|rSt|}|||s,t d|d|rd|ndd|t vrMt |}|||s(t d|dd||st d|ϐn?|rdd lm}|D]$}|t|%|D]}t|rSt|}| ||s,t d |d|rd|ndd|t vrMt |}| ||s(t d |dd||st d|n5dd lm}|D]$}|t|%tj || } d | D} d| vrdddl m } m} | | z} | | }t| D]<\}}|diddkr d|d| |<n=d| vrddh| z}|st| D]z\}}|diddkrH|ddd}|dd}di|dd|id| |<n{|sS| rBd| D}t dt%| dd|nt dd| Da| S)a Get tool definitions for model API calls with toolset-based filtering. All tools must be part of a toolset to be accessible. Args: enabled_toolsets: Only include tools from these toolsets. disabled_toolsets: Exclude tools from these toolsets (if enabled_toolsets is None). quiet_mode: Suppress status prints. Returns: Filtered list of OpenAI-format tool definitions. Nu✅ Enabled toolset 'z': z, zno toolsu✅ Enabled legacy toolset 'u⚠️ Unknown toolset: r)get_all_toolsetsu🚫 Disabled toolset 'u🚫 Disabled legacy toolset 'quietc*h|]}|ddSfunctionnamer.0ts r z'get_tool_definitions..s!JJJaAjM&1JJJr execute_code)SANDBOX_ALLOWED_TOOLSbuild_execute_code_schemarqrr)typerqrBr9r: descriptionzV For simple information retrieval, prefer web_search or web_extract (faster, cheaper).c*g|]}|ddSrprrss r z(get_tool_definitions..3s!HHHA!J-/HHHru🛠️ Final tool selection (z tools): u<🛠️ No tools selected (all filtered out or unavailable)c*g|]}|ddSrprrss rr~z(get_tool_definitions..9s! O O O1:v!6 O O Or)setr r updateprintjoin_LEGACY_TOOLSET_MAPtoolsetsrldifference_updater get_definitionstools.code_execution_toolrxry enumerategetreplacelenr8)rgrhritools_to_include toolset_nameresolved legacy_toolsrlts_namefiltered_toolsavailable_tool_namesrxrysandbox_enableddynamic_schemaitdweb_tools_availabledesc tool_namess rget_tool_definitionsrs& EE#, F FL -- F*<88 ''111!vt,ttZbCr499XCVCVCVhrttuuu!4442<@  '' 555!eccc$))T`JaJaccddd!FDlDDEEE F >------'')) > >G  # #OG$<$< = = = =- F FL -- F*<88 228<<<!xvLvv\dEtTYYxEXEXEXjtvvwww!4442<@  22<@@@!ge<eeDIIVbLcLceefff!FDlDDEEE F .-----'')) > >G  # #OG$<$< = = = =-.>jQQQN KJ>JJJ ---^^^^^^^^/2FF22?CC~..  EArvvj"%%))&11^CC-7^$T$Tq!D111+];>RR" ">22  266*b))--f559KKKj>--mR@@D<<pD !+$Kr*~$K}d$K$K))N1%EL R  RHHHHHJ iC4G4GiiRVR[R[\fRgRgii j j j j P Q Q Q!P O O O O r>todomemory delegate_tasksession_searchrWrZ tool_nameargsc|rt|ts|Stj|}|s|S|dpid}|s|S|D]d\}}t|t s||}|s3|d}|sKt||}||ur|||<e|S)aACoerce tool call arguments to match their JSON Schema types. LLMs frequently return numbers as strings (``"42"`` instead of ``42``) and booleans as strings (``"true"`` instead of ``true``). This compares each argument value against the tool's registered JSON Schema and attempts safe coercion when the value is a string but the schema expects a different type. Original values are preserved when coercion fails. Handles ``"type": "integer"``, ``"type": "number"``, ``"type": "boolean"``, and union types (``"type": ["integer", "string"]``). parameters propertiesrz) isinstancedictr get_schemaritemsstr _coerce_value) rrschemarkeyvalue prop_schemaexpectedcoerceds rcoerce_tool_argsrNs z$--   + +F  **\**0b55lCCJ  jjll   U%%%   nnS))   ??6**  x00 %  DI Krrct|tr|D]}t||}||ur|cS|S|dvrt||dkS|dkrt |S|S)zAttempt to coerce a string *value* to *expected_type*. Returns the original string when coercion is not applicable or fails. )integernumberr) integer_onlyboolean)rlistr_coerce_number_coerce_boolean)r expected_typerur(s rrrus -&&  A"5!,,FU"" # ---e=I3MOOOO !!u%%% Lrrc t|}n#ttf$r|cYSwxYw||ks&|tdks|tdkr|S|t|krt|S|r|S|S)zFTry to parse *value* as a number. Returns original string on failure.infz-inf)float ValueError OverflowErrorint)rrfs rrrs %LL  &  AvveEll""a5==&8&8CFF{{1vv  Hs ((cr|}|dkrdS|dkrdS|S)zGTry to parse *value* as a boolean. Returns original string on failure.trueTfalseF)striplower)rlows rrrs< ++--    C f}}t g~~u Lr function_name function_argstask_id tool_call_id session_id user_task enabled_toolsskip_pre_tool_call_hookc `t||} |tvrtjd|diS|sKd} ddlm} | |||pd|pd|pd}n#t $rYnwxYw|tjd|id Sn/ dd lm} | d |||pd|pd|pd n#t $rYnwxYw|tvr% dd l m } | |pdn#t $rYnwxYw|dkr$||nt} tj |||| } ntj ||||} dd lm} | d||| |pd|pd|pdn#t $rYnwxYw| S#t $rQ}d|dt|}t|tjd|id cYd}~Sd}~wwxYw)a Main function call dispatcher that routes calls to the tool registry. Args: function_name: Name of the function to call. function_args: Arguments for the function. task_id: Unique identifier for terminal/browser session isolation. user_task: The user's original task (for browser_snapshot context). enabled_tools: Tool names enabled for this session. When provided, execute_code uses this list to determine which sandbox tools to generate. Falls back to the process-global ``_last_resolved_tool_names`` for backward compat. Returns: Function result as a JSON string. errorz" must be handled by the agent loopNr)get_pre_tool_call_block_messager|)rrrF) ensure_ascii) invoke_hook pre_tool_call)rrrrr)notify_other_tool_calldefaultrw)rr)rrpost_tool_call)rrr(rrrzError executing z: )r_AGENT_LOOP_TOOLSjsondumpshermes_cli.pluginsr Exceptionr_READ_SEARCH_TOOLStools.file_toolsrr8r dispatchrloggerr)rrrrrrrr block_messagerrrrr(e error_msgs rhandle_function_callrs6%]MBBMSD - - -:w=(\(\(\]^^ ^ ' +/M NNNNNN ? ?!!#Mr)/R!-!3 !!!     (z7M":OOOO) :::::: #+&#Mr)/R!-!3       2 2 2 CCCCCC&&w';)<<<<     N * *0=/HmmNgO&}-FF &}#F  6 6 6 6 6 6 K '" 2%+)/R         D  DDD@}@@A@@  Yz7I.UCCCCCCCCCDs!EEAE A# E"A##EEB E B-*E,B-- E:C E CECAE!E?E E  E E  E F-AF("F-(F-c(tjS)z!Return all registered tool names.)r get_all_tool_namesrrrrrs  & ( ((rc*tj|S)z%Return the toolset a tool belongs to.)r get_toolset_for_tool)rs rrr!s  ( 3 33rc(tjS)z0Return toolset availability info for UI display.)r get_available_toolsetsrrrrr&s  * , ,,rc(tjS)z>Return {toolset: available_bool} for every registered toolset.)r check_toolset_requirementsrrrrr+s  . 0 00rrnc,tj|S)z.Return (available_toolsets, unavailable_info).rm)r check_tool_availabilityrms rrr0s  +% 8 8 88r)NNF)F)NNNNNF):__doc__rrloggingr)typingrrrrrtools.registryrr rr r getLogger__name__rrLockr localrrrr3tools.mcp_toolr4rrdebugrr5get_tool_to_toolset_mapr6r__annotations__get_toolset_requirementsr7rr8rboolrrrrrrrrrrrrrrrrrs]* 33333333333333;;;;;;;;66666666  8 $ $  ).""&y((   ,,.,.,.f5111111555 LL0!4444444453333333333 LL.222222223'Gh&F&H&HT#s(^HHH(I(I(K(Kd39oKKK(*49))) .!l%&%&$%AAA  [GFF"#->#'#'ww3iwCyww $sCx. wwwwDJII!>2$$4S>$d38n$$$$N(  # T    $3""& $#)-$)pDpDpDS>pDc]pD3- pD  pD } pDDI&pD"pD pDpDpDpDn)DI)))) 4C4HSM4444 -S$Y---- 1DdO1111 9949E$s)T$Z:O4P999999s06BB, B''B,0CC&C!!C&