Update app.js

app.js

@@ -10,15 +10,12 @@ const NOTDIAMOND_HEADERS = {
   'Content-Type': 'application/json',
   'next-action': '8189eb37107121e024940f588629a394a594e6a4'
 };
-const AUTH_KEY = process.env.AUTH_KEY;
+const AUTH_KEY = process.env.AUTH_KEY || 'sk-linuxdo';
 
-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',
@@ -122,7 +119,6 @@ async function* streamNotdiamondResponse(response, model) {
 
 app.use(express.json());
 
-// 添加根路由，提供服务使用指引
 app.get('/', (req, res) => {
   res.json({
     service: "AI Chat Completion Proxy",
@@ -149,25 +145,26 @@ app.get('/', (req, res) => {
 });
 
 app.post('/ai/v1/chat/completions', async (req, res) => {
-  const authHeader = req.headers['authorization'];
-  if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.slice(7) !== AUTH_KEY) {
-    return res.status(401).json({ error: 'Unauthorized' });
-  }
+  const startTime = new Date();
+  const clientIP = req.ip || req.connection.remoteAddress;
+  let userModel = req.body.model || DEFAULT_MODEL;
+  let status = 200;
 
   try {
-    let requestData = req.body;
+    const authHeader = req.headers['authorization'];
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.slice(7) !== AUTH_KEY) {
+      status = 401;
+      throw new Error('Unauthorized');
+    }
 
+    let requestData = req.body;
     let messages = requestData.messages;
-    let userModel = requestData.model || DEFAULT_MODEL;
     let model = MODEL_MAPPING[userModel] || userModel;
     const stream = requestData.stream || false;
 
     if (!messages || !Array.isArray(messages)) {
-      return res.status(400).json({ 
-        error: 'Invalid request body', 
-        details: 'messages should be an array of message objects',
-        receivedBody: req.body
-      });
+      status = 400;
+      throw new Error('Invalid request body: messages should be an array of message objects');
     }
 
     const payload = {
@@ -235,12 +232,17 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
           total_tokens: Math.floor(fullContent.length / 2)
         }
       });
-    } 
+    }
   } catch (error) {
-    res.status(500).json({ 
-      error: 'Internal Server Error', 
+    status = status === 200 ? 500 : status;
+    res.status(status).json({ 
+      error: 'Error',
       details: error.message
     });
+  } finally {
+    const endTime = new Date();
+    const duration = endTime - startTime;
+    console.log(`${endTime.toISOString()} - ${clientIP} - ${userModel} - ${status} - ${duration}ms`);
   }
 });