show artifacts

vision_agent/agent/vision_agent.py

@@ -307,7 +307,7 @@ def chat_with_code(
  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 @@ def load(self, file_path: Union[str, Path]) -> None:
  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