Update app.js

app.js

@@ -11,31 +11,59 @@ const NOTDIAMOND_HEADERS = {
   'next-action': '8189eb37107121e024940f588629a394a594e6a4'
 };
 
-const DEFAULT_MODEL = 'gpt-4o';
+const DEFAULT_MODEL = 'gpt-4-turbo';
 
 const MODEL_MAPPING = {
-  'gpt-4o-mini': 'gpt-4o-mini',
-  'gpt-4o': 'gpt-4o',
-  'gpt-4-turbo': 'gpt-4-turbo-2024-04-09',
-  'claude-3-haiku-20240307': 'anthropic.claude-3-haiku-20240307-v1:0',
-  'claude-3-5-sonnet-20240620': 'anthropic.claude-3-5-sonnet-20240620-v1:0',
-  'gemini-1.5-pro-latest': 'models/gemini-1.5-pro-latest',
-  'gemini-1.5-pro-exp-0801': 'models/gemini-1.5-pro-exp-0801',
-  'llama-3.1-405b-instruct': 'meta.llama3-1-405b-instruct-v1:0',
-  'llama-3.1-70b-instruct': 'meta.llama3-1-70b-instruct-v1:0',
-  'perplexity': 'llama-3.1-sonar-large-128k-online'
+    "gpt-4-turbo": {
+        "provider": "openai",
+        "mapping": "gpt-4-turbo-2024-04-09"
+    },
+    "gemini-1.5-pro-exp-0801": {
+        "provider": "google",
+        "mapping": "models/gemini-1.5-pro-exp-0801"
+    },
+    "Meta-Llama-3.1-70B-Instruct-Turbo": {
+        "provider": "togetherai",
+        "mapping": "meta.llama3-1-70b-instruct-v1:0"
+    },
+    "Meta-Llama-3.1-405B-Instruct-Turbo": {
+        "provider": "togetherai",
+        "mapping": "meta.llama3-1-405b-instruct-v1:0"
+    },
+    "llama-3.1-sonar-large-128k-online": {
+        "provider": "perplexity",
+        "mapping": "llama-3.1-sonar-large-128k-online"
+    },
+    "gemini-1.5-pro-latest": {
+        "provider": "google",
+        "mapping": "models/gemini-1.5-pro-latest"
+    },
+    "claude-3-5-sonnet-20240620": {
+        "provider": "anthropic",
+        "mapping": "anthropic.claude-3-5-sonnet-20240620-v1:0"
+    },
+    "claude-3-haiku-20240307": {
+        "provider": "anthropic",
+        "mapping": "anthropic.claude-3-haiku-20240307-v1:0"
+    },
+    "gpt-4o-mini": {
+        "provider": "openai",
+        "mapping": "gpt-4o-mini"
+    },
+    "gpt-4o": {
+        "provider": "openai",
+        "mapping": "gpt-4o"
+    },
+    "mistral-large-2407": {
+        "provider": "mistral",
+        "mapping": "mistral.mistral-large-2407-v1:0"
+    }
 };
 
 function getAuthKey(req) {
-  const cookies = req.headers.cookie;
-  if (cookies) {
-    const cookieArray = cookies.split(';');
-    for (let cookie of cookieArray) {
-      cookie = cookie.trim();
-      if (cookie.startsWith('sb-spuckhogycrxcbomznwo-auth-token=')) {
-        return cookie.split('=')[1];
-      }
-    }
+  const authHeader = req.headers['authorization'];
+  if (authHeader && authHeader.startsWith('Bearer ')) {
+    return authHeader.slice(7);
   }
   return null;
 }
@@ -143,7 +171,7 @@ app.get('/', (req, res) => {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
-        "Cookie": "sb-spuckhogycrxcbomznwo-auth-token=YOUR_AUTH_TOKEN"
+        "Authorization": "Bearer YOUR_AUTH_KEY"
       },
       body: {
         model: "One of: " + Object.keys(MODEL_MAPPING).join(", "),
@@ -156,7 +184,7 @@ app.get('/', (req, res) => {
       }
     },
     availableModels: Object.keys(MODEL_MAPPING),
-    note: "Replace YOUR_AUTH_TOKEN with your actual authentication token."
+    note: "Replace YOUR_AUTH_KEY with your actual authentication key."
   });
 });
 
@@ -168,11 +196,14 @@ app.get('/ai/v1/models', (req, res) => {
 
   res.json({
     object: "list",
-    data: Object.keys(MODEL_MAPPING).map(model => ({
-      id: model,
+    data: Object.entries(MODEL_MAPPING).map(([id, info]) => ({
+      id: id,
       object: "model",
       created: Math.floor(Date.now() / 1000),
-      owned_by: "notdiamond"
+      owned_by: info.provider,
+      permission: [],
+      root: id,
+      parent: null,
     }))
   });
 
@@ -194,7 +225,12 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
 
     let requestData = req.body;
     let messages = requestData.messages;
-    let model = MODEL_MAPPING[userModel] || userModel;
+    let modelInfo = MODEL_MAPPING[userModel];
+    if (!modelInfo) {
+      status = 400;
+      throw new Error('Invalid model specified');
+    }
+    let model = modelInfo.mapping;
     const stream = requestData.stream || false;
 
     if (!messages || !Array.isArray(messages)) {
@@ -215,7 +251,7 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
     const headers = {
       'Content-Type': 'application/json',
       'next-action': '4e63dabc37fef18cae74cbfd41d1bace49acf47e',
-      'Cookie': `sb-spuckhogycrxcbomznwo-auth-token=${authKey}`
+      'Authorization': `Bearer ${authKey}`
     };
 
     const response = await fetch(NOTDIAMOND_URL, {
@@ -224,7 +260,7 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
       body: JSON.stringify([payload])
     });
 
-    const generator = streamNotdiamondResponse(response, model);
+    const generator = streamNotdiamondResponse(response, userModel);
     
     if (stream) {
       res.writeHead(200, {