From 8d04f4686ff2daf7709b8cf52cedc19af3f10016 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 7 May 2024 21:14:33 -0700 Subject: [PATCH] test: fix test --- litellm/tests/test_proxy_custom_logger.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_proxy_custom_logger.py b/litellm/tests/test_proxy_custom_logger.py index 64ed08897f..5496b6c834 100644 --- 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",