feat: GenAI - Release ToolConfig to GA

PiperOrigin-RevId: 640001644
googleapis · Jun 4, 2024 · bc8b14a · bc8b14a
1 parent 51431d7
commit bc8b14a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
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",
 ]