mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-04 07:31:58 +00:00
66827f8947
Remove unused imports (F401) and duplicate/shadowed import redefinitions (F811) across the codebase using ruff's safe autofixes. No behavioral changes -- imports only. - ~1400 safe autofixes applied across 644 files (net -1072 lines) - __init__.py re-exports preserved (excluded from F401 removal so public re-export surfaces stay intact) - Re-exports that are imported or monkeypatched by tests but look unused in their defining module are kept with explicit # noqa: F401 (gateway/run.py load_dotenv; run_agent re-exports from agent.message_sanitization, agent.context_compressor, agent.retry_utils, agent.prompt_builder, agent.process_bootstrap, agent.codex_responses_adapter) - Unsafe F841 (unused-variable) fixes deliberately skipped -- those can change behavior when the RHS has side effects - ruff lints remain disabled in pyproject.toml (only PLW1514 is selected); this is a one-time cleanup, not a config change Verification: - python -m compileall: clean - pytest --collect-only: all 27161 tests collect (zero import errors) - core entry points import clean (run_agent, model_tools, cli, toolsets, hermes_state, batch_runner, gateway) - static scan: every name any test imports directly from an edited module still resolves
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
"""Tests for IterationBudget thread safety.
|
|
|
|
The `used` property must acquire the lock before reading `_used` to prevent
|
|
data races with concurrent `consume()` / `refund()` calls.
|
|
"""
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
|
|
|
|
def test_iteration_budget_used_is_thread_safe():
|
|
"""Iterating `used` while other threads consume/refund must not crash.
|
|
|
|
Before the fix, `used` returned `_used` directly without holding the lock,
|
|
so a concurrent `consume()` could observe a partially-updated value or
|
|
cause the C-level `list.append` to raise a ValueError ("list size changed").
|
|
"""
|
|
from run_agent import IterationBudget
|
|
|
|
budget = IterationBudget(max_total=1000)
|
|
num_threads = 10
|
|
operations_per_thread = 200
|
|
|
|
errors = []
|
|
|
|
def worker(consume: bool):
|
|
try:
|
|
for _ in range(operations_per_thread):
|
|
if consume:
|
|
budget.consume()
|
|
else:
|
|
budget.refund()
|
|
# Also read `used` to exercise the property
|
|
_ = budget.used
|
|
except Exception as exc:
|
|
errors.append(exc)
|
|
|
|
with ThreadPoolExecutor(max_workers=num_threads * 2) as executor:
|
|
# Half the threads consume, half refund
|
|
futures = []
|
|
for i in range(num_threads):
|
|
consume = i < num_threads // 2
|
|
futures.append(executor.submit(worker, consume))
|
|
futures.append(executor.submit(worker, consume))
|
|
|
|
for f in futures:
|
|
f.result()
|
|
|
|
assert not errors, f"Thread safety violation: {errors}"
|
|
# Final value should be within expected bounds
|
|
assert 0 <= budget.used <= budget.max_total
|
|
|
|
|
|
def test_iteration_budget_consume_returns_false_when_exhausted():
|
|
"""consume() must return False once the budget is exhausted."""
|
|
from run_agent import IterationBudget
|
|
|
|
budget = IterationBudget(max_total=3)
|
|
assert budget.consume() is True
|
|
assert budget.consume() is True
|
|
assert budget.consume() is True
|
|
assert budget.consume() is False
|
|
|
|
|
|
def test_iteration_budget_refund_restores_consume():
|
|
"""refund() after consume() must allow one more consume()."""
|
|
from run_agent import IterationBudget
|
|
|
|
budget = IterationBudget(max_total=2)
|
|
assert budget.consume() is True
|
|
assert budget.consume() is True
|
|
assert budget.consume() is False # exhausted
|
|
budget.refund()
|
|
assert budget.consume() is True
|
|
|
|
|
|
def test_iteration_budget_used_reflects_consume_and_refund():
|
|
"""used property must accurately reflect consume() and refund() calls."""
|
|
from run_agent import IterationBudget
|
|
|
|
budget = IterationBudget(max_total=10)
|
|
|
|
assert budget.used == 0
|
|
budget.consume()
|
|
assert budget.used == 1
|
|
budget.consume()
|
|
assert budget.used == 2
|
|
budget.refund()
|
|
assert budget.used == 1
|
|
budget.refund()
|
|
assert budget.used == 0
|
|
|
|
|
|
def test_iteration_budget_remaining():
|
|
"""remaining property must equal max_total - used."""
|
|
from run_agent import IterationBudget
|
|
|
|
budget = IterationBudget(max_total=5)
|
|
|
|
assert budget.remaining == 5
|
|
budget.consume()
|
|
assert budget.remaining == 4
|
|
budget.consume()
|
|
budget.consume()
|
|
assert budget.remaining == 2
|
|
budget.refund()
|
|
assert budget.remaining == 3
|