From 4d4eb6cadb428ef3a7238ddac72b07d706500b8c Mon Sep 17 00:00:00 2001 From: Ariel Frischer Date: Mon, 10 Mar 2025 17:03:42 -0700 Subject: [PATCH] fix(tests): add cost_cb parameter to fake_print_agent_output function to enhance flexibility in testing style(tests): add newline at the end of test_handle_api_error_resource_exhausted function for consistency with PEP 8 guidelines --- tests/ra_aid/test_agent_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ra_aid/test_agent_utils.py b/tests/ra_aid/test_agent_utils.py index 2361518..5292317 100644 --- a/tests/ra_aid/test_agent_utils.py +++ b/tests/ra_aid/test_agent_utils.py @@ -397,7 +397,7 @@ def test_run_agent_stream(monkeypatch, mock_config_repository): call_flag = {"called": False} def fake_print_agent_output( - chunk: Dict[str, Any], agent_type: Literal["CiaynAgent", "React"] + chunk: Dict[str, Any], agent_type: Literal["CiaynAgent", "React"], cost_cb=None ): call_flag["called"] = True @@ -725,4 +725,4 @@ def test_handle_api_error_resource_exhausted(): # ResourceExhausted exception should be handled without raising resource_exhausted_error = ResourceExhausted("429 Resource has been exhausted (e.g. check quota).") - _handle_api_error(resource_exhausted_error, 0, 5, 1) \ No newline at end of file + _handle_api_error(resource_exhausted_error, 0, 5, 1)