From 524505d82efa241773a4d2b3b33ba10b2996667a Mon Sep 17 00:00:00 2001 From: Paolo Dettori Date: Tue, 8 Apr 2025 10:31:11 -0400 Subject: [PATCH] handle case where 'data' is not an attribute in unregister_toolgroup Signed-off-by: Paolo Dettori --- llama_stack/distribution/routers/routing_tables.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index 529b5ccbf..f6adae49d 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -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)