Update Dockerfile

Dockerfile

@@ -1,20 +1,12 @@
-# Use the official Node.js 20.18.0 image
-FROM node:20.18.0
+ARG BASE=node:20.18.0
+FROM ${BASE} AS base
 
-# Install pnpm
-RUN curl -fsSL https://get.pnpm.io/install.sh | sh -
-
-# Update PATH to include pnpm
-ENV PATH=/home/node/.local/share/pnpm/lib/node_modules/pnpm/dist/bin:$PATH
-
-# Set the working directory
 WORKDIR /app
 
-# Copy package.json and pnpm-lock.yaml
-COPY package.json pnpm-lock.yaml.
+# Install dependencies (this step is cached as long as the dependencies don't change)
+COPY package.json pnpm-lock.yaml ./
 
-# Install dependencies
-RUN pnpm install
+RUN corepack enable pnpm && pnpm install
 
 # Copy the rest of your app's source code
 COPY . .
@@ -22,6 +14,9 @@ COPY . .
 # Expose the port the app runs on
 EXPOSE 5173
 
+# Production image
+FROM base AS bolt-ai-production
+
 # Define environment variables with default values or let them be overridden
 ARG GROQ_API_KEY
 ARG HuggingFace_API_KEY
@@ -37,7 +32,7 @@ ARG DEFAULT_NUM_CTX
 
 ENV WRANGLER_SEND_METRICS=false \
     GROQ_API_KEY=${GROQ_API_KEY} \
-    HuggingFace_API_KEY=${HuggingFace_API_KEY} \
+    HuggingFace_KEY=${HuggingFace_API_KEY} \
     OPENAI_API_KEY=${OPENAI_API_KEY} \
     ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY} \
     OPEN_ROUTER_API_KEY=${OPEN_ROUTER_API_KEY} \
@@ -52,8 +47,37 @@ ENV WRANGLER_SEND_METRICS=false \
 RUN mkdir -p /root/.config/.wrangler && \
     echo '{"enabled":false}' > /root/.config/.wrangler/metrics.json
 
-# Build the application
-RUN pnpm run build
+RUN npm run build
+
+CMD [ "pnpm", "run", "dockerstart"]
+
+# Development image
+FROM base AS bolt-ai-development
+
+# Define the same environment variables for development
+ARG GROQ_API_KEY
+ARG HuggingFace 
+ARG OPENAI_API_KEY
+ARG ANTHROPIC_API_KEY
+ARG OPEN_ROUTER_API_KEY
+ARG GOOGLE_GENERATIVE_AI_API_KEY
+ARG OLLAMA_API_BASE_URL
+ARG TOGETHER_API_KEY
+ARG TOGETHER_API_BASE_URL
+ARG VITE_LOG_LEVEL=debug
+ARG DEFAULT_NUM_CTX
+
+ENV GROQ_API_KEY=${GROQ_API_KEY} \
+    HuggingFace_API_KEY=${HuggingFace_API_KEY} \
+    OPENAI_API_KEY=${OPENAI_API_KEY} \
+    ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY} \
+    OPEN_ROUTER_API_KEY=${OPEN_ROUTER_API_KEY} \
+    GOOGLE_GENERATIVE_AI_API_KEY=${GOOGLE_GENERATIVE_AI_API_KEY} \
+    OLLAMA_API_BASE_URL=${OLLAMA_API_BASE_URL} \
+    TOGETHER_API_KEY=${TOGETHER_API_KEY} \
+    TOGETHER_API_BASE_URL=${TOGETHER_API_BASE_URL} \
+    VITE_LOG_LEVEL=${VITE_LOG_LEVEL} \
+    DEFAULT_NUM_CTX=${DEFAULT_NUM_CTX}
 
-# Command to run the application
-CMD ["pnpm", "run", "dockerstart"]
+RUN mkdir -p ${WORKDIR}/run
+CMD pnpm run dev --host