initial commit

Author: mishushakov

Makefile

@@ -1,5 +1,5 @@
 start-template-server:
-	docker run --rm -e E2B_LOCAL=true -p 49999:49999 -it $$(docker build . -q -f ./template/test.Dockerfile)
+	docker run --rm -e E2B_LOCAL=true -p 49999:49999 -it $$(python template/build_docker.py | docker build -q ./template -f -) 
 
 kill-template-server:
 	docker kill $(shell docker ps --filter expose=49999 --format {{.ID}})

template/Dockerfile

@@ -0,0 +1,5 @@
+from template import make_template
+from e2b import Template
+
+tmp = make_template(kernels=["python", "javascript"])
+print(Template.to_dockerfile(tmp))

template/build_docker.py

@@ -0,0 +1,13 @@
+from dotenv import load_dotenv
+from e2b_template import Template
+from template import make_template
+
+load_dotenv()
+
+Template.build(
+    make_template(),
+    alias="code-interpreter",
+    cpu_count=1,
+    memory_mb=1024,
+    on_build_logs=lambda log_entry: print(log_entry),
+)

template/build_prod.py

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

template/build_test.py

@@ -0,0 +1 @@
+e2b==2.6.0

template/e2b.Dockerfile

@@ -13,10 +13,10 @@ function start_jupyter_server() {
 		response=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:8888/api/status")
 	done
 
-	cd /root/.server/
-	/root/.server/.venv/bin/uvicorn main:app --host 0.0.0.0 --port 49999 --workers 1 --no-access-log --no-use-colors --timeout-keep-alive 640
+	cd .server/
+	.venv/bin/uvicorn main:app --host 0.0.0.0 --port 49999 --workers 1 --no-access-log --no-use-colors --timeout-keep-alive 640
 }
 
 echo "Starting Code Interpreter server..."
 start_jupyter_server &
-MATPLOTLIBRC=/root/.config/matplotlib/.matplotlibrc jupyter server --IdentityProvider.token="" >/dev/null 2>&1
+MATPLOTLIBRC=.config/matplotlib/.matplotlibrc jupyter server --IdentityProvider.token="" >/dev/null 2>&1

template/e2b.toml

@@ -0,0 +1,124 @@
+from e2b import Template, wait_for_port
+
+
+def make_template(
+    kernels: list[str] = ["python", "r", "javascript", "deno", "bash", "java"],
+):
+    # Start with base template
+    template = (
+        Template()
+        .from_image("python:3.12")
+        .set_envs(
+            {
+                "PIP_DEFAULT_TIMEOUT": "100",
+                "PIP_DISABLE_PIP_VERSION_CHECK": "1",
+                "PIP_NO_CACHE_DIR": "1",
+                "JUPYTER_CONFIG_PATH": ".jupyter",
+                "IPYTHON_CONFIG_PATH": ".ipython",
+                "SERVER_PATH": ".server",
+                "R_VERSION": "4.4.2",
+                "R_HOME": "/opt/R/4.4.2",
+                "JAVA_HOME": "/opt/java/openjdk",
+                "DENO_INSTALL": "$HOME/.deno",
+            }
+        )
+        .apt_install(
+            [
+                "build-essential",
+                "curl",
+                "git",
+                "util-linux",
+                "jq",
+                "sudo",
+                "fonts-noto-cjk",
+            ]
+        )
+        .run_cmd("curl -fsSL https://deb.nodesource.com/setup_20.x | bash -")
+        .apt_install("nodejs")
+        .copy("requirements.txt", "requirements.txt")
+        .pip_install("--no-cache-dir -r requirements.txt")
+    )
+
+    if "python" in kernels:
+        template = template.run_cmd("ipython kernel install --name 'python3' --user")
+
+    # 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")
+            .run_cmd(
+                [
+                    "R -e \"install.packages('IRkernel', repos='https://cloud.r-project.org')\"",
+                    "R -e \"IRkernel::installspec(user = FALSE, name = 'r', displayname = 'R')\"",
+                ]
+            )
+        )
+
+    # Install JavaScript Kernel if requested
+    if "javascript" in kernels:
+        template = template.npm_install(
+            "--unsafe-perm git+https://github.com/e2b-dev/ijavascript.git",
+            g=True,
+        ).run_cmd("ijsinstall --install=global")
+
+    # Install Deno Kernel if requested
+    if "deno" in kernels:
+        template = template.run_cmd(
+            [
+                "curl -fsSL https://deno.land/x/install/install.sh | sh",
+                "PATH=$HOME/.deno/bin:$PATH",
+                "deno jupyter --unstable --install",
+            ]
+        )
+        template = template.copy(
+            "deno.json", ".local/share/jupyter/kernels/deno/kernel.json"
+        )
+
+    # Install Bash Kernel if requested
+    if "bash" in kernels:
+        template = template.pip_install("bash_kernel").run_cmd(
+            "python -m bash_kernel.install"
+        )
+
+    # Install Java and Java Kernel if requested
+    if "java" in kernels:
+        template = template.apt_install("default-jdk")
+        template = template.run_cmd(
+            [
+                "wget https://github.com/SpencerPark/IJava/releases/download/v1.3.0/ijava-1.3.0.zip",
+                "unzip ijava-1.3.0.zip",
+                "python install.py --sys-prefix",
+            ]
+        )
+
+    # Common setup steps (always run)
+    template = (
+        template
+        # Create server virtual environment
+        .copy("server", ".server")
+        .run_cmd(
+            [
+                "python -m venv .server/.venv",
+                ". .server/.venv/bin/activate",
+            ]
+        )
+        # Copy and install server requirements
+        .run_cmd(
+            ".server/.venv/bin/pip install --no-cache-dir -r .server/requirements.txt"
+        )
+        # Copy configuration files
+        .copy("matplotlibrc", ".config/matplotlib/.matplotlibrc")
+        .copy("start-up.sh", ".jupyter/start-up.sh", mode=0o755, user="root")
+        .run_cmd("chmod +x .jupyter/start-up.sh")
+        .copy("jupyter_server_config.py", ".jupyter/")
+        .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