Spaces:
Sleeping
Sleeping
link0518
commited on
Commit
·
82ecee3
1
Parent(s):
8f02d2f
修复中转
Browse files- src/server/index.js +12 -1
src/server/index.js
CHANGED
|
@@ -130,13 +130,24 @@ app.get('/v1/models', async (req, res) => {
|
|
| 130 |
});
|
| 131 |
|
| 132 |
app.post('/v1/chat/completions', async (req, res) => {
|
| 133 |
-
|
| 134 |
try {
|
| 135 |
|
| 136 |
if (!messages) {
|
| 137 |
return res.status(400).json({ error: 'messages is required' });
|
| 138 |
}
|
| 139 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 140 |
const requestBody = generateRequestBody(messages, model, params, tools);
|
| 141 |
|
| 142 |
|
|
|
|
| 130 |
});
|
| 131 |
|
| 132 |
app.post('/v1/chat/completions', async (req, res) => {
|
| 133 |
+
let { messages, model, stream = true, tools, ...params } = req.body;
|
| 134 |
try {
|
| 135 |
|
| 136 |
if (!messages) {
|
| 137 |
return res.status(400).json({ error: 'messages is required' });
|
| 138 |
}
|
| 139 |
|
| 140 |
+
// 智能检测:NewAPI测速请求通常消息很简单,强制使用非流式响应
|
| 141 |
+
// 检测条件:单条消息 + 内容很短(如 "hi", "test" 等)
|
| 142 |
+
const isSingleShortMessage = messages.length === 1 &&
|
| 143 |
+
messages[0].content &&
|
| 144 |
+
messages[0].content.length < 20;
|
| 145 |
+
|
| 146 |
+
// 如果检测到可能是测速请求,且未明确要求流式,则使用非流式
|
| 147 |
+
if (isSingleShortMessage && req.body.stream === undefined) {
|
| 148 |
+
stream = false;
|
| 149 |
+
}
|
| 150 |
+
|
| 151 |
const requestBody = generateRequestBody(messages, model, params, tools);
|
| 152 |
|
| 153 |
|