test: fix test

2024-05-07 21:14:33 -07:00 · 2024-05-07 21:14:33 -07:00 · 8d04f4686f
commit 8d04f4686f
parent dc8239ecc6
1 changed files with 2 additions and 0 deletions
--- a/litellm/tests/test_proxy_custom_logger.py
+++ b/litellm/tests/test_proxy_custom_logger.py
@ -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",