Spaces:
Running
Running
Update index.js
Browse files
index.js
CHANGED
@@ -185,7 +185,7 @@ class WebSocketUtils {
|
|
185 |
static async createWebSocketClient(requestPayload) {
|
186 |
// 检查当前连接数是否达到上限
|
187 |
if (this.activeConnections.size >= this.MAX_CONNECTIONS) {
|
188 |
-
throw new Error(`当前连接数已达到上限 (${this.MAX_CONNECTIONS})
|
189 |
}
|
190 |
|
191 |
let timeoutId;
|
@@ -258,7 +258,8 @@ class WebSocketUtils {
|
|
258 |
case 'error':
|
259 |
console.error('WebSocket错误:', parsedMessage.payload[0].message);
|
260 |
ws.close();
|
261 |
-
|
|
|
262 |
}
|
263 |
});
|
264 |
|
@@ -270,7 +271,7 @@ class WebSocketUtils {
|
|
270 |
if (ws.readyState === WebSocket.OPEN) {
|
271 |
ws.close();
|
272 |
}
|
273 |
-
|
274 |
});
|
275 |
|
276 |
ws.on('close', (code, reason) => {
|
@@ -279,7 +280,7 @@ class WebSocketUtils {
|
|
279 |
this.activeConnections.delete(ws);
|
280 |
console.log(`连接关闭,当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
|
281 |
if (!isComplete) {
|
282 |
-
|
283 |
}
|
284 |
});
|
285 |
});
|
@@ -365,6 +366,7 @@ app.use(cors({
|
|
365 |
allowedHeaders: ['Content-Type', 'Authorization']
|
366 |
}));
|
367 |
|
|
|
368 |
// 获取模型列表路由
|
369 |
app.get('/hf/v1/models', (req, res) => {
|
370 |
res.json({
|
@@ -405,7 +407,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
|
|
405 |
}
|
406 |
} catch (error) {
|
407 |
console.error('处理请求时发生错误:', error);
|
408 |
-
res.status(500).json({ error: "
|
409 |
}
|
410 |
});
|
411 |
|
|
|
185 |
static async createWebSocketClient(requestPayload) {
|
186 |
// 检查当前连接数是否达到上限
|
187 |
if (this.activeConnections.size >= this.MAX_CONNECTIONS) {
|
188 |
+
throw new Error(`当前连接数已达到上限 (${this.MAX_CONNECTIONS}),请稍后重试喵!`);
|
189 |
}
|
190 |
|
191 |
let timeoutId;
|
|
|
258 |
case 'error':
|
259 |
console.error('WebSocket错误:', parsedMessage.payload[0].message);
|
260 |
ws.close();
|
261 |
+
reject(new Error(`WebSocket错误: ${parsedMessage.payload[0].message}`));
|
262 |
+
break;
|
263 |
}
|
264 |
});
|
265 |
|
|
|
271 |
if (ws.readyState === WebSocket.OPEN) {
|
272 |
ws.close();
|
273 |
}
|
274 |
+
reject(err);
|
275 |
});
|
276 |
|
277 |
ws.on('close', (code, reason) => {
|
|
|
280 |
this.activeConnections.delete(ws);
|
281 |
console.log(`连接关闭,当前活跃连接数: ${this.activeConnections.size}/${this.MAX_CONNECTIONS}`);
|
282 |
if (!isComplete) {
|
283 |
+
reject(new Error('WebSocket closed unexpectedly'));
|
284 |
}
|
285 |
});
|
286 |
});
|
|
|
366 |
allowedHeaders: ['Content-Type', 'Authorization']
|
367 |
}));
|
368 |
|
369 |
+
|
370 |
// 获取模型列表路由
|
371 |
app.get('/hf/v1/models', (req, res) => {
|
372 |
res.json({
|
|
|
407 |
}
|
408 |
} catch (error) {
|
409 |
console.error('处理请求时发生错误:', error);
|
410 |
+
res.status(500).json({ error: "内部服务器错误,请查询日志记录!", details: error.message });
|
411 |
}
|
412 |
});
|
413 |
|