mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-04-25 00:51:20 +00:00
feat(delegate): cross-agent file state coordination for concurrent subagents (#13718)
* feat(models): hide OpenRouter models that don't advertise tool support Port from Kilo-Org/kilocode#9068. hermes-agent is tool-calling-first — every provider path assumes the model can invoke tools. Models whose OpenRouter supported_parameters doesn't include 'tools' (e.g. image-only or completion-only models) cannot be driven by the agent loop and fail at the first tool call. Filter them out of fetch_openrouter_models() so they never appear in the model picker (`hermes model`, setup wizard, /model slash command). Permissive when the field is missing — OpenRouter-compatible gateways (Nous Portal, private mirrors, older snapshots) don't always populate supported_parameters. Treat missing as 'unknown → allow' rather than silently emptying the picker on those gateways. Only hide models whose supported_parameters is an explicit list that omits tools. Tests cover: tools present → kept, tools absent → dropped, field missing → kept, malformed non-list → kept, non-dict item → kept, empty list → dropped. * feat(delegate): cross-agent file state coordination for concurrent subagents Prevents mangled edits when concurrent subagents touch the same file (same process, same filesystem — the mangle scenario from #11215). Three layers, all opt-out via HERMES_DISABLE_FILE_STATE_GUARD=1: 1. FileStateRegistry (tools/file_state.py) — process-wide singleton tracking per-agent read stamps and the last writer globally. check_stale() names the sibling subagent in the warning when a non-owning agent wrote after this agent's last read. 2. Per-path threading.Lock wrapped around the read-modify-write region in write_file_tool and patch_tool. Concurrent siblings on the same path serialize; different paths stay fully parallel. V4A multi-file patches lock in sorted path order (deadlock-free). 3. Delegate-completion reminder in tools/delegate_tool.py: after a subagent returns, writes_since(parent, child_start, parent_reads) appends '[NOTE: subagent modified files the parent previously read — re-read before editing: ...]' to entry.summary when the child touched anything the parent had already seen. Complements (does not replace) the existing path-overlap check in run_agent._should_parallelize_tool_batch — batch check prevents same-file parallel dispatch within one agent's turn (cheap prevention, zero API cost), registry catches cross-subagent and cross-turn staleness at write time (detection). Behavior is warning-only, not hard-failing — matches existing project style. Errors surface naturally: sibling writes often invalidate the old_string in patch operations, which already errors cleanly. Tests: tests/tools/test_file_state_registry.py — 16 tests covering registry state transitions, per-path locking, per-path-not-global locking, writes_since filtering, kill switch, and end-to-end integration through the real read_file/write_file/patch handlers.
This commit is contained in:
parent
35a4b093d8
commit
9c9d9b7ddf
6 changed files with 785 additions and 39 deletions
|
|
@ -27,6 +27,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
|
|||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from toolsets import TOOLSETS
|
||||
from tools import file_state
|
||||
from utils import base_url_hostname
|
||||
|
||||
|
||||
|
|
@ -728,7 +729,22 @@ def _run_single_child(
|
|||
except Exception as e:
|
||||
logger.debug("Progress callback start failed: %s", e)
|
||||
|
||||
result = child.run_conversation(user_message=goal)
|
||||
# File-state coordination: generate a stable child task_id so the
|
||||
# file_state registry can attribute writes back to this subagent,
|
||||
# and snapshot the parent's read set at launch time. After the
|
||||
# child returns we compare to detect "sibling modified files the
|
||||
# parent previously read" and surface it as a reminder on the
|
||||
# returned summary.
|
||||
import uuid as _uuid
|
||||
child_task_id = f"subagent-{task_index}-{_uuid.uuid4().hex[:8]}"
|
||||
parent_task_id = getattr(parent_agent, "_current_task_id", None)
|
||||
wall_start = time.time()
|
||||
parent_reads_snapshot = (
|
||||
list(file_state.known_reads(parent_task_id))
|
||||
if parent_task_id else []
|
||||
)
|
||||
|
||||
result = child.run_conversation(user_message=goal, task_id=child_task_id)
|
||||
|
||||
# Flush any remaining batched progress to gateway
|
||||
if child_progress_cb and hasattr(child_progress_cb, '_flush'):
|
||||
|
|
@ -826,6 +842,36 @@ def _run_single_child(
|
|||
if status == "failed":
|
||||
entry["error"] = result.get("error", "Subagent did not produce a response.")
|
||||
|
||||
# Cross-agent file-state reminder. If this subagent wrote any
|
||||
# files the parent had already read, surface it so the parent
|
||||
# knows to re-read before editing — the scenario that motivated
|
||||
# the registry. We check writes by ANY non-parent task_id (not
|
||||
# just this child's), which also covers transitive writes from
|
||||
# nested orchestrator→worker chains.
|
||||
try:
|
||||
if parent_task_id and parent_reads_snapshot:
|
||||
sibling_writes = file_state.writes_since(
|
||||
parent_task_id, wall_start, parent_reads_snapshot
|
||||
)
|
||||
if sibling_writes:
|
||||
mod_paths = sorted(
|
||||
{p for paths in sibling_writes.values() for p in paths}
|
||||
)
|
||||
if mod_paths:
|
||||
reminder = (
|
||||
"\n\n[NOTE: subagent modified files the parent "
|
||||
"previously read — re-read before editing: "
|
||||
+ ", ".join(mod_paths[:8])
|
||||
+ (f" (+{len(mod_paths) - 8} more)" if len(mod_paths) > 8 else "")
|
||||
+ "]"
|
||||
)
|
||||
if entry.get("summary"):
|
||||
entry["summary"] = entry["summary"] + reminder
|
||||
else:
|
||||
entry["stale_paths"] = mod_paths
|
||||
except Exception:
|
||||
logger.debug("file_state sibling-write check failed", exc_info=True)
|
||||
|
||||
if child_progress_cb:
|
||||
try:
|
||||
child_progress_cb(
|
||||
|
|
|
|||
332
tools/file_state.py
Normal file
332
tools/file_state.py
Normal file
|
|
@ -0,0 +1,332 @@
|
|||
"""Cross-agent file state coordination.
|
||||
|
||||
Prevents mangled edits when concurrent subagents (same process, same
|
||||
filesystem) touch the same file. Complements the single-agent path-overlap
|
||||
check in ``run_agent._should_parallelize_tool_batch`` — this module catches
|
||||
the case where subagent B writes a file that subagent A already read, so
|
||||
A's next write would overwrite B's changes with stale content.
|
||||
|
||||
Design
|
||||
------
|
||||
A process-wide singleton ``FileStateRegistry`` tracks, per resolved path:
|
||||
|
||||
* per-agent read stamps: {task_id: {path: (mtime, read_ts, partial)}}
|
||||
* last writer globally: {path: (task_id, write_ts)}
|
||||
* per-path ``threading.Lock`` for read→modify→write critical sections
|
||||
|
||||
Three public hooks are used by the file tools:
|
||||
|
||||
* ``record_read(task_id, path, *, partial)`` — called by read_file
|
||||
* ``note_write(task_id, path)`` — called after write_file / patch
|
||||
* ``check_stale(task_id, path)`` — called BEFORE write_file / patch
|
||||
|
||||
Plus ``lock_path(path)`` — a context-manager returning a per-path lock to
|
||||
wrap the whole read→modify→write block. And ``writes_since(task_id,
|
||||
since_ts, paths)`` for the subagent-completion reminder in delegate_tool.
|
||||
|
||||
All methods are no-ops when ``HERMES_DISABLE_FILE_STATE_GUARD=1`` is set.
|
||||
|
||||
This module is intentionally separate from ``_read_tracker`` in
|
||||
``file_tools.py`` — that tracker is per-task and handles consecutive-read
|
||||
loop detection, which is a different concern.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import threading
|
||||
import time
|
||||
from collections import defaultdict
|
||||
from contextlib import contextmanager
|
||||
from pathlib import Path
|
||||
from typing import Dict, Iterable, List, Optional, Tuple
|
||||
|
||||
|
||||
# ── Public stamp type ────────────────────────────────────────────────
|
||||
# (mtime, read_ts, partial). partial=True when read_file returned a
|
||||
# windowed view (offset > 1 or limit < total_lines) — writes that happen
|
||||
# after a partial read should still warn so the model re-reads in full.
|
||||
ReadStamp = Tuple[float, float, bool]
|
||||
|
||||
# Number of resolved-path entries retained per agent. Bounded to keep
|
||||
# long sessions from accumulating unbounded state. On overflow we drop
|
||||
# the oldest entries by insertion order.
|
||||
_MAX_PATHS_PER_AGENT = 4096
|
||||
|
||||
# Global last-writer map cap. Same policy.
|
||||
_MAX_GLOBAL_WRITERS = 4096
|
||||
|
||||
|
||||
class FileStateRegistry:
|
||||
"""Process-wide coordinator for cross-agent file edits."""
|
||||
|
||||
def __init__(self) -> None:
|
||||
self._reads: Dict[str, Dict[str, ReadStamp]] = defaultdict(dict)
|
||||
self._last_writer: Dict[str, Tuple[str, float]] = {}
|
||||
self._path_locks: Dict[str, threading.Lock] = {}
|
||||
self._meta_lock = threading.Lock() # guards _path_locks
|
||||
self._state_lock = threading.Lock() # guards _reads + _last_writer
|
||||
|
||||
# ── Path lock management ────────────────────────────────────────
|
||||
def _lock_for(self, resolved: str) -> threading.Lock:
|
||||
with self._meta_lock:
|
||||
lock = self._path_locks.get(resolved)
|
||||
if lock is None:
|
||||
lock = threading.Lock()
|
||||
self._path_locks[resolved] = lock
|
||||
return lock
|
||||
|
||||
@contextmanager
|
||||
def lock_path(self, resolved: str):
|
||||
"""Acquire the per-path lock for a read→modify→write section.
|
||||
|
||||
Same process, same filesystem — threads on the same path serialize.
|
||||
Different paths proceed in parallel.
|
||||
"""
|
||||
lock = self._lock_for(resolved)
|
||||
lock.acquire()
|
||||
try:
|
||||
yield
|
||||
finally:
|
||||
lock.release()
|
||||
|
||||
# ── Read/write accounting ───────────────────────────────────────
|
||||
def record_read(
|
||||
self,
|
||||
task_id: str,
|
||||
resolved: str,
|
||||
*,
|
||||
partial: bool = False,
|
||||
mtime: Optional[float] = None,
|
||||
) -> None:
|
||||
if _disabled():
|
||||
return
|
||||
if mtime is None:
|
||||
try:
|
||||
mtime = os.path.getmtime(resolved)
|
||||
except OSError:
|
||||
return
|
||||
now = time.time()
|
||||
with self._state_lock:
|
||||
agent_reads = self._reads[task_id]
|
||||
agent_reads[resolved] = (float(mtime), now, bool(partial))
|
||||
_cap_dict(agent_reads, _MAX_PATHS_PER_AGENT)
|
||||
|
||||
def note_write(
|
||||
self,
|
||||
task_id: str,
|
||||
resolved: str,
|
||||
*,
|
||||
mtime: Optional[float] = None,
|
||||
) -> None:
|
||||
"""Record a successful write.
|
||||
|
||||
Updates the global last-writer map AND this agent's own read stamp
|
||||
(a write is an implicit read — the agent now knows the current
|
||||
content).
|
||||
"""
|
||||
if _disabled():
|
||||
return
|
||||
if mtime is None:
|
||||
try:
|
||||
mtime = os.path.getmtime(resolved)
|
||||
except OSError:
|
||||
return
|
||||
now = time.time()
|
||||
with self._state_lock:
|
||||
self._last_writer[resolved] = (task_id, now)
|
||||
_cap_dict(self._last_writer, _MAX_GLOBAL_WRITERS)
|
||||
# Writer's own view is now up-to-date.
|
||||
self._reads[task_id][resolved] = (float(mtime), now, False)
|
||||
_cap_dict(self._reads[task_id], _MAX_PATHS_PER_AGENT)
|
||||
|
||||
def check_stale(self, task_id: str, resolved: str) -> Optional[str]:
|
||||
"""Return a model-facing warning if this write would be stale.
|
||||
|
||||
Three staleness classes, in order of severity:
|
||||
|
||||
1. Sibling subagent wrote this file after this agent's last read.
|
||||
2. External/unknown change (mtime differs from our last read).
|
||||
3. Agent never read the file (write-without-read).
|
||||
|
||||
Returns ``None`` when the write is safe. Does not raise — callers
|
||||
decide whether to block or warn.
|
||||
"""
|
||||
if _disabled():
|
||||
return None
|
||||
with self._state_lock:
|
||||
stamp = self._reads.get(task_id, {}).get(resolved)
|
||||
last_writer = self._last_writer.get(resolved)
|
||||
|
||||
# Case 3: never read AND we have no write record — net-new file or
|
||||
# first touch by this agent. Let existing _check_sensitive_path
|
||||
# and file-exists logic handle it; nothing to warn about here.
|
||||
if stamp is None and last_writer is None:
|
||||
return None
|
||||
|
||||
try:
|
||||
current_mtime = os.path.getmtime(resolved)
|
||||
except OSError:
|
||||
# File doesn't exist — write will create it; not stale.
|
||||
return None
|
||||
|
||||
# Case 1: sibling subagent modified after our last read.
|
||||
if last_writer is not None:
|
||||
writer_tid, writer_ts = last_writer
|
||||
if writer_tid != task_id:
|
||||
if stamp is None:
|
||||
return (
|
||||
f"{resolved} was modified by sibling subagent "
|
||||
f"{writer_tid!r} but this agent never read it. "
|
||||
"Read the file before writing to avoid overwriting "
|
||||
"the sibling's changes."
|
||||
)
|
||||
read_ts = stamp[1]
|
||||
if writer_ts > read_ts:
|
||||
return (
|
||||
f"{resolved} was modified by sibling subagent "
|
||||
f"{writer_tid!r} at {_fmt_ts(writer_ts)} — after "
|
||||
f"this agent's last read at {_fmt_ts(read_ts)}. "
|
||||
"Re-read the file before writing."
|
||||
)
|
||||
|
||||
# Case 2: external / unknown modification (mtime drifted).
|
||||
if stamp is not None:
|
||||
read_mtime, _read_ts, partial = stamp
|
||||
if current_mtime != read_mtime:
|
||||
return (
|
||||
f"{resolved} was modified since you last read it "
|
||||
"on disk (external edit or unrecorded writer). "
|
||||
"Re-read the file before writing."
|
||||
)
|
||||
if partial:
|
||||
return (
|
||||
f"{resolved} was last read with offset/limit pagination "
|
||||
"(partial view). Re-read the whole file before "
|
||||
"overwriting it."
|
||||
)
|
||||
|
||||
# Case 3b: agent truly never read the file.
|
||||
if stamp is None:
|
||||
return (
|
||||
f"{resolved} was not read by this agent. "
|
||||
"Read the file first so you can write an informed edit."
|
||||
)
|
||||
|
||||
return None
|
||||
|
||||
# ── Reminder helper for delegate_tool ───────────────────────────
|
||||
def writes_since(
|
||||
self,
|
||||
exclude_task_id: str,
|
||||
since_ts: float,
|
||||
paths: Iterable[str],
|
||||
) -> Dict[str, List[str]]:
|
||||
"""Return ``{writer_task_id: [paths]}`` for writes done after
|
||||
``since_ts`` by agents OTHER than ``exclude_task_id``.
|
||||
|
||||
Used by delegate_task to append a "subagent modified files the
|
||||
parent previously read" reminder to the delegation result.
|
||||
"""
|
||||
if _disabled():
|
||||
return {}
|
||||
paths_set = set(paths)
|
||||
out: Dict[str, List[str]] = defaultdict(list)
|
||||
with self._state_lock:
|
||||
for p, (writer_tid, ts) in self._last_writer.items():
|
||||
if writer_tid == exclude_task_id:
|
||||
continue
|
||||
if ts < since_ts:
|
||||
continue
|
||||
if p in paths_set:
|
||||
out[writer_tid].append(p)
|
||||
return dict(out)
|
||||
|
||||
def known_reads(self, task_id: str) -> List[str]:
|
||||
"""Return the list of resolved paths this agent has read."""
|
||||
if _disabled():
|
||||
return []
|
||||
with self._state_lock:
|
||||
return list(self._reads.get(task_id, {}).keys())
|
||||
|
||||
# ── Testing hooks ───────────────────────────────────────────────
|
||||
def clear(self) -> None:
|
||||
"""Reset all state. Intended for tests only."""
|
||||
with self._state_lock:
|
||||
self._reads.clear()
|
||||
self._last_writer.clear()
|
||||
with self._meta_lock:
|
||||
self._path_locks.clear()
|
||||
|
||||
|
||||
# ── Module-level singleton + helpers ─────────────────────────────────
|
||||
_registry = FileStateRegistry()
|
||||
|
||||
|
||||
def get_registry() -> FileStateRegistry:
|
||||
return _registry
|
||||
|
||||
|
||||
def _disabled() -> bool:
|
||||
# Re-read each call so tests can toggle via monkeypatch.setenv.
|
||||
return os.environ.get("HERMES_DISABLE_FILE_STATE_GUARD", "").strip() == "1"
|
||||
|
||||
|
||||
def _fmt_ts(ts: float) -> str:
|
||||
# Short relative wall-clock for error messages; avoids pulling in
|
||||
# datetime formatting overhead on the hot path.
|
||||
return time.strftime("%H:%M:%S", time.localtime(ts))
|
||||
|
||||
|
||||
def _cap_dict(d: dict, limit: int) -> None:
|
||||
"""Trim a dict to ``limit`` entries by dropping insertion-order oldest."""
|
||||
over = len(d) - limit
|
||||
if over <= 0:
|
||||
return
|
||||
# dict preserves insertion order (PY>=3.7) — pop the oldest keys.
|
||||
it = iter(d)
|
||||
for _ in range(over):
|
||||
try:
|
||||
d.pop(next(it))
|
||||
except (StopIteration, KeyError):
|
||||
break
|
||||
|
||||
|
||||
# ── Convenience wrappers (short names used at call sites) ────────────
|
||||
def record_read(task_id: str, resolved_or_path: str | Path, *, partial: bool = False) -> None:
|
||||
_registry.record_read(task_id, str(resolved_or_path), partial=partial)
|
||||
|
||||
|
||||
def note_write(task_id: str, resolved_or_path: str | Path) -> None:
|
||||
_registry.note_write(task_id, str(resolved_or_path))
|
||||
|
||||
|
||||
def check_stale(task_id: str, resolved_or_path: str | Path) -> Optional[str]:
|
||||
return _registry.check_stale(task_id, str(resolved_or_path))
|
||||
|
||||
|
||||
def lock_path(resolved_or_path: str | Path):
|
||||
return _registry.lock_path(str(resolved_or_path))
|
||||
|
||||
|
||||
def writes_since(
|
||||
exclude_task_id: str,
|
||||
since_ts: float,
|
||||
paths: Iterable[str | Path],
|
||||
) -> Dict[str, List[str]]:
|
||||
return _registry.writes_since(exclude_task_id, since_ts, [str(p) for p in paths])
|
||||
|
||||
|
||||
def known_reads(task_id: str) -> List[str]:
|
||||
return _registry.known_reads(task_id)
|
||||
|
||||
|
||||
__all__ = [
|
||||
"FileStateRegistry",
|
||||
"get_registry",
|
||||
"record_read",
|
||||
"note_write",
|
||||
"check_stale",
|
||||
"lock_path",
|
||||
"writes_since",
|
||||
"known_reads",
|
||||
]
|
||||
|
|
@ -12,6 +12,7 @@ from typing import Optional
|
|||
from agent.file_safety import get_read_block_error
|
||||
from tools.binary_extensions import has_binary_extension
|
||||
from tools.file_operations import ShellFileOperations
|
||||
from tools import file_state
|
||||
from agent.redact import redact_sensitive_text
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
|
@ -483,6 +484,19 @@ def read_file_tool(path: str, offset: int = 1, limit: int = 500, task_id: str =
|
|||
# accumulate megabytes of dict/set state. See _cap_read_tracker_data.
|
||||
_cap_read_tracker_data(task_data)
|
||||
|
||||
# Cross-agent file-state registry (separate from per-task read
|
||||
# tracker above): records that THIS agent has read this path so
|
||||
# write/patch can detect sibling-subagent writes that happened
|
||||
# after our read. Partial read when offset>1 or the read was
|
||||
# truncated (large file with more content than limit covered).
|
||||
# Outside the _read_tracker_lock so the registry's own locking
|
||||
# isn't nested under ours.
|
||||
try:
|
||||
_partial = (offset > 1) or bool(result_dict.get("truncated"))
|
||||
file_state.record_read(task_id, resolved_str, partial=_partial)
|
||||
except Exception:
|
||||
logger.debug("file_state.record_read failed", exc_info=True)
|
||||
|
||||
if count >= 4:
|
||||
# Hard block: stop returning content to break the loop
|
||||
return json.dumps({
|
||||
|
|
@ -602,15 +616,43 @@ def write_file_tool(path: str, content: str, task_id: str = "default") -> str:
|
|||
if sensitive_err:
|
||||
return tool_error(sensitive_err)
|
||||
try:
|
||||
stale_warning = _check_file_staleness(path, task_id)
|
||||
file_ops = _get_file_ops(task_id)
|
||||
result = file_ops.write_file(path, content)
|
||||
result_dict = result.to_dict()
|
||||
if stale_warning:
|
||||
result_dict["_warning"] = stale_warning
|
||||
# Refresh the stored timestamp so consecutive writes by this
|
||||
# task don't trigger false staleness warnings.
|
||||
_update_read_timestamp(path, task_id)
|
||||
# Resolve once for the registry lock + stale check. Failures here
|
||||
# fall back to the legacy path — write proceeds, per-task staleness
|
||||
# check below still runs.
|
||||
try:
|
||||
_resolved = str(_resolve_path(path))
|
||||
except Exception:
|
||||
_resolved = None
|
||||
|
||||
if _resolved is None:
|
||||
stale_warning = _check_file_staleness(path, task_id)
|
||||
file_ops = _get_file_ops(task_id)
|
||||
result = file_ops.write_file(path, content)
|
||||
result_dict = result.to_dict()
|
||||
if stale_warning:
|
||||
result_dict["_warning"] = stale_warning
|
||||
_update_read_timestamp(path, task_id)
|
||||
return json.dumps(result_dict, ensure_ascii=False)
|
||||
|
||||
# Serialize the read→modify→write region per-path so concurrent
|
||||
# subagents can't interleave on the same file. Different paths
|
||||
# remain fully parallel.
|
||||
with file_state.lock_path(_resolved):
|
||||
# Cross-agent staleness wins over per-task warning when both
|
||||
# fire — its message names the sibling subagent.
|
||||
cross_warning = file_state.check_stale(task_id, _resolved)
|
||||
stale_warning = _check_file_staleness(path, task_id)
|
||||
file_ops = _get_file_ops(task_id)
|
||||
result = file_ops.write_file(path, content)
|
||||
result_dict = result.to_dict()
|
||||
effective_warning = cross_warning or stale_warning
|
||||
if effective_warning:
|
||||
result_dict["_warning"] = effective_warning
|
||||
# Refresh stamps after the successful write so consecutive
|
||||
# writes by this task don't trigger false staleness warnings.
|
||||
_update_read_timestamp(path, task_id)
|
||||
if not result_dict.get("error"):
|
||||
file_state.note_write(task_id, _resolved)
|
||||
return json.dumps(result_dict, ensure_ascii=False)
|
||||
except Exception as e:
|
||||
if _is_expected_write_exception(e):
|
||||
|
|
@ -637,36 +679,70 @@ def patch_tool(mode: str = "replace", path: str = None, old_string: str = None,
|
|||
if sensitive_err:
|
||||
return tool_error(sensitive_err)
|
||||
try:
|
||||
# Check staleness for all files this patch will touch.
|
||||
stale_warnings = []
|
||||
# Resolve paths for locking. Ordered + deduplicated so concurrent
|
||||
# callers lock in the same order — prevents deadlock on overlapping
|
||||
# multi-file V4A patches.
|
||||
_resolved_paths: list[str] = []
|
||||
_seen: set[str] = set()
|
||||
for _p in _paths_to_check:
|
||||
_sw = _check_file_staleness(_p, task_id)
|
||||
if _sw:
|
||||
stale_warnings.append(_sw)
|
||||
try:
|
||||
_r = str(_resolve_path(_p))
|
||||
except Exception:
|
||||
_r = None
|
||||
if _r and _r not in _seen:
|
||||
_resolved_paths.append(_r)
|
||||
_seen.add(_r)
|
||||
_resolved_paths.sort()
|
||||
|
||||
file_ops = _get_file_ops(task_id)
|
||||
|
||||
if mode == "replace":
|
||||
if not path:
|
||||
return tool_error("path required")
|
||||
if old_string is None or new_string is None:
|
||||
return tool_error("old_string and new_string required")
|
||||
result = file_ops.patch_replace(path, old_string, new_string, replace_all)
|
||||
elif mode == "patch":
|
||||
if not patch:
|
||||
return tool_error("patch content required")
|
||||
result = file_ops.patch_v4a(patch)
|
||||
else:
|
||||
return tool_error(f"Unknown mode: {mode}")
|
||||
|
||||
result_dict = result.to_dict()
|
||||
if stale_warnings:
|
||||
result_dict["_warning"] = stale_warnings[0] if len(stale_warnings) == 1 else " | ".join(stale_warnings)
|
||||
# Refresh stored timestamps for all successfully-patched paths so
|
||||
# consecutive edits by this task don't trigger false warnings.
|
||||
if not result_dict.get("error"):
|
||||
# Acquire per-path locks in sorted order via ExitStack. On single
|
||||
# path this degenerates to one lock; on empty list (unresolvable)
|
||||
# it's a no-op and execution falls through unchanged.
|
||||
from contextlib import ExitStack
|
||||
with ExitStack() as _locks:
|
||||
for _r in _resolved_paths:
|
||||
_locks.enter_context(file_state.lock_path(_r))
|
||||
|
||||
# Collect warnings — cross-agent registry first (names sibling),
|
||||
# then per-task tracker as a fallback.
|
||||
stale_warnings: list[str] = []
|
||||
_path_to_resolved: dict[str, str] = {}
|
||||
for _p in _paths_to_check:
|
||||
_update_read_timestamp(_p, task_id)
|
||||
try:
|
||||
_r = str(_resolve_path(_p))
|
||||
except Exception:
|
||||
_r = None
|
||||
_path_to_resolved[_p] = _r
|
||||
_cross = file_state.check_stale(task_id, _r) if _r else None
|
||||
_sw = _cross or _check_file_staleness(_p, task_id)
|
||||
if _sw:
|
||||
stale_warnings.append(_sw)
|
||||
|
||||
file_ops = _get_file_ops(task_id)
|
||||
|
||||
if mode == "replace":
|
||||
if not path:
|
||||
return tool_error("path required")
|
||||
if old_string is None or new_string is None:
|
||||
return tool_error("old_string and new_string required")
|
||||
result = file_ops.patch_replace(path, old_string, new_string, replace_all)
|
||||
elif mode == "patch":
|
||||
if not patch:
|
||||
return tool_error("patch content required")
|
||||
result = file_ops.patch_v4a(patch)
|
||||
else:
|
||||
return tool_error(f"Unknown mode: {mode}")
|
||||
|
||||
result_dict = result.to_dict()
|
||||
if stale_warnings:
|
||||
result_dict["_warning"] = stale_warnings[0] if len(stale_warnings) == 1 else " | ".join(stale_warnings)
|
||||
# Refresh stored timestamps for all successfully-patched paths so
|
||||
# consecutive edits by this task don't trigger false warnings.
|
||||
if not result_dict.get("error"):
|
||||
for _p in _paths_to_check:
|
||||
_update_read_timestamp(_p, task_id)
|
||||
_r = _path_to_resolved.get(_p)
|
||||
if _r:
|
||||
file_state.note_write(task_id, _r)
|
||||
result_json = json.dumps(result_dict, ensure_ascii=False)
|
||||
# Hint when old_string not found — saves iterations where the agent
|
||||
# retries with stale content instead of re-reading the file.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue