feat: log more tools info back to client (#101)

* feat: log more tools info back to client * fix lint
landing-ai · May 30, 2024 · a4b6539 · a4b6539
1 parent 93c365d
commit a4b6539
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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: