fix: Make all subclass of BaseToolset to call parent constructor

So that all member field are created / initialized in the base toolset.

PiperOrigin-RevId: 794388671
This commit is contained in:
Xiang (Sean) Zhou
2025-08-12 21:31:43 -07:00
committed by Copybara-Service
parent ebd726f1f5
commit 8c65967cdc
4 changed files with 3 additions and 4 deletions
@@ -134,7 +134,6 @@ class ApplicationIntegrationToolset(BaseToolset):
self._connection = connection
self._entity_operations = entity_operations
self._actions = actions
self._tool_name_prefix = tool_name_prefix
self._tool_instructions = tool_instructions
self._service_account_json = service_account_json
self._auth_scheme = auth_scheme
@@ -43,7 +43,7 @@ class BigQueryToolset(BaseToolset):
credentials_config: Optional[BigQueryCredentialsConfig] = None,
bigquery_tool_config: Optional[BigQueryToolConfig] = None,
):
self.tool_filter = tool_filter
super().__init__(tool_filter=tool_filter)
self._credentials_config = credentials_config
self._tool_settings = (
bigquery_tool_config if bigquery_tool_config else BigQueryToolConfig()
@@ -47,13 +47,13 @@ class GoogleApiToolset(BaseToolset):
tool_filter: Optional[Union[ToolPredicate, List[str]]] = None,
service_account: Optional[ServiceAccount] = None,
):
super().__init__(tool_filter=tool_filter)
self.api_name = api_name
self.api_version = api_version
self._client_id = client_id
self._client_secret = client_secret
self._service_account = service_account
self._openapi_toolset = self._load_toolset_with_oidc_auth()
self.tool_filter = tool_filter
@override
async def get_tools(
@@ -44,7 +44,7 @@ class SpannerToolset(BaseToolset):
credentials_config: Optional[SpannerCredentialsConfig] = None,
spanner_tool_settings: Optional[SpannerToolSettings] = None,
):
self.tool_filter = tool_filter
super().__init__(tool_filter=tool_filter)
self._credentials_config = credentials_config
self._tool_settings = (
spanner_tool_settings