Update app.js
Browse files
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 |
-
|
17 |
-
|
18 |
-
];
|
19 |
|
20 |
-
//
|
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,
|