diff --git a/tests/test_litellm/llms/vertex_ai/vertex_gemma_models/test_vertex_gemma_transformation.py b/tests/test_litellm/llms/vertex_ai/vertex_gemma_models/test_vertex_gemma_transformation.py index 1ff9009bb8..fb98848579 100644 --- a/tests/test_litellm/llms/vertex_ai/vertex_gemma_models/test_vertex_gemma_transformation.py +++ b/tests/test_litellm/llms/vertex_ai/vertex_gemma_models/test_vertex_gemma_transformation.py @@ -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"