refactor: move e2b code interpreter to vas

vision_agent/agent/vision_agent.py

@@ -197,6 +197,7 @@ def __init__(
  local_artifacts_path: Optional[Union[str, Path]] = None,
  code_sandbox_runtime: Optional[str] = None,
  callback_message: Optional[Callable[[Dict[str, Any]], None]] = None,
+ code_interpreter: Optional[CodeInterpreter] = None,
  ) -> None:
  """Initialize the VisionAgent.
 
@@ -207,12 +208,14 @@ def __init__(
  local_artifacts_path (Optional[Union[str, Path]]): The path to the local
  artifacts file.
  code_sandbox_runtime (Optional[str]): The code sandbox runtime to use.
+ code_interpreter (Optional[CodeInterpreter]): if not None, use this CodeInterpreter
  """
 
  self.agent = AnthropicLMM(temperature=0.0) if agent is None else agent
  self.max_iterations = 12
  self.verbosity = verbosity
  self.code_sandbox_runtime = code_sandbox_runtime
+ self.code_interpreter = code_interpreter
  self.callback_message = callback_message
  if self.verbosity >= 1:
  _LOGGER.setLevel(logging.INFO)
@@ -284,9 +287,14 @@ def chat_with_code(
  # this is setting remote artifacts path
  artifacts = Artifacts(WORKSPACE / "artifacts.pkl")
 
- with CodeInterpreterFactory.new_instance(
- code_sandbox_runtime=self.code_sandbox_runtime,
- ) as code_interpreter:
+ code_interpreter = (
+ self.code_interpreter
+ if self.code_interpreter is not None
+ else CodeInterpreterFactory.new_instance(
+ code_sandbox_runtime=self.code_sandbox_runtime,
+ )
+ )
+ with code_interpreter:
  orig_chat = copy.deepcopy(chat)
  int_chat = copy.deepcopy(chat)
  last_user_message = chat[-1]

vision_agent/tools/meta_tools.py

@@ -28,6 +28,7 @@
 CURRENT_LINE = 0
 DEFAULT_WINDOW_SIZE = 100
 ZMQ_PORT = os.environ.get("ZMQ_PORT", None)
+VERBOSITY = os.environ.get("VERBOSITY", 0)
 
 
 def report_progress_callback(port: int, inp: Dict[str, Any]) -> None:
@@ -375,7 +376,7 @@ def detect_dogs(image_path: str):
  )
  )
  else:
- agent = va.agent.VisionAgentCoder()
+ agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY))
 
  fixed_chat: List[Message] = [{"role": "user", "content": chat, "media": media}]
  response = agent.chat_with_workflow(
@@ -438,7 +439,7 @@ def detect_dogs(image_path: str):
  return dogs
  """
 
- agent = va.agent.VisionAgentCoder()
+ agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY))
  if name not in artifacts:
  print(f"[Artifact {name} does not exist]")
  return f"[Artifact {name} does not exist]"