From 82b4bc8dbc3715f8f49a98343992faa717553793 Mon Sep 17 00:00:00 2001 From: Paul Swingle Date: Mon, 22 Apr 2024 16:47:37 -0700 Subject: [PATCH] fix tests --- tests/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 089295956..234e661f7 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, None) + return SpiceResponse(SpiceCallArgs("gpt-3.5-turbo", [], False), value, 1, 0, 0, True, 1) 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, None) + return SpiceResponse(SpiceCallArgs("gpt-3.5-turbo", [], True), "".join(values), 1, 0, 0, True, 1) mock_spice_response = MockStreamingSpiceResponse() return mock_spice_response