Update app.js

app.js

@@ -10,7 +10,6 @@ const NOTDIAMOND_HEADERS = {
   'Content-Type': 'application/json',
   'next-action': '8189eb37107121e024940f588629a394a594e6a4'
 };
-const AUTH_KEY = process.env.AUTH_KEY;
 
 const DEFAULT_MODEL = 'gpt-4-turbo';
 
@@ -27,6 +26,14 @@ const MODEL_MAPPING = {
   'perplexity': 'llama-3.1-sonar-large-128k-online'
 };
 
+function getAuthKey(req) {
+  const authHeader = req.headers['authorization'];
+  if (authHeader && authHeader.startsWith('Bearer ')) {
+    return authHeader.slice(7);
+  }
+  return null;
+}
+
 function createOpenAIChunk(content, model, finishReason = null) {
   return {
     id: `chatcmpl-${crypto.randomUUID()}`,
@@ -143,13 +150,13 @@ app.get('/', (req, res) => {
       }
     },
     availableModels: Object.keys(MODEL_MAPPING),
-    note: "Replace YOUR_AUTH_KEY with the actual authentication key."
+    note: "Replace YOUR_AUTH_KEY with your actual authentication key."
   });
 });
 
 app.get('/ai/v1/models', (req, res) => {
-  const authHeader = req.headers['authorization'];
-  if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.slice(7) !== AUTH_KEY) {
+  const authKey = getAuthKey(req);
+  if (!authKey) {
     return res.status(401).json({ error: 'Unauthorized' });
   }
 
@@ -173,8 +180,8 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
   let status = 200;
 
   try {
-    const authHeader = req.headers['authorization'];
-    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.slice(7) !== AUTH_KEY) {
+    const authKey = getAuthKey(req);
+    if (!authKey) {
       status = 401;
       throw new Error('Unauthorized');
     }
@@ -201,7 +208,8 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
 
     const headers = {
       'Content-Type': 'application/json',
-      'next-action': '4e63dabc37fef18cae74cbfd41d1bace49acf47e'
+      'next-action': '4e63dabc37fef18cae74cbfd41d1bace49acf47e',
+      'Authorization': `Bearer ${authKey}`
     };
 
     const response = await fetch(NOTDIAMOND_URL, {