NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 04:02:55 -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

28
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,16 +212,18 @@ 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(
name=rest_api_tool.name, IntegrationConnectorTool(
description=rest_api_tool.description, name=rest_api_tool.name,
connection_name=connection_details["name"], description=rest_api_tool.description,
connection_host=connection_details["host"], connection_name=connection_details["name"],
connection_service_name=connection_details["serviceName"], connection_host=connection_details["host"],
entity=entity, connection_service_name=connection_details["serviceName"],
action=action, entity=entity,
operation=operation, action=action,
rest_api_tool=rest_api_tool, operation=operation,
rest_api_tool=rest_api_tool,
)
) )
@override @override
@ -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)
) )