From 8db9990764dd95db2caac82de0b969bf456a7800 Mon Sep 17 00:00:00 2001 From: Stefano Fiorucci Date: Thu, 31 Oct 2024 10:47:06 +0100 Subject: [PATCH] chore: fix Vertex tests (#1163) * try fixing tests * more fixes --- integrations/google_vertex/tests/chat/test_gemini.py | 1 + integrations/google_vertex/tests/test_gemini.py | 1 + 2 files changed, 2 insertions(+) diff --git a/integrations/google_vertex/tests/chat/test_gemini.py b/integrations/google_vertex/tests/chat/test_gemini.py index 0d77bd9c6..73c99fe2f 100644 --- a/integrations/google_vertex/tests/chat/test_gemini.py +++ b/integrations/google_vertex/tests/chat/test_gemini.py @@ -170,6 +170,7 @@ def test_to_dict_with_params(_mock_vertexai_init, _mock_generative_model): "unit": {"type_": "STRING", "enum": ["celsius", "fahrenheit"]}, }, "required": ["location"], + "property_ordering": ["location", "unit"], }, } ] diff --git a/integrations/google_vertex/tests/test_gemini.py b/integrations/google_vertex/tests/test_gemini.py index b3d6dd5f5..277851224 100644 --- a/integrations/google_vertex/tests/test_gemini.py +++ b/integrations/google_vertex/tests/test_gemini.py @@ -157,6 +157,7 @@ def test_to_dict_with_params(_mock_vertexai_init, _mock_generative_model): "unit": {"type_": "STRING", "enum": ["celsius", "fahrenheit"]}, }, "required": ["location"], + "property_ordering": ["location", "unit"], }, } ]