NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 20:22:56 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix application integration toolset

Browse Source 
PiperOrigin-RevId: 759450502
This commit is contained in:
Xiang (Sean) Zhou 2025-05-15 22:23:13 -07:00 committed by Copybara-Service
parent fe5a5494e9
commit eb076a9ee8
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/google/adk/tools/application_integration_tool/application_integration_toolset.py
View File

@ -162,7 +162,7 @@ class ApplicationIntegrationToolset(BaseToolset):
" (entity_operations or actions)) should be provided." " (entity_operations or actions)) should be provided."
) )
self._openapi_toolset = None self._openapi_toolset = None
self._tool = None self._tools = []
self._parse_spec_to_toolset(spec, connection_details) self._parse_spec_to_toolset(spec, connection_details)
def _parse_spec_to_toolset(self, spec_dict, connection_details): def _parse_spec_to_toolset(self, spec_dict, connection_details):
@ -212,7 +212,8 @@ class ApplicationIntegrationToolset(BaseToolset):
rest_api_tool.configure_auth_scheme(auth_scheme) rest_api_tool.configure_auth_scheme(auth_scheme)
if auth_credential: if auth_credential:
rest_api_tool.configure_auth_credential(auth_credential) rest_api_tool.configure_auth_credential(auth_credential)
self._tool = IntegrationConnectorTool( self._tools.append(
IntegrationConnectorTool(
name=rest_api_tool.name, name=rest_api_tool.name,
description=rest_api_tool.description, description=rest_api_tool.description,
connection_name=connection_details["name"], connection_name=connection_details["name"],
@ -223,6 +224,7 @@ class ApplicationIntegrationToolset(BaseToolset):
operation=operation, operation=operation,
rest_api_tool=rest_api_tool, rest_api_tool=rest_api_tool,
) )
)
@override @override
async def get_tools( async def get_tools(
@ -230,8 +232,8 @@ class ApplicationIntegrationToolset(BaseToolset):
readonly_context: Optional[ReadonlyContext] = None, readonly_context: Optional[ReadonlyContext] = None,
) -> List[RestApiTool]: ) -> List[RestApiTool]:
return ( return (
[self._tool] self._tools
if self._tool if self._openapi_toolset is None
else await self._openapi_toolset.get_tools(readonly_context) else await self._openapi_toolset.get_tools(readonly_context)
) )