smgc commited on
Commit
882f772
1 Parent(s): 44ef370

Update app.js

Browse files
Files changed (1) hide show
  1. app.js +5 -18
app.js CHANGED
@@ -5,19 +5,17 @@ const crypto = require('crypto');
5
  const app = express();
6
  const port = 8000;
7
 
8
- // 这里粘贴您提供的所有常量和函数定义
9
  const NOTDIAMOND_URL = 'https://chat.notdiamond.ai/mini-chat';
10
  const NOTDIAMOND_HEADERS = {
11
  'Content-Type': 'application/json',
12
  'next-action': '8189eb37107121e024940f588629a394a594e6a4'
13
  };
14
- const AUTH_KEY = process.env.AUTH_KEY;
15
 
16
- const MODEL_LIST = [
17
- {"provider":"openai","model":"gpt-4-turbo-2024-04-09"},{"provider":"google","model":"gemini-1.5-pro-exp-0801"},{"provider":"togetherai","model":"Meta-Llama-3.1-70B-Instruct-Turbo"},{"provider":"togetherai","model":"Meta-Llama-3.1-405B-Instruct-Turbo"},{"provider":"perplexity","model":"llama-3.1-sonar-large-128k-online"},{"provider":"google","model":"gemini-1.5-pro-latest"},{"provider":"anthropic","model":"claude-3-5-sonnet-20240620"},{"provider":"anthropic","model":"claude-3-haiku-20240307"},{"provider":"openai","model":"gpt-4o-mini"},{"provider":"openai","model":"gpt-4o"},{"provider":"mistral","model":"mistral-large-2407"}
18
- ];
19
 
20
- // 这里粘贴所有的函数定义(getNotdiamondModel, createOpenAIChunk, streamNotdiamondResponse
21
 
22
  app.use(express.json());
23
 
@@ -31,7 +29,7 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
31
  let requestData = req.body;
32
 
33
  let messages = requestData.messages;
34
- let model = requestData.model || '';
35
  const stream = requestData.stream || false;
36
 
37
  if (!messages || !Array.isArray(messages)) {
@@ -42,17 +40,6 @@ app.post('/ai/v1/chat/completions', async (req, res) => {
42
  });
43
  }
44
 
45
- try {
46
- const { model: resolvedModel } = await getNotdiamondModel(messages, model);
47
- if (resolvedModel) {
48
- model = resolvedModel;
49
- } else {
50
- throw new Error("Failed to get model from NotDiamond");
51
- }
52
- } catch (error) {
53
- return res.status(400).json({ error: error.message });
54
- }
55
-
56
  const payload = {
57
  messages: messages,
58
  model: model,
 
5
  const app = express();
6
  const port = 8000;
7
 
 
8
  const NOTDIAMOND_URL = 'https://chat.notdiamond.ai/mini-chat';
9
  const NOTDIAMOND_HEADERS = {
10
  'Content-Type': 'application/json',
11
  'next-action': '8189eb37107121e024940f588629a394a594e6a4'
12
  };
13
+ const AUTH_KEY = process.env.AUTH_KEY || 'sk-linuxdo';
14
 
15
+ // 默认模型,如果用户没有指定模型,就使用这个
16
+ const DEFAULT_MODEL = 'gpt-4-turbo-2024-04-09';
 
17
 
18
+ // 保留其他函数定义:createOpenAIChunk, streamNotdiamondResponse
19
 
20
  app.use(express.json());
21
 
 
29
  let requestData = req.body;
30
 
31
  let messages = requestData.messages;
32
+ let model = requestData.model || DEFAULT_MODEL;
33
  const stream = requestData.stream || false;
34
 
35
  if (!messages || !Array.isArray(messages)) {
 
40
  });
41
  }
42
 
 
 
 
 
 
 
 
 
 
 
 
43
  const payload = {
44
  messages: messages,
45
  model: model,