KiK UdZddlZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl mZmZejeZdaejZejZdZdZdZd Ze 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+ d4dee%d ee%d!e,d"eee%effd#Z-hd$Z.d%d&hZ/ d5d'e%d(ee%efd)e e%d*e e%d+e ee%d"e%f d,Z0d"ee%fd-Z1d.e%d"e e%fd/Z2d"ee%e)ffd0Z3d"ee%e,ffd1Z4d6d2e,d"e ee%ee)ffd3Z5dS)7a 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)registry)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_loop(/home/ubuntu/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_asyncr2QsV,'))  .!!.!!!!   2 2q 2 A A .T[[d33F===-- . . . . . . . . . . . . . . . .!!)>)@)@@@&(( --d333  I  ' ' - --s %%6B  B$'B$cgd}ddl}|D]K} ||#t$r&}td||Yd}~Dd}~wwxYwdS)zImport all tool modules to trigger their registry.register() calls. Wrapped in a function so import errors in optional tools (e.g., fal_client not installed) don't prevent the rest from loading. )ztools.web_toolsztools.terminal_toolztools.file_toolsztools.vision_toolsztools.mixture_of_agents_toolztools.image_generation_toolztools.skills_toolztools.skill_manager_toolztools.browser_toolztools.cronjob_toolsztools.rl_training_toolztools.tts_toolztools.todo_toolztools.memory_toolztools.session_search_toolztools.clarify_toolztools.code_execution_toolztools.delegate_toolztools.process_registryztools.send_message_toolztools.homeassistant_toolrNz#Could not import tool module %s: %s) importlib import_module Exceptionloggerwarning)_modulesr4mod_namees r_discover_toolsr<s H0OO O  # #H - - - - O O O NN@(A N N N N N N N N OOOs$ AAA)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_press browser_closebrowser_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..4s!JJJaAjM&1JJJr execute_code)SANDBOX_ALLOWED_TOOLSbuild_execute_code_schemar{r|)typer{rKrBrC descriptionzV For simple information retrieval, prefer web_search or web_extract (faster, cheaper).c*g|]}|ddSrzrr}s r z(get_tool_definitions..Ys!HHHA!J-/HHHru🛠️ Final tool selection (z tools): u<🛠️ No tools selected (all filtered out or unavailable)c*g|]}|ddSrzrr}s rrz(get_tool_definitions.._s! O O O1:v!6 O O Or)setr r updateprintjoin_LEGACY_TOOLSET_MAPtoolsetsrvdifference_updaterget_definitionstools.code_execution_toolrr enumerategetreplacelenrA)rqrrrstools_to_include toolset_nameresolved legacy_toolsrvts_namefiltered_toolsavailable_tool_namesrrsandbox_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_searchrard function_name function_argstask_id user_task enabled_toolsc|tvr% ddlm}||pdn#t$rYnwxYw |tvrt jd|diS ddlm}|d|||pd n#t$rYnwxYw|d kr$||nt}tj |||| }ntj |||| } ddlm}|d||||pdn#t$rYnwxYw|S#t$rQ} d|dt| } t| t jd| idcYd } ~ 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. r)notify_other_tool_calldefaulterrorz" must be handled by the agent loop) invoke_hook pre_tool_callr) tool_nameargsrrN)rr)rrpost_tool_call)rrr'rzError executing z: F) ensure_ascii)_READ_SEARCH_TOOLStools.file_toolsrr6_AGENT_LOOP_TOOLSjsondumpshermes_cli.pluginsrrArdispatchstrr7r) rrrrrrrrr'r; error_msgs rhandle_function_callrps.2...  ? ? ? ? ? ? " "7#7i 8 8 8 8    D %D - - -:w=(\(\(\]^^ ^  6 6 6 6 6 6 K=}^e^kik l l l l l    D  N * *0=/HmmNgO&}-FF &}#F   6 6 6 6 6 6 K(M ^dnun{y{ | | | | |    D  DDD@}@@A@@  Yz7I.UCCCCCCCCCDss ,,!C)A+*C)+ A85C)7A88AC)>CC) C$!C)#C$$C)) E3AD?9E?Ec(tjS)z!Return all registered tool names.)rget_all_tool_namesrrrrrs  & ( ((rrc*tj|S)z%Return the toolset a tool belongs to.)rget_toolset_for_tool)rs rrrs  ( 3 33rc(tjS)z0Return toolset availability info for UI display.)rget_available_toolsetsrrrrrs  * , ,,rc(tjS)z>Return {toolset: available_bool} for every registered toolset.)rcheck_toolset_requirementsrrrrrs  . 0 00rrxc,tj|S)z.Return (available_toolsets, unavailable_info).rw)rcheck_tool_availabilityrws rrrs  +% 8 8 88r)NNF)NNN)F)6__doc__rrloggingr(typingrrrrrtools.registryrrr r getLogger__name__r7r Lockr localrrrr2r<tools.mcp_toolr=r6r;debugrr>get_tool_to_toolset_mapr?r__annotations__get_toolset_requirementsr@dictrArboolrrrrrrrrrrrrrs* 33333333333333######66666666  8 $ $  ).""&y((   ,,.,.,.f#O#O#OL5111111555 LL0!4444444453333333333 LL.222222223'Gh&F&H&HT#s(^HHH(I(I(K(Kd39oKKK(*49))) .!l%&%&$%AAA  [GFF"#->#'#'ww3iwCyww $sCx. wwwwDJII!>2 "#)- EDEDEDS>EDc]ED} ED DI& ED  EDEDEDEDX)DI)))) 4C4HSM4444 -S$Y---- 1DdO1111 9949E$s)T$Z:O4P999999s07BB- B((B-1CC'C""C'