j`UdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZejeZdej$defdZdedefd Zdd e ede efd ZGd d ZdZiZe e eeeffed<ej<Zde defdZ ddZ!GddZ"e"Z#defdZ$ddefdZ%y)aVCentral registry for all hermes-agent tools. Each tool file calls ``registry.register()`` at module level to declare its schema, handler, toolset membership, and availability check. ``model_tools.py`` queries the registry instead of maintaining its own parallel data structures. Import chain (circular-import safe): tools/registry.py (no imports from model_tools or tool files) ^ tools/*.py (import from tools.registry at module level) ^ model_tools.py (imports tools.registry + all tool modules) ^ run_agent.py, cli.py, batch_runner.py, etc. N)Path)CallableDictListOptionalSetnodereturnct|tjr$t|jtjsy|jj }t|tj xrP|jdk(xr?t|jtjxr|jjdk(S)zHReturn True when *node* is a ``registry.register(...)`` call expression.Fregisterregistry) isinstanceastExprvalueCallfunc AttributeattrNameid)r rs /home/cube/projects/richard/traning coach/.omo/evidence/nutricoach-v150-combined/st_01a0560c-r63-installed-wheel-first-claim-qa/venv/lib/python3.12/site-packages/tools/registry.py_is_registry_register_callrs dCHH %Z CHH-M ::??D4' ( II # ( tzz388 , ( JJMMZ '  module_pathc |jd}tj|t|}t d|jDS#tt f$rYywxYw)zReturn True when the module contains a top-level ``registry.register(...)`` call. Only inspects module-body statements so that helper modules which happen to call ``registry.register()`` inside a function are not picked up. zutf-8)encoding)filenameFc32K|]}t|ywN)r).0stmts r z*_module_registers_tools..6sFID)$/Is) read_textrparsestrOSError SyntaxErroranybody)rsourcetrees r_module_registers_toolsr-*s_ &&&8yy#k*:; FDIIF FF [ !s2AA"!A" tools_dirc| t|n&ttjj}t |j dDcgc]*}|j dvrt|rd|j,}}g}|D])} tj||j|+|Scc}w#t$r!}tjd||Yd}~Xd}~wwxYw)zLImport built-in self-registering tool modules and return their module names.Nz*.py> __init__.py mcp_tool.py registry.pyztools.z#Could not import tool module %s: %s)r__file__resolveparentsortedglobnamer-stem importlib import_moduleappend Exceptionloggerwarning)r. tools_pathpath module_namesimportedmod_namees rdiscover_builtin_toolsrF9s$-$9itH~?U?U?W?^?^J:??6233D 99I I #D )  3H  O  # #H - OOH %! O O NN@(A N N Os/B4 &B99 C#CC#ceZdZdZdZ ddZy) ToolEntryz&Metadata for a single registered tool. r8toolsetschemahandlercheck_fn requires_envis_async descriptionemojimax_result_size_charsdynamic_schema_overridesNc ||_||_||_||_||_||_||_||_| |_| |_ | |_ yr rI) selfr8rJrKrLrMrNrOrPrQrRrSs r__init__zToolEntry.__init__VsX      (  & %:")A%r)NN)__name__ __module__ __qualname____doc__ __slots__rVrrrHrHMs0IGKArrHg>@_check_fn_cachefnc\tj}t5tj |}||\}}||z t kr |cdddSddd t |}t5||ft|<ddd|S#1swY6xYw#t$rd}Y5wxYw#1swY|SxYw)zIReturn bool(fn()), TTL-cached across calls. Swallows exceptions as False.NF)time monotonic_check_fn_cache_lockr]get_CHECK_FN_TTL_SECONDSboolr=)r^nowcachedtsrs r_check_fn_cachedri~s .. C  $$R(  IBRx//    RT  "El  L     Ls)*BB. B!B  BB!B+cbt5tjdddy#1swYyxYw)zDrop all cached ``check_fn`` results. Call after config changes that affect tool availability (e.g. ``hermes tools enable``).N)rbr]clearr\rrinvalidate_check_fn_cacherls    s%.ceZdZdZdZdeeeee e fffdZ deefdZ dee e ffdZ de de d zdefd Zd e deefd Zdee fd Zde dee fdZde de dd fdZdee e ffdZde dee fdZ d0d e de dede de dedede de deezd zde defdZd e dd fdZd1dee d edeefd!Zd e d"ede fd#Zd2d e d$eezd zdeezfd%Z dee fd&Z!d e deefd'Z"d e dee fd(Z#d3d e d$e de fd)Z$dee e ffd*Z%de defd+Z&dee effd,Z'dee effd-Z(dee effd.Z)d1d efd/Z*y )4 ToolRegistryzISingleton registry that collects tool schemas + handlers from tool files.cni|_i|_i|_tj|_d|_y)Nr)_tools_toolset_checks_toolset_aliases threadingRLock_lock _generationrUs rrVzToolRegistry.__init__s3,. 4602__& !"rr c|j5t|jjt |j fcdddS#1swYyxYw)zBReturn a coherent snapshot of registry entries and toolset checks.N)rulistrpvaluesdictrqrws r_snapshot_statezToolRegistry._snapshot_states8 ZZ **,-tD4H4H/IIZZs 8AAc(|jdS)z4Return a stable snapshot of registered tool entries.rr|rws r_snapshot_entrieszToolRegistry._snapshot_entries##%a((rc(|jdS)z8Return a stable snapshot of toolset availability checks.r~rws r_snapshot_toolset_checksz%ToolRegistry._snapshot_toolset_checksrrrJcheckNct|sy t|S#t$rtjd|YywxYw)zQRun a toolset check, treating missing or failing checks as unavailable/available.Tz,Toolset %s check raised; marking unavailableF)rer=r>debugrUrJrs r_evaluate_toolset_checkz$ToolRegistry._evaluate_toolset_checks: =   LLG Q s 77r8c||j5|jj|cdddS#1swYyxYw)z0Return a registered tool entry by name, or None.N)rurprc)rUr8s r get_entryzToolRegistry.get_entrys# ZZ;;??4(ZZ2;cnt|jDchc]}|jc}Scc}w)z;Return sorted unique toolset names present in the registry.)r6rrJrUentrys rget_registered_toolset_namesz)ToolRegistry.get_registered_toolset_namess.$2H2H2JK2Ju}}2JKLLKs2cHtfd|jDS)z:Return sorted tool names registered under a given toolset.c3TK|]}|jk(r|j!ywr )rJr8)r!rrJs rr#z:ToolRegistry.get_tool_names_for_toolset..s) $<5}}' JJ$r?rv)rUrrJexistings rregister_toolset_aliasz#ToolRegistry.register_toolset_aliassl ZZ,,007HH/J8W,3D ! !% (    ! ZZs AA55A>cp|j5t|jcdddS#1swYyxYw)z=Return a snapshot of ``{alias: canonical_toolset}`` mappings.N)rur{rrrws rget_registered_toolset_aliasesz+ToolRegistry.get_registered_toolset_aliasess! ZZ--.ZZs,5c||j5|jj|cdddS#1swYyxYw)z8Return the canonical toolset name for an alias, or None.N)rurrrc)rUrs rget_toolset_alias_targetz%ToolRegistry.get_toolset_alias_targets' ZZ((,,U3ZZrrKrLrMrNrOrPrQrRrSoverridec |j5|jj|} | r| j|k7r| jj dxr|j d}|r#t j d||| jnQ| r#t jd||| jn,t jd||| j dddyt||||||xsg||xs|jdd| | |  |j|<|r||jvr||j|<|xjd z c_ dddy#1swYyxYw) aRegister a tool. Called at module-import time by each tool file. ``override=True`` is an explicit opt-in for plugins that intend to replace an existing built-in tool implementation (e.g. swap the default browser tool for a headed-Chrome CDP backend). Without it, registrations that would shadow an existing tool from a different toolset are rejected to prevent accidental overwrites. zmcp-z8Tool '%s': MCP toolset '%s' overwriting MCP toolset '%s'zOTool '%s': toolset '%s' overriding existing toolset '%s' (override=True opt-in)zTool registration REJECTED: '%s' (toolset '%s') would shadow existing tool from toolset '%s'. Pass override=True to register() if the replacement is intentional, or deregister the existing tool first.NrPrIr) rurprcrJ startswithr>rinfoerrorrHrqrv)rUr8rJrKrLrMrNrOrPrQrRrSrrboth_mcps rr zToolRegistry.registers[.ZZ{{t,HH,,7$$//73**62LLRgx'7'7KK1gx'7'7LLNgx'7'7 AZB!*!)/R!'H6::mR+H&;)A !DKK G4+?+??08$$W-    ! aZZsCEA/EEc&|j5|jj|d dddytfd|jj D}|sh|j jj d|jjDcic]\}}|j k7r||c}}|_|xjdz c_ dddtjd|ycc}}w#1swY&xYw)aRemove a tool from the registry. Also cleans up the toolset check if no other tools remain in the same toolset. Used by MCP dynamic tool discovery to nuke-and-repave when a server sends ``notifications/tools/list_changed``. Nc3PK|]}|jjk(ywr )rJ)r!rErs rr#z*ToolRegistry.deregister..@s#'4Hq U]]*4Hs#&rzDeregistered tool: %s) rurppopr)rzrqrJrritemsrvr>r)rUr8toolset_still_existsrtargetrs @r deregisterzToolRegistry.deregister3sZZKKOOD$/E}Z $''48KK4F4F4H'$ ($$((=*.)>)>)D)D)F))F v.6M)F)%    ! !"  ,d3 )Zs# DA1D(DDDD tool_namesquietcg}i}|jDcic]}|j|}}t|D]}|j|}|s|jrX|j|vr"t |j||j<||js|st jd|{i|jd|ji}|j2 |j} t| tr|j| |jd|d|Scc}w#t$r!} t jd|| Yd} ~ Bd} ~ wwxYw)a"Return OpenAI-format tool schemas for the requested tool names. Only tools whose ``check_fn()`` returns True (or have no check_fn) are included. ``check_fn()`` results are cached for ~30 s via :func:`_check_fn_cached` to amortize repeat probes (check_terminal_ requirements probes modal/docker, browser checks probe playwright, etc.); TTL chosen so env-var changes (``hermes tools enable foo``) still take effect in near-real-time without forcing a full cache flush on every call. z"Tool %s unavailable (check failed)r8NzCdynamic_schema_overrides for tool %s raised %s; using static schemafunction)typer)rr8r6rcrMrir>rrKrSrr{updater=r?r<) rUrrresult check_resultsrentries_by_namer8schema_with_name overridesexcs rget_definitionszToolRegistry.get_definitionsQsI/1 :>:P:P:RS:R5::u,:RS:&D#''-E~~>>64DU^^4TM%..1$U^^4  %I4PC%,,C C  --9  % > > @I!)T2(// : MM:;KL M;'< ?T0!NN.csD%1D** E3EEargsc |j|}|stjdd|iS |jrddlm}||j |fi|S|j |fi|S#t$rt}tjd||dt|jd|} ddlm }||} n#t$r|} YnwxYwtjd| icYd }~Sd }~wwxYw) zExecute a tool handler by name. * Async handlers are bridged automatically via ``_run_async()``. * All exceptions are caught and returned as ``{"error": "..."}`` for consistent error format. rzUnknown tool: r) _run_asynczTool %s dispatch error: %szTool execution failed: z: )_sanitize_tool_errorN) rjsondumpsrO model_toolsrrLr=r> exceptionrrWr) rUr8rkwargsrrrErawr sanitizeds rdispatchzToolRegistry.dispatchst$::w.(?@A A 4~~2!-%--"?"?@@ 5==00 0 4   94 C,DG,<,<+=RsCC <05    ::w 23 3 4sG*A-A-- C*62C%)B87C%8 CC%CC%C*%C*defaultcp|j|}|r|j |jS||Sddlm}|S)zBReturn per-tool max result size, or *default* (or global default).r)DEFAULT_RESULT_SIZE_CHARS)rrRtools.budget_configr)rUr8rrrs rget_max_result_sizez ToolRegistry.get_max_result_sizes>t$ U00<.. .  NA((rcBtd|jDS)z0Return sorted list of all registered tool names.c34K|]}|jywr )r8)r!rs rr#z2ToolRegistry.get_all_tool_names..sG.FUejj.Fsrrws rget_all_tool_nameszToolRegistry.get_all_tool_namessGd.D.D.FGGGrcD|j|}|r |jSdS)uReturn a tool's raw schema dict, bypassing check_fn filtering. Useful for token estimation and introspection where availability doesn't matter — only the schema content does. N)rrKrUr8rs r get_schemazToolRegistry.get_schemas# t$$u||.$.rcD|j|}|r |jSdS)z.Return the toolset a tool belongs to, or None.N)rrJrs rget_toolset_for_toolz!ToolRegistry.get_toolset_for_tools!t$ %u}}/4/rc\|j|}|r|jr |jS|S)z3Return the emoji for a tool, or *default* if unset.)rrQ)rUr8rrs r get_emojizToolRegistry.get_emojis't$$ B'Brcr|jDcic]}|j|jc}Scc}w)z?Return ``{tool_name: toolset_name}`` for every registered tool.)rr8rJrs rget_tool_to_toolset_mapz$ToolRegistry.get_tool_to_toolset_maps27;7M7M7OP7Oe EMM)7OPPPs4c|j5|jj|}ddd|j|S#1swYxYw)zCheck if a toolset's requirements are met. Returns False (rather than crashing) when the check function raises an unexpected exception (e.g. network error, missing import, bad config). N)rurqrcrrs ris_toolset_availablez!ToolRegistry.is_toolset_availablesA ZZ((,,W5E++GU;;Zs AA c |j\}}t|Dchc]}|jc}}|Dcic]$}||j||j |&c}Scc}wcc}w)z7Return ``{toolset: available_bool}`` for every toolset.)r|r6rJrrc)rUentriestoolset_checksrtoolsetsrJs rcheck_toolset_requirementsz'ToolRegistry.check_toolset_requirementss{"&"6"6"8g>gU5==g>?$ # T11'>;M;Mg;VW W#  ? s A')A,cri}|j\}}|D]}|j}||vr)|j||j|gdgd||<||dj |j |j sj|j D]$}|||dvs||dj |&|S)z'Return toolset metadata for UI display.r) availabletoolsrP requirementsrr)r|rJrrcr<r8rN)rUrrrrrhenvs rget_available_toolsetsz#ToolRegistry.get_available_toolsetss$&"&"6"6"8EB!!%!=!=N..r2" #%$&    RL ! ( ( 4!! --C(2,~">>  ^4;;C@. rcbi}|j\}}|D]}|j}||vr|g|j|dgd||<|j||dvr!||dj |j|j D]$}|||dvs||dj |&|S)zABuild a TOOLSET_REQUIREMENTS-compatible dict for backward compat.N)r8env_varsrM setup_urlrrr)r|rJrcr8r<rN)rUrrrrrhrs rget_toolset_requirementsz%ToolRegistry.get_toolset_requirementss"$"&"6"6"8EB " . 2 22 6!% r zzG!44r 7#**5::6))fRj442Jz*11#6* rc g}g}t}|j\}}|D]}|j}||vr|j||j ||j |r|j |X|j ||j|D cgc]} | j|k(s| j c} d||fScc} w)zDReturn (available_toolsets, unavailable_info) like the old function.)r8rr) setr|rJaddrrcr<rNr8) rUrr unavailableseenrrrrhrEs rcheck_tool_availabilityz$ToolRegistry.check_tool_availability s  u"&"6"6"8EBTz HHRL++B0B0B20FG  $"" % 2 2.5IgbaffgI$+%%Js C ,C )NNFrrNNF)Fr )u⚡)+rWrXrYrZrVtuplerrHrr&rr|rrrerrrrrrrrr{ryintfloatr rrrrrrrrrrrrrrrr\rrrnrnsS "JtIS(]8K'K!LJ )4 ?))$sH}*=)s8d?t)c)hy&9) Md3iM # $s)  "C "# "$ "/S#X/ 4c4hsm4"!48-1G"G"G" G"  G"  G"G"G"G"G" #U{T1G"#+G"G"R4s4t4>> tool_error("file not found") '{"error": "file not found"}' >>> tool_error("bad input", success=False) '{"error": "bad input", "success": false}' rF ensure_ascii)r&rrr)messageextrars r tool_errorr3s2s7| $F  e ::f5 11rc b|tj|dStj|dS)aReturn a JSON result string for tool handlers. Accepts a dict positional arg *or* keyword arguments (not both): >>> tool_result(success=True, count=42) '{"success": true, "count": 42}' >>> tool_result({"key": "value"}) '{"key": "value"}' Fr)rr)datars r tool_resultrAs, zz$U33 ::f5 11rr )r N)&rZrr:rloggingrsr`pathlibrtypingrrrrr getLoggerrWr>ASTrerr-r&rFrHrdr]rr__annotations__Lockrbrirlrnr rrr\rrrs    66   8 $ SWW   G G$ GhtnS (AAX68heTk 2238%y~~'d$ E&E&R >& 2C 2 2 2r