scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

test fixes

Browse source
This commit is contained in:
Ishaan Jaffer 2026-01-10 15:15:41 -08:00
parent 1ebe979eee
commit f1d516f0ae
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/proxy_unit_tests/test_proxy_custom_logger.py
View file

@ -91,7 +91,9 @@ def test_embedding(client):
assert "headers" in metadata
proxy_server_request = litellm_params.get("proxy_server_request")
model_info = litellm_params.get("model_info")
assert proxy_server_request == {
# Remove arrival_time from comparison as it's dynamic
proxy_server_request_copy = {k: v for k, v in proxy_server_request.items() if k != "arrival_time"}
assert proxy_server_request_copy == {
"url": "http://testserver/embeddings",
"method": "POST",
"headers": {
@ -188,7 +190,9 @@ def test_chat_completion(client):
}
assert "authorization" not in proxy_server_request_object["headers"]
assert proxy_server_request_object == {
# Remove arrival_time from comparison as it's dynamic
proxy_server_request_copy = {k: v for k, v in proxy_server_request_object.items() if k != "arrival_time"}
assert proxy_server_request_copy == {
"url": "http://testserver/chat/completions",
"method": "POST",
"headers": {