From a4b6539184f13633f00f056c3e8be7cd76f8f4f2 Mon Sep 17 00:00:00 2001
From: wuyiqunLu <132986242+wuyiqunLu@users.noreply.github.com>
Date: Thu, 30 May 2024 07:43:49 -0500
Subject: [PATCH] feat: log more tools info back to client (#101)

* feat: log more tools info back to client

* fix lint
---
 vision_agent/agent/vision_agent.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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: