vision-agent 0.2.157__tar.gz → 0.2.158__tar.gz

{vision_agent-0.2.157 → vision_agent-0.2.158}/PKG-INFO

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

{vision_agent-0.2.157 → vision_agent-0.2.158}/pyproject.toml

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

{vision_agent-0.2.157 → vision_agent-0.2.158}/vision_agent/agent/vision_agent.py

@@ -383,7 +383,7 @@ class VisionAgent(Agent):
                     obs_chat_elt: Message = {"role": "observation", "content": obs}
                     if media_obs and result.success:
                         obs_chat_elt["media"] = [
-                            Path(code_interpreter.remote_path) / media_ob
+                            Path(self.local_artifacts_path).parent / media_ob
                             for media_ob in media_obs
                         ]
 
@@ -407,6 +407,8 @@ class VisionAgent(Agent):
             code_interpreter.download_file(
                 str(remote_artifacts_path.name), str(self.local_artifacts_path)
             )
+            artifacts.load(self.local_artifacts_path)
+            artifacts.save()
         return orig_chat, artifacts
 
     def streaming_message(self, message: Dict[str, Any]) -> None: