Fix for backend tests

Signed-off-by: Samuel Monson <[email protected]>

Author: sjmonson

tests/unit/conftest.py

@@ -132,7 +132,6 @@ async def _mock_completions_response(request) -> AsyncIterable[str]:
             assert payload["stream_options"] == {"include_usage": True}
             assert payload["prompt"] is not None
             assert len(payload["prompt"]) > 0
-            assert payload["max_completion_tokens"] > 0
             assert payload["max_tokens"] > 0
 
             return httpx.Response(  # type: ignore
@@ -141,7 +140,7 @@ async def _mock_completions_response(request) -> AsyncIterable[str]:
                     type_="text",
                     prompt=payload["prompt"],
                     output_token_count=(
-                        payload["max_completion_tokens"]
+                        payload["max_tokens"]
                         if payload.get("ignore_eos", False)
                         else None
                     ),
@@ -162,7 +161,6 @@ async def _mock_chat_completions_response(request):
             assert payload["messages"] is not None
             assert len(payload["messages"]) > 0
             assert payload["max_completion_tokens"] > 0
-            assert payload["max_tokens"] > 0
 
             return httpx.Response(  # type: ignore
                 200,