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

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
This commit is contained in:
Ariel Frischer 2025-03-10 17:03:42 -07:00
parent 8f2adc7f61
commit 4d4eb6cadb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/ra_aid/test_agent_utils.py
View File

@ -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)
_handle_api_error(resource_exhausted_error, 0, 5, 1)