handle case where 'data' is not an attribute in unregister_toolgroup

Signed-off-by: Paolo Dettori <dettori@us.ibm.com>
This commit is contained in:
Paolo Dettori 2025-04-08 10:31:11 -04:00
parent 0c3f9f46f5
commit 524505d82e

View file

@ -609,7 +609,7 @@ class ToolGroupsRoutingTable(CommonRoutingTableImpl, ToolGroups):
if tool_group is None:
raise ValueError(f"Tool group {toolgroup_id} not found")
tools = await self.list_tools(toolgroup_id)
for tool in tools.data:
for tool in getattr(tools, "data", []):
await self.unregister_object(tool)
await self.unregister_object(tool_group)