log retrieved functions

landing-ai · May 15, 2024 · 16a9285 · 16a9285
1 parent c8d3d9d
commit 16a9285
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py
@@ -235,9 +235,11 @@ def run_plan(
  f"""
 {tabulate(tabular_data=[task], headers="keys", tablefmt="mixed_grid", maxcolwidths=_MAX_TABULATE_COL_WIDTH)}"""
  )
- tool_info = "\n".join(
- [e["doc"] for e in tool_recommender.top_k(task["instruction"])]
- )
+ tools = tool_recommender.top_k(task["instruction"])
+ tool_info = "\n".join([e["doc"] for e in tools])
+
+ if verbosity == 2:
+ _LOGGER.info(f"Tools retrieved: {[e['desc'] for e in tools]}")
 
  if long_term_memory is not None:
  retrieved_ltm = "\n".join(