feat: GenAI - Release ToolConfig to GA

matthew29tang · copybara-github · commit bc8b14a7c9c6 · 2024-06-03T20:58:01.000-07:00
PiperOrigin-RevId: 640001644
diff --git a/tests/unit/vertexai/test_generative_models.py b/tests/unit/vertexai/test_generative_models.py
@@ -749,7 +749,7 @@ def test_chat_function_calling(self, generative_models: generative_models):
     )
     @pytest.mark.parametrize(
         "generative_models",
-        [preview_generative_models],
+        [generative_models, preview_generative_models],
     )
     def test_chat_forced_function_calling(self, generative_models: generative_models):
         get_current_weather_func = generative_models.FunctionDeclaration(
diff --git a/vertexai/generative_models/__init__.py b/vertexai/generative_models/__init__.py
@@ -32,6 +32,7 @@
     ResponseValidationError,
     SafetySetting,
     Tool,
+    ToolConfig,
     grounding,
 )
 
@@ -51,5 +52,6 @@
     "ResponseValidationError",
     "SafetySetting",
     "Tool",
+    "ToolConfig",
     "grounding",
 ]

Original file line number	Diff line number	Diff line change
`@@ -749,7 +749,7 @@ def test_chat_function_calling(self, generative_models: generative_models):`
`749`	`749`	`)`
`750`	`750`	`@pytest.mark.parametrize(`
`751`	`751`	`"generative_models",`
`752`		`- [preview_generative_models],`
	`752`	`+ [generative_models, preview_generative_models],`
`753`	`753`	`)`
`754`	`754`	`def test_chat_forced_function_calling(self, generative_models: generative_models):`
`755`	`755`	`get_current_weather_func = generative_models.FunctionDeclaration(`