yxmiler commited on
Commit
f842b96
·
verified ·
1 Parent(s): 0a81d64

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +7 -5
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
- throw new Error(`WebSocket错误: ${parsedMessage.payload[0].message}`);
 
262
  }
263
  });
264
 
@@ -270,7 +271,7 @@ class WebSocketUtils {
270
  if (ws.readyState === WebSocket.OPEN) {
271
  ws.close();
272
  }
273
- throw new Error(`WebSocket错误: ${err}`);
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
- throw new Error('WebSocket closed unexpectedly');
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: "内部服务器错误", details: error.message });
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