simplifier

landing-ai · Jul 22, 2024 · 6d1ccba · 6d1ccba
1 parent 80e0405
commit 6d1ccba
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -156,6 +156,7 @@ def write_plans(
  tool_desc: str,
  working_memory: str,
  model: LMM,
+ test_multi_plan: bool,
 ) -> Dict[str, Any]:
  chat = copy.deepcopy(chat)
  if chat[-1]["role"] != "user":
@@ -737,14 +738,6 @@ def chat_with_workflow(
  plan_i = plans[k]
  tool_info = tool_infos[k]
 
- if test_multi_plan:
- self.log_progress(
- {
- "type": "plans",
- "status": "completed",
- "payload": tool_lists[best_plan],
- }
- )
  self.log_progress(
  {
  "type": "tools",