PL j-dZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZejeZedhZehdZehd Zejd ejZejd Zd/dZd0dZd1dZd2dZd3dZ d4dZ!d5dZ"d6d#Z#d7d&Z$d8d9d+Z%d:d-Z&gd.Z'dS);uTool-dispatch helpers — parallelism gating, multimodal envelopes, mutation tracking. Pure module-level utilities extracted from ``run_agent.py``: * ``_is_destructive_command`` — terminal-command heuristic used to gate parallel batch dispatch. * ``_should_parallelize_tool_batch`` / ``_extract_parallel_scope_path`` / ``_paths_overlap`` — the rules engine deciding when a multi-tool batch can run concurrently. * ``_is_multimodal_tool_result`` / ``_multimodal_text_summary`` / ``_append_subdir_hint_to_multimodal`` — envelope helpers for the ``{"_multimodal": True, "content": [...], "text_summary": ...}`` dict shape returned by tools like ``computer_use``. * ``_extract_file_mutation_targets`` / ``_extract_error_preview`` — per-turn file-mutation verifier inputs. * ``_trajectory_normalize_msg`` — strip image blobs from a message for trajectory saving. All helpers are stateless. ``run_agent`` re-exports each name so existing ``from run_agent import ...`` imports in tests and other modules keep working unchanged. ) annotationsN)Path)AnyDictListOptional)FILE_MUTATING_TOOL_NAMESclarify> read_file skill_view web_search skills_list web_extract ha_get_state search_filessession_searchvision_analyzeha_list_entitiesha_list_services>patchr write_filez(?:^|\s|&&|\|\||;|`)(?: rm\s|rmdir\s| cp\s|install\s| mv\s| sed\s+-i| truncate\s| dd\s| shred\s| git\s+(?:reset|clean|checkout)\s )z[^>]>[^>]|^>[^>]cmdstrreturnboolc~|sdSt|rdSt|rdSdS)zJHeuristic: does this terminal command look like it modifies/deletes files?FT)_DESTRUCTIVE_PATTERNSsearch_REDIRECT_OVERWRITE)rs ?/home/kuhnn/.hermes/hermes-agent/agent/tool_dispatch_helpers.py_is_destructive_commandr!OsJ u##C((t!!#&&t 5 tool_namecH ddlm}||S#t$rYdSwxYw)zCheck if an MCP tool comes from a server with parallel tool calls enabled. Lazy-imports from ``tools.mcp_tool`` to avoid circular dependencies. Returns False if the MCP module is not available. r)is_mcp_tool_parallel_safeF)tools.mcp_toolr% Exception)r#r%s r _is_mcp_tool_parallel_safer(ZsL <<<<<<((333 uus  !!c t|dkrdSd|D}td|DrdSg}|D]}|jj} t j|jj}n:#t$r-tj d||jjddYdSwxYwt|ts+tj d|t|j dS|tvrIt||dStfd |DrdS||t"vrt%|sdSd S) z?Return True when a tool-call batch is safe to run concurrently.Fc&g|]}|jjS)functionname).0tcs r z2_should_parallelize_tool_batch..ls888r"+"888r"c3(K|] }|tvVdSN)_NEVER_PARALLEL_TOOLS)r/r.s r z1_should_parallelize_tool_batch..ms( @ @T4( ( @ @ @ @ @ @r"u@Could not parse args for %s — defaulting to sequential; raw=%sNu6Non-dict args for %s (%s) — defaulting to sequentialc38K|]}t|VdSr3)_paths_overlap)r/existing scoped_paths r r5z1_should_parallelize_tool_batch..s-XXX>+x88XXXXXXr"T)lenanyr-r.jsonloads argumentsr'loggingdebug isinstancedicttype__name___PATH_SCOPED_TOOLS_extract_parallel_scope_pathappend_PARALLEL_SAFE_TOOLSr() tool_calls tool_namesreserved_paths tool_callr# function_argsr:s @r _should_parallelize_tool_batchrOgs :!u88Z888J @ @Z @ @ @@@u!#N &+   Jy'9'CDDMM    MR",TcT2    555  -..  MH]##,    55 * * *6y-PPK"uuXXXXXXXXX uu  ! !+ . . .  0 0 0-i88 uu 4sA112B('B(rNrCOptional[Path]c|tvrdS|d}t|tr|sdSt |}|r9t tj t|St tj tt j |z S)z8Return the normalized file target for path-scoped tools.Npath) rFgetrBrstripr expanduser is_absoluteosrRabspathcwd)r#rNraw_pathexpandeds r rGrGs***t  ((H h $ $HNN,<,<tH~~((**H4BGOOCMM22333 DHJJ$9 : :;; < < >>r"valuerct|to>|dduo't|dtS)a True if the value is a multimodal tool result envelope. Multimodal handlers (e.g. tools/computer_use) return a dict with `_multimodal=True`, a `content` key holding OpenAI-style content parts, and an optional `text_summary` for string-only fallbacks. _multimodalTcontent)rBrCrSlist)rds r _is_multimodal_tool_resultrisM 5$ 3 IIm $ $ , 3 uyy++T 2 2r"c Jt|r|drt|dSg}|dpgD]f}t|trO|ddkr6|t|ddg|rd|SdSt|tr|S tj|tS#t$rt|cYSwxYw) uExtract a plain text view of a multimodal tool result. Used wherever downstream code needs a string — logging, previews, persistence size heuristics, fall-back content for providers that don't support multipart tool messages. text_summaryrgrDtext z[multimodal tool result])default) rirSrrBrCrHjoinr=dumpsr')rdr_ps r _multimodal_text_summaryrss-"%(( * 99^ $ $ .u^,-- -9%%+ 5 5A!T"" 5quuV}}'>'> Svr!2!233444  $99U## #))% z%---- 5zzs+DD"!D"Dict[str, Any]hintNonect|sdS|dpg}|D][}t|trD|ddkr+t |dd|z|d<n\|dd|d||d<t|dtr|d|z|d<dSdS) zMutate a multimodal tool-result envelope to append a subdir hint. The hint is added to the first text part so the model sees it; image parts are left untouched. `text_summary` is also updated for string-fallback callers. NrgrDrlrmrrDrlrk)rirSrBrCrinsert)rdrur_rrs r !_append_subdir_hint_to_multimodalrzs &e , , IIi &BE !! a   155==F#:#:AEE&"--..5AfI E Q66777 i%))N++S11= %n 5 <n==r"args List[str]c<|tvrgS|dkr)|d}|rt|gngS|dpd}|dkr)|d}|rt|gngS|dkr|dpd}t|tr|sgSg}t jd|tjD]@}|d}|r| |A|SgS) ayReturn the file paths a ``write_file`` or ``patch`` call is targeting. For ``write_file`` and ``patch`` in replace mode this is just ``args["path"]``. For ``patch`` in V4A patch mode we parse the patch content for ``*** Update File:`` / ``*** Add File:`` / ``*** Delete File:`` headers so the verifier can track each file in a multi-file patch separately. rrRmodereplacerrmz/^\*\*\*\s+(?:Update|Add|Delete)\s+File:\s*(.+)$r*) _FILE_MUTATING_TOOLSrSrrBrefinditer MULTILINEgrouprTrH)r#r{rrr~bodypaths_ms r _extract_file_mutation_targetsrs8,,, L  HHV  $Axx"$ 88F   (yD y HHV  $Axx"$ wxx  &B$$$ D I+ >  L    B  !!##A  Q Ir"resultmax_lenintcV|t|nd}t|ts" t|}n#t$rYdSwxYw|}|drk t j|}t|tr0t| dtr|d}n#t$rYnwxYwd | }t||kr|d|dz dz}|S)zFPull a one-line error summary out of a tool result for footer display.Nrm{error r*u…) rsrBrr'rT startswithr=r>rCrSrpsplitr;)rrrlstrippeddatas r _extract_error_previewrs:/5/A #F + + +rD dC  t99DD   22 zz||H3 :h''D$%% %*TXXg5F5F*L*L %G}    D  88DJJLL ! !D 4yy7MgkM"U* Ks": AA5AC CCmsgct|ts|S|d}t|ri|dt |iSt|t rig}|D]\}t|tr0|ddvr|dddG||]i|d|iS|S)aStrip image blobs from a message for trajectory saving. Returns a shallow copy with multimodal tool results replaced by their text_summary, and image parts in content lists replaced by `[screenshot]` placeholders. Keeps the message schema otherwise intact. rgrD>image image_url input_imagerlz [screenshot]rx)rBrCrSrirsrhrH)rrgcleanedrrs r _trajectory_normalize_msgr)s c4  ggi  G!'**ED#Dy":7"C"CDDD'4  + " "A!T"" "quuV}}8]']']GGHHHHq!!!!*#*y'*** Jr")r4rIrFrrr!rOrGr8rirsrzrrr)rrrr)r#rrr)rr)r#rrNrCrrP)r\rr]rrr)rdrrr)rdrrr)rdrtrurrrv)r#rr{rtrr|)r)rrrrrr)rrtrrt)(__doc__ __future__rr=r@rWrpathlibrtypingrrrr agent.tool_result_classificationr r getLoggerrElogger frozensetr4rIrFcompileVERBOSErrr!r(rOrGr8rirsrzrrr__all__r,r"r rs..#"""""  ,,,,,,,,,,,,  8 $ $" 9+..!y " " "  YCCCDD#   J  !bj!455    ++++\===="????    2====*    F2.   r"