diff --git a/src/google/adk/tools/bigquery/bigquery_credentials.py b/src/google/adk/tools/bigquery/bigquery_credentials.py index 27c3a786..0a99136c 100644 --- a/src/google/adk/tools/bigquery/bigquery_credentials.py +++ b/src/google/adk/tools/bigquery/bigquery_credentials.py @@ -31,12 +31,14 @@ from ...auth.auth_credential import AuthCredential from ...auth.auth_credential import AuthCredentialTypes from ...auth.auth_credential import OAuth2Auth from ...auth.auth_tool import AuthConfig +from ...utils.feature_decorator import experimental from ..tool_context import ToolContext BIGQUERY_TOKEN_CACHE_KEY = "bigquery_token_cache" BIGQUERY_DEFAULT_SCOPE = ["https://www.googleapis.com/auth/bigquery"] +@experimental class BigQueryCredentialsConfig(BaseModel): """Configuration for Google API tools. (Experimental)""" diff --git a/src/google/adk/tools/bigquery/bigquery_tool.py b/src/google/adk/tools/bigquery/bigquery_tool.py index 55301220..18273418 100644 --- a/src/google/adk/tools/bigquery/bigquery_tool.py +++ b/src/google/adk/tools/bigquery/bigquery_tool.py @@ -12,6 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +from __future__ import annotations import inspect from typing import Any @@ -21,6 +22,7 @@ from typing import Optional from google.oauth2.credentials import Credentials from typing_extensions import override +from ...utils.feature_decorator import experimental from ..function_tool import FunctionTool from ..tool_context import ToolContext from .bigquery_credentials import BigQueryCredentialsConfig @@ -28,6 +30,7 @@ from .bigquery_credentials import BigQueryCredentialsManager from .config import BigQueryToolConfig +@experimental class BigQueryTool(FunctionTool): """GoogleApiTool class for tools that call Google APIs. diff --git a/src/google/adk/tools/bigquery/bigquery_toolset.py b/src/google/adk/tools/bigquery/bigquery_toolset.py index 5543d103..313cf499 100644 --- a/src/google/adk/tools/bigquery/bigquery_toolset.py +++ b/src/google/adk/tools/bigquery/bigquery_toolset.py @@ -26,11 +26,13 @@ from . import query_tool from ...tools.base_tool import BaseTool from ...tools.base_toolset import BaseToolset from ...tools.base_toolset import ToolPredicate +from ...utils.feature_decorator import experimental from .bigquery_credentials import BigQueryCredentialsConfig from .bigquery_tool import BigQueryTool from .config import BigQueryToolConfig +@experimental class BigQueryToolset(BaseToolset): """BigQuery Toolset contains tools for interacting with BigQuery data and metadata."""