use complete env variable key in template

Author: marwan37

omni-reader/Dockerfile.sandbox

@@ -1,44 +1,47 @@
 # Sandbox base image
-FROM safoinext/zenml-sandbox:latest
+FROM zenmldocker/zenml-sandbox:latest
+
+# Project metadata
+LABEL project_name="omni-reader"
+LABEL project_version="0.1.0"
 
 # Install project-specific dependencies
-# Install polars-lts-cpu instead of polars (version compiled for CPU compatibility)
 RUN pip install --no-cache-dir \
-    "instructor==1.7.7" \
-    "jiwer==3.0.5" \
-    "jiter==0.8.2" \
+    "instructor" \
+    "jiwer" \
+    "jiter" \
     "importlib-metadata<7.0,>=1.4.0" \
-    "litellm==1.64.1" \
+    "litellm" \
     "mistralai==1.0.3" \
     "numpy<2.0,>=1.9.0" \
     "openai==1.69.0" \
     "Pillow==11.1.0" \
     "polars-lts-cpu==1.26.0" \
     "pyarrow>=7.0.0" \
-    "python-dotenv==1.0.1" \
+    "python-dotenv" \
     "streamlit==1.44.0" \
     "pydantic>=2.8.2,<2.9.0" \
     "tqdm==4.66.4" \
-    "zenml>=0.80.0" \
-    uv
+    "zenml>=0.80.0"
 
 # Set workspace directory
 WORKDIR /workspace
 
-# Clone only the omni-reader directory and reorganize
+# Clone only the project directory and reorganize
 RUN git clone --depth 1 https://github.com/zenml-io/zenml-projects.git /tmp/zenml-projects && \
     cp -r /tmp/zenml-projects/omni-reader/* /workspace/ && \
     rm -rf /tmp/zenml-projects
 
 # Create a template .env file for API keys
-RUN echo "OPENAI_API_KEY=YOUR_OPENAI_API_KEY_HERE" > .env && \
-    echo "MISTRAL_API_KEY=YOUR_MISTRAL_API_KEY_HERE" >> .env
+RUN echo "OPENAI_API_KEY=YOUR_OPENAI_API_KEY" && \
+    echo "MISTRAL_API_KEY=YOUR_MISTRAL_API_KEY" > .env
 
-# Create a .vscode directory (mainly to auto-apply the dark theme)
-RUN mkdir -p /workspace/.vscode
-# Copy settings file
-COPY settings.json /workspace/.vscode/settings.json
+# Create a .vscode directory and settings.json file
+RUN mkdir -p /workspace/.vscode && \
+    echo '{\n'\
+    '  "workbench.colorTheme": "Default Dark Modern"\n'\
+    '}' > /workspace/.vscode/settings.json
 
-# Set environment variable to skip CPU checks for Polars as a fallback
-ENV POLARS_SKIP_CPU_CHECK=1
 
+# Set environment variables for compatibility and performance
+ENV POLARS_SKIP_CPU_CHECK=1

oncoclear/Dockerfile.sandbox

@@ -1,5 +1,5 @@
 # Sandbox base image
-FROM safoinext/zenml-sandbox:latest
+FROM zenmldocker/zenml-sandbox:latest
 
 # Project metadata
 LABEL project_name="oncoclear"
@@ -22,9 +22,9 @@ RUN git clone --depth 1 https://github.com/zenml-io/zenml-projects.git /tmp/zenm
     rm -rf /tmp/zenml-projects
 
 # Create a template .env file for API keys
-RUN echo "ZENML_PROJECT_SECRET_NAME=YOUR_ZENML_KEY_HERE" && \
-    echo "ZENML_STORE_URL=YOUR_ZENML_KEY_HERE" && \
-    echo "ZENML_STORE_API_KEY=YOUR_ZENML_KEY_HERE" > .env
+RUN echo "ZENML_PROJECT_SECRET_NAME=YOUR_ZENML_PROJECT_SECRET_NAME" && \
+    echo "ZENML_STORE_URL=YOUR_ZENML_STORE_URL" && \
+    echo "ZENML_STORE_API_KEY=YOUR_ZENML_STORE_API_KEY" > .env
 
 # Create a .vscode directory and settings.json file
 RUN mkdir -p /workspace/.vscode && \

scripts/generate_sandbox_dockerfile.py

@@ -45,14 +45,11 @@
 
 def format_env_key(key):
     """Format environment variable placeholder text."""
-    # Extract the service name from the key
-    service = key.split("_")[0] if "_" in key else key
     # Special case handling
     if key == "GOOGLE_APPLICATION_CREDENTIALS":
         return f"{key}=PATH_TO_YOUR_GOOGLE_CREDENTIALS_FILE"
-    if key == "HF_TOKEN":
-        return f"{key}=YOUR_HUGGINGFACE_TOKEN_HERE"
-    return f"{key}=YOUR_{service}_KEY_HERE"
+
+    return f"{key}=YOUR_{key}"
 
 
 def parse_requirements(project_dir):