From a9c4f570c42d9bdaf4174bab052f80778133d7f0 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Sun, 24 Mar 2024 20:38:46 -0700 Subject: [PATCH] fix issues with easytool --- vision_agent/agent/easytool.py | 2 +- vision_agent/agent/easytool_prompts.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/vision_agent/agent/easytool.py b/vision_agent/agent/easytool.py index 7a7cf7d1..63aad9ba 100644 --- a/vision_agent/agent/easytool.py +++ b/vision_agent/agent/easytool.py @@ -45,7 +45,7 @@ def format_tools(tools: Dict[int, Any]) -> str: # Format this way so it's clear what the ID's are tool_str = "" for key in tools: - tool_str += f"ID: {key}, {tools[key]}\n" + tool_str += f"ID: {key} - {tools[key]}\n" return tool_str diff --git a/vision_agent/agent/easytool_prompts.py b/vision_agent/agent/easytool_prompts.py index acc0a111..445e42b3 100644 --- a/vision_agent/agent/easytool_prompts.py +++ b/vision_agent/agent/easytool_prompts.py @@ -14,7 +14,7 @@ Output: """ TASK_TOPOLOGY = """Given a complex user's question, I have decompose this question into some simple subtasks. I think there exists a logical connections and order amontg the tasks. Thus you need to help me output this logical connections and order. -You must ONLY output in a parsible JSON format with the following format:" +You must ONLY output in a parsible JSON format with the following format: {{"Tasks": [{{"task": task, "id", task_id, "dep": [dependency_task_id1, dependency_task_id2, ...]}}]}} @@ -31,7 +31,6 @@ CHOOSE_TOOL = """This is the user's question: {question} These are the tools you can select to solve the question: -Tool List: {tools} Please note that: