Spaces:
Sleeping
Sleeping
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 |
|