diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 1fef0852..59d5b519 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -264,15 +264,19 @@ def retrieve_tools( ) tool_info = [] tool_desc = [] + tool_list: List[Dict[str, str]] = [] for task in plan: tools = tool_recommender.top_k(task["instructions"], k=2, thresh=0.3) tool_info.extend([e["doc"] for e in tools]) tool_desc.extend([e["desc"] for e in tools]) + tool_list.extend( + {"description": e["desc"], "documentation": e["doc"]} for e in tools + ) log_progress( { "type": "tools", "status": "completed", - "payload": tools, + "payload": tool_list, } ) if verbosity == 2: