vision-agent 0.2.136__tar.gz → 0.2.138__tar.gz

{vision_agent-0.2.136 → vision_agent-0.2.138}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vision-agent
-Version: 0.2.136
+Version: 0.2.138
 Summary: Toolset for Vision Agent
 Author: Landing AI
 Author-email: dev@landing.ai
@@ -13,6 +13,7 @@ Requires-Dist: anthropic (>=0.31.0,<0.32.0)
 Requires-Dist: av (>=11.0.0,<12.0.0)
 Requires-Dist: e2b (>=0.17.2a50,<0.18.0)
 Requires-Dist: e2b-code-interpreter (==0.0.11a37)
+Requires-Dist: flake8 (>=7.0.0,<8.0.0)
 Requires-Dist: ipykernel (>=6.29.4,<7.0.0)
 Requires-Dist: langsmith (>=0.1.58,<0.2.0)
 Requires-Dist: nbclient (>=0.10.0,<0.11.0)

{vision_agent-0.2.136 → vision_agent-0.2.138}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.2.136"
+version = "0.2.138"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <dev@landing.ai>"]
 readme = "README.md"
@@ -24,6 +24,7 @@ requests = "2.*"
 tqdm = ">=4.64.0,<5.0.0"
 pandas = "2.*"
 openai = "1.*"
+flake8 = "^7.0.0"
 typing_extensions = "4.*"
 opencv-python = "4.*"
 tabulate = "^0.9.0"
@@ -47,7 +48,6 @@ av = "^11.0.0"
 autoflake = "1.*"
 pytest = "7.*"
 black = ">=23,<25"
-flake8 = "5.*"
 isort = "5.*"
 responses = "^0.23.1"
 mypy = "<1.8.0"

{vision_agent-0.2.136 → vision_agent-0.2.138}/vision_agent/agent/vision_agent.py

@@ -276,8 +276,11 @@ class VisionAgent(Agent):
                 # sometimes it gets stuck in a loop, so we force it to exit
                 if last_response == response:
                     response["let_user_respond"] = True
-
-                self.streaming_message({"role": "assistant", "content": response})
+                    self.streaming_message(
+                        {"role": "assistant", "error": "Stuck in loop"}
+                    )
+                else:
+                    self.streaming_message({"role": "assistant", "content": response})
 
                 if response["let_user_respond"]:
                     break

{vision_agent-0.2.136 → vision_agent-0.2.138}/vision_agent/tools/meta_tools.py

@@ -214,7 +214,11 @@ def create_code_artifact(artifacts: Artifacts, name: str) -> str:
     display(
         {
             MimeType.APPLICATION_ARTIFACT: json.dumps(
-                {"name": name, "content": artifacts[name]}
+                {
+                    "name": name,
+                    "content": artifacts[name],
+                    "action": "create",
+                }
             )
         },
         raw=True,
@@ -305,7 +309,11 @@ def edit_code_artifact(
     display(
         {
             MimeType.APPLICATION_ARTIFACT: json.dumps(
-                {"name": name, "content": artifacts[name]}
+                {
+                    "name": name,
+                    "content": artifacts[name],
+                    "action": "edit",
+                }
             )
         },
         raw=True,
@@ -366,7 +374,16 @@ def generate_vision_code(
     total_lines = len(code_lines)
 
     display(
-        {MimeType.APPLICATION_ARTIFACT: json.dumps({"name": name, "content": code})},
+        {
+            MimeType.APPLICATION_ARTIFACT: json.dumps(
+                {
+                    "name": name,
+                    "content": code,
+                    "contentType": "vision_code",
+                    "action": "create",
+                }
+            )
+        },
         raw=True,
     )
     return view_lines(code_lines, 0, total_lines, name, total_lines)
@@ -434,7 +451,15 @@ def edit_vision_code(
     total_lines = len(code_lines)
 
     display(
-        {MimeType.APPLICATION_ARTIFACT: json.dumps({"name": name, "content": code})},
+        {
+            MimeType.APPLICATION_ARTIFACT: json.dumps(
+                {
+                    "name": name,
+                    "content": code,
+                    "action": "edit",
+                }
+            )
+        },
         raw=True,
     )
     return view_lines(code_lines, 0, total_lines, name, total_lines)

{vision_agent-0.2.136 → vision_agent-0.2.138}/vision_agent/utils/execute.py

@@ -104,7 +104,7 @@ class Result:
     latex: Optional[str] = None
     json: Optional[Dict[str, Any]] = None
     javascript: Optional[str] = None
-    artifact_name: Optional[str] = None
+    artifact: Optional[str] = None
     extra: Optional[Dict[str, Any]] = None
     "Extra data that can be included. Not part of the standard types."