From c6f3d583080ca41496858838e3072d6f9541eb6c Mon Sep 17 00:00:00 2001 From: Paul Swingle Date: Mon, 22 Apr 2024 13:25:17 -0700 Subject: [PATCH] fix tests again --- tests/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 234e661f7..089295956 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -98,7 +98,7 @@ def mock_call_llm_api(mocker): completion_mock = mocker.patch.object(LlmApiHandler, "call_llm_api") def wrap_unstreamed_string(value): - return SpiceResponse(SpiceCallArgs("gpt-3.5-turbo", [], False), value, 1, 0, 0, True, 1) + return SpiceResponse(SpiceCallArgs("gpt-3.5-turbo", [], False), value, 1, 0, 0, True, 1, None) def wrap_streamed_strings(values): class MockStreamingSpiceResponse: @@ -115,7 +115,7 @@ async def __anext__(self): return values[self.cur_value - 1] def current_response(self): - return SpiceResponse(SpiceCallArgs("gpt-3.5-turbo", [], True), "".join(values), 1, 0, 0, True, 1) + return SpiceResponse(SpiceCallArgs("gpt-3.5-turbo", [], True), "".join(values), 1, 0, 0, True, 1, None) mock_spice_response = MockStreamingSpiceResponse() return mock_spice_response