NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -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."
)
self._openapi_toolset = None
self._tool = None
self._tools = []
self._parse_spec_to_toolset(spec, 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)
if auth_credential:
rest_api_tool.configure_auth_credential(auth_credential)
self._tool = IntegrationConnectorTool(
name=rest_api_tool.name,
description=rest_api_tool.description,
connection_name=connection_details["name"],
connection_host=connection_details["host"],
connection_service_name=connection_details["serviceName"],
entity=entity,
action=action,
operation=operation,
rest_api_tool=rest_api_tool,
self._tools.append(
IntegrationConnectorTool(
name=rest_api_tool.name,
description=rest_api_tool.description,
connection_name=connection_details["name"],
connection_host=connection_details["host"],
connection_service_name=connection_details["serviceName"],
entity=entity,
action=action,
operation=operation,
rest_api_tool=rest_api_tool,
)
)
@override
@ -230,8 +232,8 @@ class ApplicationIntegrationToolset(BaseToolset):
readonly_context: Optional[ReadonlyContext] = None,
) -> List[RestApiTool]:
return (
[self._tool]
if self._tool
self._tools
if self._openapi_toolset is None
else await self._openapi_toolset.get_tools(readonly_context)
)