phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-07 21:14:33 -07:00
parent dc8239ecc6
commit 8d04f4686f
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_proxy_custom_logger.py
View file

@ -117,6 +117,7 @@ def test_embedding(client):
"input_cost_per_token": 0.002, "input_cost_per_token": 0.002,
"mode": "embedding", "mode": "embedding",
"id": "hello", "id": "hello",
"db_model": False,
} }
result = response.json() result = response.json()
print(f"Received response: {result}") print(f"Received response: {result}")
@ -191,6 +192,7 @@ def test_chat_completion(client):
"id": "gm", "id": "gm",
"input_cost_per_token": 0.0002, "input_cost_per_token": 0.0002,
"mode": "chat", "mode": "chat",
"db_model": False,
} }
assert proxy_server_request_object == { assert proxy_server_request_object == {
"url": "http://testserver/chat/completions", "url": "http://testserver/chat/completions",