Desktop Template E2B SDK

.github/workflows/template.yml

@@ -3,8 +3,8 @@ name: Build and push Desktop template
 on:
   push:
     paths:
-      - "template/**"
-      - ".github/workflows/template.yml"
+      - 'template/**'
+      - '.github/workflows/template.yml'
     branches:
       - main
 
@@ -23,6 +23,22 @@ jobs:
       - name: Checkout repository
         uses: actions/checkout@v3
 
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.12'
+
+      - name: Install and configure Poetry
+        uses: snok/install-poetry@v1
+        with:
+          version: 2.1.1
+          virtualenvs-create: true
+          virtualenvs-in-project: true
+          installer-parallel: true
+
+      - name: Install dependencies
+        run: poetry install
+
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
@@ -35,16 +51,12 @@ jobs:
       - name: Build and push to DockerHub
         run: |
           docker pull ${{ secrets.DOCKERHUB_USERNAME }}/desktop:latest || true
-          docker buildx build \
-            --file Dockerfile \
+          poetry run python build_docker.py | docker buildx build \
             --platform linux/amd64 \
             --push \
-            --tag ${{ secrets.DOCKERHUB_USERNAME }}/desktop:latest .
-
-      - name: Install E2B CLI
-        run: npm install -g @e2b/cli
+            --tag ${{ secrets.DOCKERHUB_USERNAME }}/desktop:latest -f - files
 
-      - name: Build e2b
-        run: e2b template build
+      - name: Build E2B template
+        run: poetry run python build_prod.py
         env:
-          E2B_ACCESS_TOKEN: ${{ secrets.E2B_ACCESS_TOKEN }}
+          E2B_API_KEY: ${{ secrets.E2B_API_KEY }}

.gitignore

@@ -295,3 +295,5 @@ cython_debug/
 
 # SDK reference artifacts
 sdk_ref/
+
+.ruff_cache/

template/.python-version

@@ -0,0 +1 @@
+3.12

template/README.md

@@ -0,0 +1,15 @@
+# E2B Desktop Template
+
+This is the template for the E2B Desktop Sandbox.
+
+## Building the template
+
+```bash
+poetry run python build_dev.py
+```
+
+## Building the production image
+
+```bash
+poetry run python build_prod.py
+```

template/build_dev.py

@@ -0,0 +1,13 @@
+from dotenv import load_dotenv
+from e2b import Template, default_build_logger
+from template import template_with_user_workdir
+
+load_dotenv()
+
+Template.build(
+    template=template_with_user_workdir,
+    alias="desktop-dev",
+    cpu_count=8,
+    memory_mb=8192,
+    on_build_logs=default_build_logger(),
+)

template/build_docker.py

@@ -0,0 +1,5 @@
+from e2b import Template
+from template import template
+
+# output the template to stdout to pipe into docker buildx
+print(Template.to_dockerfile(template))

template/build_prod.py

@@ -0,0 +1,13 @@
+from dotenv import load_dotenv
+from e2b import Template, default_build_logger
+from template import template_with_user_workdir
+
+load_dotenv()
+
+Template.build(
+    template_with_user_workdir,
+    alias="desktop",
+    cpu_count=8,
+    memory_mb=8192,
+    on_build_logs=default_build_logger(),
+)