scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

test_acompletion_basic_request

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-10 11:10:37 -07:00
parent 68a6e4aa27
commit fdaa5bc49e
1 changed files with 8 additions and 10 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,6 +135,7 @@ 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
@ -145,17 +146,14 @@ class TestVertexGemmaCompletion:
assert "instances" in request_data
assert len(request_data["instances"]) == 1
outer_instance = request_data["instances"][0]
assert outer_instance["@requestFormat"] == "chatCompletions"
instance = request_data["instances"][0]
assert instance["@requestFormat"] == "chatCompletions"
# 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
# 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
# Validate LiteLLM Response (OpenAI format)
assert response.id == "chatcmpl-aaa4288f-2b8e-4bc0-8b14-4e444decd2c4"