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

Revert "test_acompletion_basic_request"

Browse source 
This reverts commit fdaa5bc49e.
This commit is contained in:
Ishaan Jaffer 2025-10-10 11:11:00 -07:00
parent fdaa5bc49e
commit e9ac4bfc0e
1 changed files with 10 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/llms/vertex_ai/vertex_gemma_models/test_vertex_gemma_transformation.py
View file

@ -135,7 +135,6 @@ class TestVertexGemmaCompletion:
assert call_args is not None, "HTTP handler was not called"
request_data = call_args.kwargs["json"]
print("request body=", json.dumps(request_data, indent=4))
request_url = call_args.kwargs["url"]
# Validate exact URL matches what we sent
@ -146,14 +145,17 @@ class TestVertexGemmaCompletion:
assert "instances" in request_data
assert len(request_data["instances"]) == 1
instance = request_data["instances"][0]
assert instance["@requestFormat"] == "chatCompletions"
outer_instance = request_data["instances"][0]
assert outer_instance["@requestFormat"] == "chatCompletions"
# Messages should be directly in the instance, not double-nested
assert "messages" in instance
assert instance["messages"][0]["role"] == "user"
assert instance["messages"][0]["content"] == "What is machine learning?"
assert instance["max_tokens"] == 100
# The actual instance with messages is nested inside
assert "instances" in outer_instance
inner_instance = outer_instance["instances"][0]
assert inner_instance["@requestFormat"] == "chatCompletions"
assert "messages" in inner_instance
assert inner_instance["messages"][0]["role"] == "user"
assert inner_instance["messages"][0]["content"] == "What is machine learning?"
assert inner_instance["max_tokens"] == 100
# Validate LiteLLM Response (OpenAI format)
assert response.id == "chatcmpl-aaa4288f-2b8e-4bc0-8b14-4e444decd2c4"