NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 01:41:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor apihub toolset to hide non public field

Browse Source 
PiperOrigin-RevId: 758378396
This commit is contained in:
Xiang (Sean) Zhou 2025-05-13 14:31:55 -07:00 committed by Copybara-Service
parent cf06cc507a
commit f6bdd555fb
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/google/adk/tools/apihub_tool/apihub_toolset.py
View File

@ -133,19 +133,19 @@ class APIHubToolset(BaseToolset):
""" """
self.name = name self.name = name
self.description = description self.description = description
self.apihub_resource_name = apihub_resource_name self._apihub_resource_name = apihub_resource_name
self.lazy_load_spec = lazy_load_spec self._lazy_load_spec = lazy_load_spec
self.apihub_client = apihub_client or APIHubClient( self._apihub_client = apihub_client or APIHubClient(
access_token=access_token, access_token=access_token,
service_account_json=service_account_json, service_account_json=service_account_json,
) )
self.openapi_toolset = None self._openapi_toolset = None
self.auth_scheme = auth_scheme self._auth_scheme = auth_scheme
self.auth_credential = auth_credential self._auth_credential = auth_credential
self.tool_filter = tool_filter self.tool_filter = tool_filter
if not self.lazy_load_spec: if not self._lazy_load_spec:
self._prepare_toolset() self._prepare_toolset()
@override @override
@ -157,16 +157,16 @@ class APIHubToolset(BaseToolset):
Returns: Returns:
A list of all available RestApiTool objects. A list of all available RestApiTool objects.
""" """
if not self.openapi_toolset: if not self._openapi_toolset:
self._prepare_toolset() self._prepare_toolset()
if not self.openapi_toolset: if not self._openapi_toolset:
return [] return []
return await self.openapi_toolset.get_tools(readonly_context) return await self._openapi_toolset.get_tools(readonly_context)
def _prepare_toolset(self) -> None: def _prepare_toolset(self) -> None:
"""Fetches the spec from API Hub and generates the toolset.""" """Fetches the spec from API Hub and generates the toolset."""
# For each API, get the first version and the first spec of that version. # For each API, get the first version and the first spec of that version.
spec_str = self.apihub_client.get_spec_content(self.apihub_resource_name) spec_str = self._apihub_client.get_spec_content(self._apihub_resource_name)
spec_dict = yaml.safe_load(spec_str) spec_dict = yaml.safe_load(spec_str)
if not spec_dict: if not spec_dict:
return return
@ -177,14 +177,14 @@ class APIHubToolset(BaseToolset):
self.description = self.description or spec_dict.get('info', {}).get( self.description = self.description or spec_dict.get('info', {}).get(
'description', '' 'description', ''
) )
self.openapi_toolset = OpenAPIToolset( self._openapi_toolset = OpenAPIToolset(
spec_dict=spec_dict, spec_dict=spec_dict,
auth_credential=self.auth_credential, auth_credential=self._auth_credential,
auth_scheme=self.auth_scheme, auth_scheme=self._auth_scheme,
tool_filter=self.tool_filter, tool_filter=self.tool_filter,
) )
@override @override
async def close(self): async def close(self):
if self.openapi_toolset: if self._openapi_toolset:
await self.openapi_toolset.close() await self._openapi_toolset.close()