From b0104ffd3fddfb2c705b5aded7227fa102e9976a Mon Sep 17 00:00:00 2001
From: Zhichao <Zhichao.lin@landing.ai>
Date: Thu, 10 Oct 2024 21:53:25 +0800
Subject: [PATCH] remove e2b

---
 vision_agent/utils/execute.py | 39 -----------------------------------
 1 file changed, 39 deletions(-)

diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py
index 38c38150..17b9d3f0 100644
--- a/vision_agent/utils/execute.py
+++ b/vision_agent/utils/execute.py
@@ -13,8 +13,6 @@
 
 import nbformat
 from dotenv import load_dotenv
-from e2b_code_interpreter import Execution as E2BExecution
-from e2b_code_interpreter import Result as E2BResult
 from nbclient import NotebookClient
 from nbclient import __version__ as nbclient_version
 from nbclient.exceptions import CellTimeoutError, DeadKernelError
@@ -200,23 +198,6 @@ def formats(self) -> Iterable[str]:
             formats.extend(iter(self.extra))
         return formats
 
-    @staticmethod
-    def from_e2b_result(result: E2BResult) -> "Result":
-        """
-        Creates a Result object from an E2BResult object.
-        """
-        data = {
-            MimeType.TEXT_PLAIN.value: result.text,
-            MimeType.IMAGE_PNG.value: result.png,
-            MimeType.APPLICATION_JSON.value: result.json,
-        }
-        for k, v in result.extra.items():
-            data[k] = v
-        return Result(
-            is_main_result=result.is_main_result,
-            data=data,
-        )
-
 
 class Logs(BaseModel):
     """Data printed to stdout and stderr during execution, usually by print statements,
@@ -357,26 +338,6 @@ def from_exception(exec: Exception, traceback_raw: List[str]) -> "Execution":
             )
         )
 
-    @staticmethod
-    def from_e2b_execution(exec: E2BExecution) -> "Execution":
-        """Creates an Execution object from an E2BResult object."""
-        return Execution(
-            results=[Result.from_e2b_result(res) for res in exec.results],
-            logs=Logs(stdout=exec.logs.stdout, stderr=exec.logs.stderr),
-            error=(
-                Error(
-                    name=exec.error.name,
-                    value=_remove_escape_and_color_codes(exec.error.value),
-                    traceback_raw=[
-                        _remove_escape_and_color_codes(line)
-                        for line in exec.error.traceback.split("\n")
-                    ],
-                )
-                if exec.error
-                else None
-            ),
-        )
-
 
 class CodeInterpreter(abc.ABC):
     """Code interpreter interface."""