mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-04-27 01:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

fix: test for suppress_status_output should capture stdout, not mock _vprint

Browse source 
The test was mocking _vprint entirely, bypassing the suppress guard.
Switch to capturing _print_fn output so the real _vprint runs and
the guard suppresses retry noise as intended.
This commit is contained in:
Teknium 2026-04-09 15:18:30 -07:00 • committed by Teknium
parent 69a0092c38
commit 3b554bf839
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/run_agent/test_run_agent.py
View file

@ -5,6 +5,7 @@ pieces. The OpenAI client and tool loading are mocked so no network calls
are made. are made.
""" """
import io
import json import json
import logging import logging
import re import re
@ -1120,15 +1121,17 @@ class TestExecuteToolCalls:
agent._save_trajectory = lambda *args, **kwargs: None agent._save_trajectory = lambda *args, **kwargs: None
agent._save_session_log = lambda *args, **kwargs: None agent._save_session_log = lambda *args, **kwargs: None
with patch("run_agent.time.sleep", return_value=None), \ captured = io.StringIO()
patch.object(agent, "_vprint") as mock_vprint: agent._print_fn = lambda *args, **kw: print(*args, file=captured, **kw)
with patch("run_agent.time.sleep", return_value=None):
result = agent.run_conversation("hello") result = agent.run_conversation("hello")
assert result["completed"] is True assert result["completed"] is True
assert result["final_response"] == "Recovered" assert result["final_response"] == "Recovered"
rendered = [" ".join(str(arg) for arg in call.args) for call in mock_vprint.call_args_list] output = captured.getvalue()
assert not any("API call failed" in line for line in rendered) assert "API call failed" not in output
assert not any("Rate limit reached" in line for line in rendered) assert "Rate limit reached" not in output
class TestConcurrentToolExecution: class TestConcurrentToolExecution: