pin version (#1124)

deepset-ai · Oct 4, 2024 · cc5defb · cc5defb
1 parent dbd844c
commit cc5defb
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 13 deletions.
diff --git a/integrations/google_vertex/pyproject.toml b/integrations/google_vertex/pyproject.toml
@@ -22,7 +22,7 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: CPython",
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
-dependencies = ["haystack-ai", "google-cloud-aiplatform>=1.38", "pyarrow>3"]
+dependencies = ["haystack-ai", "google-cloud-aiplatform>=1.61", "pyarrow>3"]
 
 [project.urls]
 Documentation = "https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/google_vertex#readme"

diff --git a/integrations/google_vertex/tests/chat/test_gemini.py b/integrations/google_vertex/tests/chat/test_gemini.py
@@ -22,11 +22,11 @@
     name="get_current_weather",
     description="Get the current weather in a given location",
     parameters={
-        "type_": "OBJECT",
+        "type": "object",
         "properties": {
-            "location": {"type_": "STRING", "description": "The city and state, e.g. San Francisco, CA"},
+            "location": {"type": "string", "description": "The city and state, e.g. San Francisco, CA"},
             "unit": {
-                "type_": "STRING",
+                "type": "string",
                 "enum": [
                     "celsius",
                     "fahrenheit",
@@ -238,13 +238,19 @@ def test_from_dict_with_param(_mock_vertexai_init, _mock_generative_model):
                                 "name": "get_current_weather",
                                 "description": "Get the current weather in a given location",
                                 "parameters": {
-                                    "type_": "OBJECT",
+                                    "type": "object",
                                     "properties": {
                                         "location": {
-                                            "type_": "STRING",
+                                            "type": "string",
                                             "description": "The city and state, e.g. San Francisco, CA",
                                         },
-                                        "unit": {"type_": "STRING", "enum": ["celsius", "fahrenheit"]},
+                                        "unit": {
+                                            "type": "string",
+                                            "enum": [
+                                                "celsius",
+                                                "fahrenheit",
+                                            ],
+                                        },
                                     },
                                     "required": ["location"],
                                 },

diff --git a/integrations/google_vertex/tests/test_gemini.py b/integrations/google_vertex/tests/test_gemini.py
@@ -18,11 +18,11 @@
     name="get_current_weather",
     description="Get the current weather in a given location",
     parameters={
-        "type_": "OBJECT",
+        "type": "object",
         "properties": {
-            "location": {"type_": "STRING", "description": "The city and state, e.g. San Francisco, CA"},
+            "location": {"type": "string", "description": "The city and state, e.g. San Francisco, CA"},
             "unit": {
-                "type_": "STRING",
+                "type": "string",
                 "enum": [
                     "celsius",
                     "fahrenheit",
@@ -226,13 +226,19 @@ def test_from_dict_with_param(_mock_vertexai_init, _mock_generative_model):
                             {
                                 "name": "get_current_weather",
                                 "parameters": {
-                                    "type_": "OBJECT",
+                                    "type": "object",
                                     "properties": {
-                                        "unit": {"type_": "STRING", "enum": ["celsius", "fahrenheit"]},
                                         "location": {
-                                            "type_": "STRING",
+                                            "type": "string",
                                             "description": "The city and state, e.g. San Francisco, CA",
                                         },
+                                        "unit": {
+                                            "type": "string",
+                                            "enum": [
+                                                "celsius",
+                                                "fahrenheit",
+                                            ],
+                                        },
                                     },
                                     "required": ["location"],
                                 },