Skip to content

Commit abcf14c

Browse files
google-genai-botcopybara-github
authored andcommitted
feat: Promote BigQuery tools to Stable
In preparation to launching GA of the BigQuery agentic tools in ADK, this change marks BIG_QUERY_TOOLSET and BIG_QUERY_TOOL_CONFIG as stable features. The experimental decorators and warnings associated with these features have been removed. PiperOrigin-RevId: 897865928
1 parent e7d8160 commit abcf14c

File tree

6 files changed

+4
-18
lines changed

6 files changed

+4
-18
lines changed

src/google/adk/features/_feature_registry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,10 +99,10 @@ class FeatureConfig:
9999
FeatureStage.EXPERIMENTAL, default_on=True
100100
),
101101
FeatureName.BIG_QUERY_TOOLSET: FeatureConfig(
102-
FeatureStage.EXPERIMENTAL, default_on=True
102+
FeatureStage.STABLE, default_on=True
103103
),
104104
FeatureName.BIG_QUERY_TOOL_CONFIG: FeatureConfig(
105-
FeatureStage.EXPERIMENTAL, default_on=True
105+
FeatureStage.STABLE, default_on=True
106106
),
107107
FeatureName.BIGTABLE_TOOL_SETTINGS: FeatureConfig(
108108
FeatureStage.EXPERIMENTAL, default_on=True

src/google/adk/tools/bigquery/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
"""BigQuery Tools (Experimental).
15+
"""BigQuery Tools.
1616
1717
BigQuery Tools under this module are hand crafted and customized while the tools
1818
under google.adk.tools.google_api_tool are auto generated based on API

src/google/adk/tools/bigquery/bigquery_credentials.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414

1515
from __future__ import annotations
1616

17-
from ...features import experimental
1817
from ...features import FeatureName
1918
from .._google_credentials import BaseGoogleCredentialsConfig
2019

@@ -26,9 +25,8 @@
2625
BIGQUERY_DEFAULT_SCOPE = ["https://www.googleapis.com/auth/bigquery"]
2726

2827

29-
@experimental(FeatureName.GOOGLE_CREDENTIALS_CONFIG)
3028
class BigQueryCredentialsConfig(BaseGoogleCredentialsConfig):
31-
"""BigQuery Credentials Configuration for Google API tools (Experimental).
29+
"""BigQuery Credentials Configuration for Google API tools.
3230
3331
Please do not use this in production, as it may be deprecated later.
3432
"""

src/google/adk/tools/bigquery/bigquery_toolset.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from . import metadata_tool
2626
from . import query_tool
2727
from . import search_tool
28-
from ...features import experimental
2928
from ...features import FeatureName
3029
from ...tools.base_tool import BaseTool
3130
from ...tools.base_toolset import BaseToolset
@@ -35,7 +34,6 @@
3534
from .config import BigQueryToolConfig
3635

3736

38-
@experimental(FeatureName.BIG_QUERY_TOOLSET)
3937
class BigQueryToolset(BaseToolset):
4038
"""BigQuery Toolset contains tools for interacting with BigQuery data and metadata."""
4139

src/google/adk/tools/bigquery/config.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from pydantic import ConfigDict
2222
from pydantic import field_validator
2323

24-
from ...features import experimental
2524
from ...features import FeatureName
2625

2726

@@ -48,7 +47,6 @@ class WriteMode(Enum):
4847
"""All write operations are allowed."""
4948

5049

51-
@experimental(FeatureName.BIG_QUERY_TOOL_CONFIG)
5250
class BigQueryToolConfig(BaseModel):
5351
"""Configuration for BigQuery tools."""
5452

tests/unittests/tools/bigquery/test_bigquery_tool_config.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,6 @@ def reset_warned_features():
2727
_WARNED_FEATURES.clear()
2828

2929

30-
def test_bigquery_tool_config_experimental_warning():
31-
"""Test BigQueryToolConfig experimental warning."""
32-
with warnings.catch_warnings(record=True) as w:
33-
BigQueryToolConfig()
34-
assert len(w) == 1
35-
assert "BIG_QUERY_TOOL_CONFIG is enabled." in str(w[0].message)
36-
37-
3830
def test_bigquery_tool_config_invalid_property():
3931
"""Test BigQueryToolConfig raises exception when setting invalid property."""
4032
with pytest.raises(

0 commit comments

Comments
 (0)