From b4392d2b67db86f62a58611f63b7fb9d5ad29ccc Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 7 Aug 2024 18:39:04 -0700 Subject: [PATCH] test(test_amazing_vertex_completion.py): fix test for json schema validation in openai schema --- litellm/tests/test_amazing_vertex_completion.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 53bb9fd8030..bad2428fbe2 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -1488,6 +1488,9 @@ async def test_gemini_pro_json_schema_args_sent_httpx_openai_schema( ): from typing import List + if enforce_validation: + litellm.enable_json_schema_validation = True + from pydantic import BaseModel load_vertex_ai_credentials()