From 04bd7686d752baea5293cd6c72d0d1d2cb260244 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 29 Aug 2024 10:17:03 -0700 Subject: [PATCH] make remote path execute variable --- vision_agent/utils/execute.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py index b2da6f11..37c8d260 100644 --- a/vision_agent/utils/execute.py +++ b/vision_agent/utils/execute.py @@ -384,8 +384,15 @@ def from_e2b_execution(exec: E2BExecution) -> "Execution": class CodeInterpreter(abc.ABC): """Code interpreter interface.""" - def __init__(self, timeout: int, *args: Any, **kwargs: Any) -> None: + def __init__( + self, + timeout: int, + remote_path: Optional[Union[str, Path]] = None, + *args: Any, + **kwargs: Any, + ) -> None: self.timeout = timeout + self.remote_path = Path(remote_path if remote_path is not None else WORKSPACE) def __enter__(self) -> Self: return self