vision-agent 0.2.149__tar.gz → 0.2.151__tar.gz

{vision_agent-0.2.149 → vision_agent-0.2.151}/PKG-INFO

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

{vision_agent-0.2.149 → vision_agent-0.2.151}/pyproject.toml

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

{vision_agent-0.2.149 → vision_agent-0.2.151}/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-0.2.149 → vision_agent-0.2.151}/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-0.2.149 → vision_agent-0.2.151}/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.149 → vision_agent-0.2.151}/vision_agent/utils/sim.py

@@ -125,7 +125,9 @@ class AzureSim(Sim):
             raise ValueError("key is required if no column 'embs' is present.")
 
         if sim_key is not None:
-            self.df["embs"] = self.df[sim_key].apply(lambda x: get_embedding(client, x))
+            self.df["embs"] = self.df[sim_key].apply(
+                lambda x: get_embedding(self.emb_call, x)
+            )
 
 
 class OllamaSim(Sim):