From 8e413796d698a35648e835ae8a8a72a6b49db52c Mon Sep 17 00:00:00 2001
From: chand1012 <chandler@chand1012.dev>
Date: Thu, 20 Mar 2025 21:55:14 -0400
Subject: [PATCH 1/2] Add optional env var system

---
 src/lib/config.ts | 52 +++++++++++++++++++++++++++++++++++++----------
 1 file changed, 41 insertions(+), 11 deletions(-)

diff --git a/src/lib/config.ts b/src/lib/config.ts
index ef99eed..ac02f82 100644
--- a/src/lib/config.ts
+++ b/src/lib/config.ts
@@ -45,32 +45,62 @@ const loadConfig = () =>
     fs.readFileSync(path.join(process.cwd(), `${configFileName}`), 'utf-8'),
   ) as any as Config;
 
+const getEnvVar = (key: string): string | undefined => {
+  return process.env[key];
+};
+
+const getConfigValue = (path: string[], defaultValue: string): string => {
+  // Convert path to environment variable name (e.g., ['MODELS', 'GROQ', 'API_KEY'] -> 'GROQ_API_KEY')
+  const envKey = path.slice(1).join('_').toUpperCase();
+  const envValue = getEnvVar(envKey);
+
+  if (envValue !== undefined) {
+    return envValue;
+  }
+
+  // Fall back to config.toml
+  let value: any = loadConfig();
+  for (const key of path) {
+    value = value[key];
+    if (value === undefined) {
+      return defaultValue;
+    }
+  }
+  return value;
+};
+
 export const getSimilarityMeasure = () =>
-  loadConfig().GENERAL.SIMILARITY_MEASURE;
+  getConfigValue(['GENERAL', 'SIMILARITY_MEASURE'], 'cosine');
 
-export const getKeepAlive = () => loadConfig().GENERAL.KEEP_ALIVE;
+export const getKeepAlive = () =>
+  getConfigValue(['GENERAL', 'KEEP_ALIVE'], '30s');
 
-export const getOpenaiApiKey = () => loadConfig().MODELS.OPENAI.API_KEY;
+export const getOpenaiApiKey = () =>
+  getConfigValue(['MODELS', 'OPENAI', 'API_KEY'], '');
 
-export const getGroqApiKey = () => loadConfig().MODELS.GROQ.API_KEY;
+export const getGroqApiKey = () =>
+  getConfigValue(['MODELS', 'GROQ', 'API_KEY'], '');
 
-export const getAnthropicApiKey = () => loadConfig().MODELS.ANTHROPIC.API_KEY;
+export const getAnthropicApiKey = () =>
+  getConfigValue(['MODELS', 'ANTHROPIC', 'API_KEY'], '');
 
-export const getGeminiApiKey = () => loadConfig().MODELS.GEMINI.API_KEY;
+export const getGeminiApiKey = () =>
+  getConfigValue(['MODELS', 'GEMINI', 'API_KEY'], '');
 
 export const getSearxngApiEndpoint = () =>
-  process.env.SEARXNG_API_URL || loadConfig().API_ENDPOINTS.SEARXNG;
+  getConfigValue(['API_ENDPOINTS', 'SEARXNG'], '');
 
-export const getOllamaApiEndpoint = () => loadConfig().MODELS.OLLAMA.API_URL;
+export const getOllamaApiEndpoint = () =>
+  getConfigValue(['MODELS', 'OLLAMA', 'API_URL'], 'http://localhost:11434');
 
 export const getCustomOpenaiApiKey = () =>
-  loadConfig().MODELS.CUSTOM_OPENAI.API_KEY;
+  getConfigValue(['MODELS', 'CUSTOM_OPENAI', 'API_KEY'], '');
 
 export const getCustomOpenaiApiUrl = () =>
-  loadConfig().MODELS.CUSTOM_OPENAI.API_URL;
+  getConfigValue(['MODELS', 'CUSTOM_OPENAI', 'API_URL'], '');
 
 export const getCustomOpenaiModelName = () =>
-  loadConfig().MODELS.CUSTOM_OPENAI.MODEL_NAME;
+  getConfigValue(['MODELS', 'CUSTOM_OPENAI', 'MODEL_NAME'], '');
 
 const mergeConfigs = (current: any, update: any): any => {
   if (update === null || update === undefined) {

From a4c0fd7ea443ade55f2621db85887a4430f4b240 Mon Sep 17 00:00:00 2001
From: chand1012 <chandler@chand1012.dev>
Date: Fri, 21 Mar 2025 09:51:07 -0400
Subject: [PATCH 2/2] Add .env to dockerfile and fix SEARXNG URL var

---
 docker-compose.yaml |  2 ++
 src/lib/config.ts   | 13 ++++++++-----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index b702b4e..aab8f28 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -16,6 +16,8 @@ services:
       dockerfile: app.dockerfile
     environment:
       - SEARXNG_API_URL=http://searxng:8080
+    env_file:
+      - .env
     ports:
       - 3000:3000
     networks:
diff --git a/src/lib/config.ts b/src/lib/config.ts
index ac02f82..615dc4a 100644
--- a/src/lib/config.ts
+++ b/src/lib/config.ts
@@ -40,10 +40,13 @@ type RecursivePartial<T> = {
   [P in keyof T]?: RecursivePartial<T[P]>;
 };
 
-const loadConfig = () =>
-  toml.parse(
-    fs.readFileSync(path.join(process.cwd(), `${configFileName}`), 'utf-8'),
-  ) as any as Config;
+const loadConfig = () => {
+  const configPath = path.join(process.cwd(), configFileName);
+  if (!fs.existsSync(configPath) || fs.lstatSync(configPath).isDirectory()) {
+    return {} as Config;
+  }
+  return toml.parse(fs.readFileSync(configPath, 'utf-8')) as any as Config;
+};
 
 const getEnvVar = (key: string): string | undefined => {
   return process.env[key];
@@ -88,7 +91,7 @@ export const getGeminiApiKey = () =>
   getConfigValue(['MODELS', 'GEMINI', 'API_KEY'], '');
 
 export const getSearxngApiEndpoint = () =>
-  getConfigValue(['API_ENDPOINTS', 'SEARXNG'], '');
+  process.env.SEARXNG_API_URL || getConfigValue(['API_ENDPOINTS', 'SEARXNG'], '');
 
 export const getOllamaApiEndpoint = () =>
   getConfigValue(['MODELS', 'OLLAMA', 'API_URL'], 'http://localhost:11434');