From 270cecab9a92dd53120e4c58d1f5e0566790c110 Mon Sep 17 00:00:00 2001
From: json <avezard.jeason@gmail.com>
Date: Mon, 10 Mar 2025 12:43:22 +0100
Subject: [PATCH] Add VITE_ALLOWED_HOSTS configuration and update environment
 variables

---
 .env.example   | 11 ++++++++---
 Dockerfile     | 10 +++++++---
 vite.config.ts | 13 +++++++++++++
 3 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/.env.example b/.env.example
index 2d736a72cc..ca9568ee3f 100644
--- a/.env.example
+++ b/.env.example
@@ -55,7 +55,7 @@ TOGETHER_API_KEY=
 #Get your Hyperbolics API Key at https://app.hyperbolic.xyz/settings
 #baseURL="https://api.hyperbolic.xyz/v1/chat/completions"
 HYPERBOLIC_API_KEY=
-HYPERBOLIC_API_BASE_URL= 
+HYPERBOLIC_API_BASE_URL=
 
 # Get your Mistral API Key by following these instructions -
 # https://console.mistral.ai/api-keys/
@@ -78,7 +78,7 @@ LMSTUDIO_API_BASE_URL=
 # You only need this environment variable set if you want to use xAI models
 XAI_API_KEY=
 
-# Get your Perplexity API Key here - 
+# Get your Perplexity API Key here -
 # https://www.perplexity.ai/settings/api
 # You only need this environment variable set if you want to use Perplexity models
 PERPLEXITY_API_KEY=
@@ -97,8 +97,13 @@ AWS_BEDROCK_CONFIG=
 # Include this environment variable if you want more logging for debugging locally
 VITE_LOG_LEVEL=debug
 
+# Set this to the allowed hosts for your application, separated by commas
+# Example: VITE_ALLOWED_HOSTS=localhost,127.0.0.1,myapp.com
+# Leave empty if you don't need to restrict allowed hosts
+VITE_ALLOWED_HOSTS=
+
 # Example Context Values for qwen2.5-coder:32b
-# 
+#
 # DEFAULT_NUM_CTX=32768 # Consumes 36GB of VRAM
 # DEFAULT_NUM_CTX=24576 # Consumes 32GB of VRAM
 # DEFAULT_NUM_CTX=12288 # Consumes 26GB of VRAM
diff --git a/Dockerfile b/Dockerfile
index 1cd3f0bfca..91e362f7fd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -34,6 +34,7 @@ ARG TOGETHER_API_BASE_URL
 ARG AWS_BEDROCK_CONFIG
 ARG VITE_LOG_LEVEL=debug
 ARG DEFAULT_NUM_CTX
+ARG VITE_ALLOWED_HOSTS
 
 ENV WRANGLER_SEND_METRICS=false \
     GROQ_API_KEY=${GROQ_API_KEY} \
@@ -49,7 +50,8 @@ ENV WRANGLER_SEND_METRICS=false \
     AWS_BEDROCK_CONFIG=${AWS_BEDROCK_CONFIG} \
     VITE_LOG_LEVEL=${VITE_LOG_LEVEL} \
     DEFAULT_NUM_CTX=${DEFAULT_NUM_CTX}\
-    RUNNING_IN_DOCKER=true
+    RUNNING_IN_DOCKER=true\
+    VITE_ALLOWED_HOSTS=${VITE_ALLOWED_HOSTS}
 
 # Pre-configure wrangler to disable metrics
 RUN mkdir -p /root/.config/.wrangler && \
@@ -64,7 +66,7 @@ FROM base AS bolt-ai-development
 
 # Define the same environment variables for development
 ARG GROQ_API_KEY
-ARG HuggingFace 
+ARG HuggingFace
 ARG OPENAI_API_KEY
 ARG ANTHROPIC_API_KEY
 ARG OPEN_ROUTER_API_KEY
@@ -75,6 +77,7 @@ ARG TOGETHER_API_KEY
 ARG TOGETHER_API_BASE_URL
 ARG VITE_LOG_LEVEL=debug
 ARG DEFAULT_NUM_CTX
+ARG VITE_ALLOWED_HOSTS
 
 ENV GROQ_API_KEY=${GROQ_API_KEY} \
     HuggingFace_API_KEY=${HuggingFace_API_KEY} \
@@ -89,7 +92,8 @@ ENV GROQ_API_KEY=${GROQ_API_KEY} \
     AWS_BEDROCK_CONFIG=${AWS_BEDROCK_CONFIG} \
     VITE_LOG_LEVEL=${VITE_LOG_LEVEL} \
     DEFAULT_NUM_CTX=${DEFAULT_NUM_CTX}\
-    RUNNING_IN_DOCKER=true
+    RUNNING_IN_DOCKER=true\
+    VITE_ALLOWED_HOSTS=${VITE_ALLOWED_HOSTS}
 
 RUN mkdir -p ${WORKDIR}/run
 CMD pnpm run dev --host
diff --git a/vite.config.ts b/vite.config.ts
index a9351c12ff..fbba7914f1 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -11,6 +11,16 @@ import { join } from 'path';
 
 dotenv.config();
 
+
+// Define allowed hosts based on the VITE_ALLOWED_HOSTS environment variable
+// If the environment variable is not set, default to an empty array
+const allowedHosts = process.env.VITE_ALLOWED_HOSTS
+  ?  process.env.VITE_ALLOWED_HOSTS.split(',')
+    .filter(host => host.trim() !== '')
+    .map(host => host.trim())
+  : []
+
+
 // Get detailed git info with fallbacks
 const getGitInfo = () => {
   try {
@@ -166,6 +176,9 @@ export default defineConfig((config) => {
         },
       },
     },
+    server: {
+      allowedHosts
+    }
   };
 });