test: fix test

2026-07-11 17:05:43 +00:00 · 2024-05-07 21:14:33 -07:00
parent dc8239ecc6
commit 8d04f4686f
1 changed files with 2 additions and 0 deletions
@@ -117,6 +117,7 @@ def test_embedding(client):
            "input_cost_per_token": 0.002,
            "mode": "embedding",
            "id": "hello",
+            "db_model": False,
        }
        result = response.json()
        print(f"Received response: {result}")
@@ -191,6 +192,7 @@ def test_chat_completion(client):
            "id": "gm",
            "input_cost_per_token": 0.0002,
            "mode": "chat",
+            "db_model": False,
        }
        assert proxy_server_request_object == {
            "url": "http://testserver/chat/completions",