zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix tests

This commit is contained in:
AI Christianson 2025-03-07 18:47:12 -05:00
parent 60e4616313
commit 58c0739cfc
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/ra_aid/test_agent_utils.py
View File

@ -151,6 +151,7 @@ def test_create_agent_anthropic(mock_model, mock_config_repository):
mock_react.assert_called_once_with( mock_react.assert_called_once_with(
mock_model, mock_model,
[], [],
interrupt_after=['tools'],
version="v2", version="v2",
state_modifier=mock_react.call_args[1]["state_modifier"], state_modifier=mock_react.call_args[1]["state_modifier"],
) )
@ -295,7 +296,7 @@ def test_create_agent_anthropic_token_limiting_disabled(mock_model, mock_config_
agent = create_agent(mock_model, []) agent = create_agent(mock_model, [])
assert agent == "react_agent" assert agent == "react_agent"
mock_react.assert_called_once_with(mock_model, [], version="v2") mock_react.assert_called_once_with(mock_model, [], interrupt_after=['tools'], version="v2")
def test_get_model_token_limit_research(mock_config_repository): def test_get_model_token_limit_research(mock_config_repository):
@ -372,10 +373,19 @@ def test_agent_context_depth():
def test_run_agent_stream(monkeypatch): def test_run_agent_stream(monkeypatch):
from ra_aid.agent_utils import _run_agent_stream from ra_aid.agent_utils import _run_agent_stream
# Create a dummy agent that yields one chunk # Create a simple state class with a next property
class State:
def __init__(self):
self.next = None
# Create a dummy agent that yields one chunk and has a get_state method
class DummyAgent: class DummyAgent:
def stream(self, input_data, cfg: dict): def stream(self, input_data, cfg: dict):
yield {"content": "chunk1"} yield {"content": "chunk1"}
def get_state(self, state_config=None):
# Return an object with a next property set to None
return State()
dummy_agent = DummyAgent() dummy_agent = DummyAgent()
# Set flags so that _run_agent_stream will reset them # Set flags so that _run_agent_stream will reset them