From e9ac4bfc0e253fe4b2473502c0b4886d4b419b05 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 10 Oct 2025 11:11:00 -0700 Subject: [PATCH] Revert "test_acompletion_basic_request" This reverts commit fdaa5bc49ebc61773564872ddf255fcf50c2e9e4. --- .../test_vertex_gemma_transformation.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) 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 fb988485791..1ff9009bb8f 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,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"