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

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