vision-agent 0.2.29__tar.gz → 0.2.30__tar.gz

{vision_agent-0.2.29 → vision_agent-0.2.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.29
+Version: 0.2.30
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai

{vision_agent-0.2.29 → vision_agent-0.2.30}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.29"
+version = "0.2.30"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"

{vision_agent-0.2.29 → vision_agent-0.2.30}/vision_agent/agent/agent.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 from pathlib import Path
-from typing import Dict, List, Optional, Union, Any
+from typing import Any, Dict, List, Optional, Union
 
 
 class Agent(ABC):

{vision_agent-0.2.29 → vision_agent-0.2.30}/vision_agent/agent/agent_coder.py

@@ -3,7 +3,7 @@ import logging
 import os
 import sys
 from pathlib import Path
-from typing import Dict, List, Optional, Union, Any
+from typing import Any, Dict, List, Optional, Union
 
 from rich.console import Console
 from rich.syntax import Syntax

{vision_agent-0.2.29 → vision_agent-0.2.30}/vision_agent/agent/vision_agent_v3.py

@@ -3,7 +3,7 @@ import json
 import logging
 import sys
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Union, cast, Callable, no_type_check
+from typing import Any, Callable, Dict, List, Optional, Union, cast, no_type_check
 
 from rich.console import Console
 from rich.syntax import Syntax
@@ -14,6 +14,7 @@ from vision_agent.agent.vision_agent_v3_prompts import (
     CODE,
     FEEDBACK,
     FIX_BUG,
+    FULL_TASK,
     PLAN,
     REFLECT,
     SIMPLE_TEST,
@@ -332,7 +333,7 @@ class VisionAgentV3(Agent):
                 self.verbosity,
             )
             results = write_and_test_code(
-                plan_i_str,
+                FULL_TASK.format(user_request=chat[0]["content"], subtasks=plan_i_str),
                 tool_info,
                 UTILITIES_DOCSTRING,
                 format_memory(working_memory),
@@ -350,7 +351,14 @@ class VisionAgentV3(Agent):
             plan.append({"code": code, "test": test, "plan": plan_i})
 
             if self_reflection:
-                reflection = reflect(chat, plan_i_str, code, self.planner)
+                reflection = reflect(
+                    chat,
+                    FULL_TASK.format(
+                        user_request=chat[0]["content"], subtasks=plan_i_str
+                    ),
+                    code,
+                    self.planner,
+                )
                 if self.verbosity > 0:
                     self.log_progress(
                         {

{vision_agent-0.2.29 → vision_agent-0.2.30}/vision_agent/agent/vision_agent_v3_prompts.py

@@ -3,6 +3,14 @@ USER_REQ = """
 {user_request}
 """
 
+FULL_TASK = """
+## User Request
+{user_request}
+
+## Subtasks
+{subtasks}
+"""
+
 FEEDBACK = """
 ## This contains code and feedback from previous runs and is used for providing context so you do not make the same mistake again.
 
@@ -170,7 +178,7 @@ This is the documentation for the functions you have access to. You may call any
 
 
 FIX_BUG = """
-**Role** As a coder, your job is to find the error in the code and fix it. You are running in a notebook setting so feel free to run !pip install to install missing packages.
+**Role** As a coder, your job is to find the error in the code and fix it. You are running in a notebook setting so you can run !pip install to install missing packages.
 
 **Instructions**:
 Please re-complete the code to fix the error message. Here is the previous version: