vision-agent 0.2.150__py3-none-any.whl → 0.2.152__py3-none-any.whl

vision_agent/agent/vision_agent.py

@@ -307,7 +307,7 @@ class VisionAgent(Agent):
             remote_artifacts_path = code_interpreter.upload_file(
                 self.local_artifacts_path
             )
-            artifacts_loaded = artifacts.show()
+            artifacts_loaded = artifacts.show(code_interpreter.remote_path)
             int_chat.append({"role": "observation", "content": artifacts_loaded})
             orig_chat.append({"role": "observation", "content": artifacts_loaded})
             self.streaming_message({"role": "observation", "content": artifacts_loaded})

vision_agent/tools/meta_tools.py

@@ -107,13 +107,16 @@ class Artifacts:
                 with open(self.remote_save_path.parent / k, mode) as f:
                     f.write(v)
 
-    def show(self) -> str:
+    def show(self, uploaded_file_path: Optional[Union[str, Path]] = None) -> str:
         """Shows the artifacts that have been loaded and their remote save paths."""
+        loaded_path = (
+            Path(uploaded_file_path)
+            if uploaded_file_path is not None
+            else self.remote_save_path
+        )
         output_str = "[Artifacts loaded]\n"
         for k in self.artifacts.keys():
-            output_str += (
-                f"Artifact {k} loaded to {str(self.remote_save_path.parent / k)}\n"
-            )
+            output_str += f"Artifact {k} loaded to {str(loaded_path / k)}\n"
         output_str += "[End of artifacts]\n"
         print(output_str)
         return output_str

vision_agent/tools/tools.py

@@ -1603,7 +1603,7 @@ def extract_frames_and_timestamps(
     """
 
     def reformat(
-        frames_and_timestamps: List[Tuple[np.ndarray, float]]
+        frames_and_timestamps: List[Tuple[np.ndarray, float]],
     ) -> List[Dict[str, Union[np.ndarray, float]]]:
         return [
             {"frame": frame, "timestamp": timestamp}
@@ -2017,7 +2017,7 @@ def overlay_counting_results(
         fontsize,
     )
 
-    for i, elt in enumerate(instances):
+    for i, elt in enumerate(instances, 1):
         label = f"{i}"
         box = elt["bbox"]
 

vision_agent/utils/execute.py

@@ -464,7 +464,9 @@ print(f"Vision Agent version: {va_version}")"""
         _LOGGER.info(
             f"E2BCodeInterpreter (sandbox id: {self.interpreter.sandbox_id}) initialized:\n{sys_versions}"
         )
-        self.remote_path = Path(remote_path if remote_path is not None else WORKSPACE)
+        self.remote_path = Path(
+            remote_path if remote_path is not None else "/home/user"
+        )
 
     def close(self, *args: Any, **kwargs: Any) -> None:
         try:
@@ -713,7 +715,7 @@ def _get_e2b_env() -> Union[Dict[str, str], None]:
     openai_api_key = os.getenv("OPENAI_API_KEY", "")
     anthropic_api_key = os.getenv("ANTHROPIC_API_KEY", "")
     if openai_api_key or anthropic_api_key:
-        envs = {}
+        envs = {"WORKSPACE": os.getenv("WORKSPACE", "/home/user")}
         if openai_api_key:
             envs["OPENAI_API_KEY"] = openai_api_key
         if anthropic_api_key:

{vision_agent-0.2.150.dist-info → vision_agent-0.2.152.dist-info}/METADATA

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

{vision_agent-0.2.150.dist-info → vision_agent-0.2.152.dist-info}/RECORD

@@ -2,7 +2,7 @@ vision_agent/__init__.py,sha256=EAb4-f9iyuEYkBrX4ag1syM8Syx8118_t0R6_C34M9w,57
 vision_agent/agent/__init__.py,sha256=NF2LABqHixLvbsOIO-fe-VKZ7awvShLtcT0oQT4eWtI,235
 vision_agent/agent/agent.py,sha256=2cjIOxEuSJrqbfPXYoV0qER5ihXsPFCoEFJa4jpqan0,597
 vision_agent/agent/agent_utils.py,sha256=PEUHqvnHmFL4np_TeFmKMwr5s_dWfdfJz6TF_ogd1dU,2353
-vision_agent/agent/vision_agent.py,sha256=MDXIM5md1V6y62-chyGiDg_138Rns6KKOO2wMqb6vD8,18431
+vision_agent/agent/vision_agent.py,sha256=m7apb1smJbRyj0VAellrN_mDrSPAee4DVm6FWRa-e78,18459
 vision_agent/agent/vision_agent_coder.py,sha256=9BT4gaXsqH5pvxo8WGwJN9MTvP1V3TgoJHBpjtlKP9I,38417
 vision_agent/agent/vision_agent_coder_prompts.py,sha256=BmbTMhth4v1qLexuoSeyo47QQ0kPQvL1pLbCJHMsWDw,18910
 vision_agent/agent/vision_agent_prompts.py,sha256=3n92aF-jpUyyrAy06izdHIMPEMZPKD1JV0wfQvt-PD8,11251
@@ -15,19 +15,19 @@ vision_agent/lmm/__init__.py,sha256=jyY1sJb_tYKg5-Wzs3p1lvwFkc-aUNZfMcLy3TOC4Zg,
 vision_agent/lmm/lmm.py,sha256=B5ClgwvbybVCWkf9opDMLjTtJZemUU4KUkQoRxGh43I,16787
 vision_agent/lmm/types.py,sha256=ZEXR_ptBL0ZwDMTDYkgxUCmSZFmBYPQd2jreNzr_8UY,221
 vision_agent/tools/__init__.py,sha256=zUv3aVPN1MXfyQiQi5To4rkQGtG7mxLQ1NjLI3pxM80,2412
-vision_agent/tools/meta_tools.py,sha256=rudM9heiuTfNjp741ZNcUGRJdpfDZ38BuellA1IPuIo,24747
+vision_agent/tools/meta_tools.py,sha256=yBlkRTeEfI3sAMZbz5mvOsHu9e1OrzDw6XLd6t-U0IY,24909
 vision_agent/tools/prompts.py,sha256=V1z4YJLXZuUl_iZ5rY0M5hHc_2tmMEUKr0WocXKGt4E,1430
 vision_agent/tools/tool_utils.py,sha256=5ukuDMxbEH4iKetYR9I7twzsA8ECyP4tVwYXQq54mxI,8020
-vision_agent/tools/tools.py,sha256=c7SjtZD7YfxhEAGYYe-ExVCBA4NDXmRwerBIbd-XEH8,74557
+vision_agent/tools/tools.py,sha256=CCeXdHDguZh8j7Iibi-BzfPxOakPSGnFAeVUZIHekno,74561
 vision_agent/tools/tools_types.py,sha256=JUOZWGW2q-dlJ85CHr9gvo9KQk_rXyjJhi-iwPNn4eM,2397
 vision_agent/utils/__init__.py,sha256=7fMgbZiEwbNS0fBOS_hJI5PuEYBblw36zLi_UjUzvj4,244
 vision_agent/utils/exceptions.py,sha256=booSPSuoULF7OXRr_YbC4dtKt6gM_HyiFQHBuaW86C4,2052
-vision_agent/utils/execute.py,sha256=WX4bYdmJGznnaxEOAKLc7cswUJNPZZCil1OP2xFzpRU,27943
+vision_agent/utils/execute.py,sha256=FqSOr5gtBeKB1g2hbV6-bhox6qItDQNn2o9efq1w6f4,28017
 vision_agent/utils/image_utils.py,sha256=rm9GfXvD4JrjnqKrP_f2gfq4SzmqYC0IdC1kKwdn6sk,11303
 vision_agent/utils/sim.py,sha256=ZuSS07TUXFGjipmiQoY8TKRmSes7XXCdtU9PI8PC1sw,5609
 vision_agent/utils/type_defs.py,sha256=BE12s3JNQy36QvauXHjwyeffVh5enfcvd4vTzSwvEZI,1384
 vision_agent/utils/video.py,sha256=xbMEoRk13l4fHeQlbvMQhLCn8RNndYmsDhUf01TUeR8,4781
-vision_agent-0.2.150.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vision_agent-0.2.150.dist-info/METADATA,sha256=zcfgNjECU5R-iqxWFi7uZlXpypPvuX1Sars7EJ6URyA,13758
-vision_agent-0.2.150.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-vision_agent-0.2.150.dist-info/RECORD,,
+vision_agent-0.2.152.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vision_agent-0.2.152.dist-info/METADATA,sha256=OH7wZSXkviBN5xALMJ7ysqCVUzX1dFG_91CHp8NT8jc,13758
+vision_agent-0.2.152.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+vision_agent-0.2.152.dist-info/RECORD,,