Update app.js

app.js

@@ -10,9 +10,19 @@ const NOTDIAMOND_HEADERS = {
   'Content-Type': 'application/json',
   'next-action': '8189eb37107121e024940f588629a394a594e6a4'
 };
-const AUTH_KEY = process.env.AUTH_KEY;
-
-const DEFAULT_MODEL = 'gpt-4-turbo-2024-04-09';
+const AUTH_KEY = process.env.AUTH_KEY';
+
+const DEFAULT_MODEL = 'gpt-4o';
+
+const MODEL_MAPPING = {
+  'gpt-4-turbo': 'gpt-4-turbo-2024-04-09',
+  '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'
+};
 
 function createOpenAIChunk(content, model, finishReason = null) {
   return {
@@ -109,6 +119,32 @@ async function* streamNotdiamondResponse(response, model) {
 
 app.use(express.json());
 
+// 添加根路由，提供服务使用指引
+app.get('/', (req, res) => {
+  res.json({
+    service: "AI Chat Completion Proxy",
+    usage: {
+      endpoint: "/ai/v1/chat/completions",
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": "Bearer YOUR_AUTH_KEY"
+      },
+      body: {
+        model: "One of: " + Object.keys(MODEL_MAPPING).join(", "),
+        messages: [
+          { role: "system", content: "You are a helpful assistant." },
+          { role: "user", content: "Hello, who are you?" }
+        ],
+        stream: false,
+        temperature: 0.7
+      }
+    },
+    availableModels: Object.keys(MODEL_MAPPING),
+    note: "Replace YOUR_AUTH_KEY with the actual authentication key."
+  });
+});
+
 app.post('/ai/v1/chat/completions', async (req, res) => {
   const authHeader = req.headers['authorization'];
   if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.slice(7) !== AUTH_KEY) {
@@ -119,7 +155,8 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
     let requestData = req.body;
 
     let messages = requestData.messages;
-    let model = requestData.model || DEFAULT_MODEL;
+    let userModel = requestData.model || DEFAULT_MODEL;
+    let model = MODEL_MAPPING[userModel] || userModel;
     const stream = requestData.stream || false;
 
     if (!messages || !Array.isArray(messages)) {
@@ -177,7 +214,7 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
         id: `chatcmpl-${crypto.randomUUID()}`,
         object: "chat.completion",
         created: Math.floor(Date.now() / 1000),
-        model: model,
+        model: userModel,
         system_fingerprint: "fp_4e2b2da518",
         choices: [
           {