From bc8b14a7c9c632721db9166dc9b63eec17d31afd Mon Sep 17 00:00:00 2001
From: Matthew Tang <tangmatthew@google.com>
Date: Mon, 3 Jun 2024 20:54:42 -0700
Subject: [PATCH] feat: GenAI - Release ToolConfig to GA

PiperOrigin-RevId: 640001644
---
 tests/unit/vertexai/test_generative_models.py | 2 +-
 vertexai/generative_models/__init__.py        | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/unit/vertexai/test_generative_models.py b/tests/unit/vertexai/test_generative_models.py
index 391e30e554..36a2c1dc72 100644
--- 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
index 6c3eb34ae9..71f0d07cab 100644
--- 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",
 ]