updated

Author: mishushakov

template/build_prod.py

@@ -1,13 +1,13 @@
 from dotenv import load_dotenv
-from e2b import Template
+from e2b import Template, default_build_logger
 from template import make_template
 
 load_dotenv()
 
 Template.build(
-    make_template(),
+    make_template(set_user_workdir=True),
     alias="code-interpreter",
-    cpu_count=1,
-    memory_mb=1024,
-    on_build_logs=lambda log_entry: print(log_entry),
+    cpu_count=2,
+    memory_mb=2048,
+    on_build_logs=default_build_logger(),
 )

template/build_test.py

@@ -1,13 +1,13 @@
 from dotenv import load_dotenv
-from e2b import Template
+from e2b import Template, default_build_logger
 from template import make_template
 
 load_dotenv()
 
 Template.build(
-    make_template(kernels=["python", "javascript"]),
+    make_template(kernels=["python", "javascript"], set_user_workdir=True),
     alias="code-interpreter-dev",
     cpu_count=1,
     memory_mb=1024,
-    on_build_logs=lambda log_entry: print(log_entry),
+    on_build_logs=default_build_logger(min_level="debug"),
 )

template/template.py

@@ -1,13 +1,16 @@
-from e2b import Template, wait_for_port
+from e2b import Template, wait_for_port, wait_for_url
 
 
 def make_template(
     kernels: list[str] = ["python", "r", "javascript", "deno", "bash", "java"],
+    set_user_workdir: bool = False,
 ):
     # Start with base template
     template = (
         Template()
         .from_image("python:3.12")
+        .set_user("root")
+        .set_workdir("/")
         .set_envs(
             {
                 "PIP_DEFAULT_TIMEOUT": "100",
@@ -19,7 +22,7 @@ def make_template(
                 "R_VERSION": "4.4.2",
                 "R_HOME": "/opt/R/4.4.2",
                 "JAVA_HOME": "/opt/java/openjdk",
-                "DENO_INSTALL": "$HOME/.deno",
+                "DENO_INSTALL": "/opt/deno",
             }
         )
         .apt_install(
@@ -45,11 +48,7 @@ def make_template(
     # Install R Kernel if requested
     if "r" in kernels:
         template = (
-            template.run_cmd(
-                "curl -O https://cdn.rstudio.com/r/debian-12/pkgs/r-4.4.2_1_amd64.deb"
-            )
-            .apt_install("./r-4.4.2_1_amd64.deb")
-            .make_symlink("/opt/R/4.4.2/bin/R", "/usr/bin/R")
+            template.apt_install("r-base")
             .run_cmd(
                 [
                     "R -e \"install.packages('IRkernel', repos='https://cloud.r-project.org')\"",
@@ -70,7 +69,7 @@ def make_template(
         template = template.run_cmd(
             [
                 "curl -fsSL https://deno.land/x/install/install.sh | sh",
-                "PATH=$HOME/.deno/bin:$PATH",
+                "PATH=/opt/deno/bin:$PATH",
                 "deno jupyter --unstable --install",
             ]
         )
@@ -113,7 +112,9 @@ def make_template(
         .make_dir(".ipython/profile_default/startup")
         .copy("ipython_kernel_config.py", ".ipython/profile_default/")
         .copy("startup_scripts", ".ipython/profile_default/startup")
-        .set_start_cmd(".jupyter/start-up.sh", wait_for_port(49999))
     )
 
-    return template
+    if set_user_workdir:
+        template = template.set_user("user").set_workdir("/home/user")
+
+    return template.set_start_cmd(".jupyter/start-up.sh", wait_for_url("http://localhost:49999/health"))