yxmiler commited on
Commit
fb1e69d
·
verified ·
1 Parent(s): 0cbee1c

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +14 -2
index.js CHANGED
@@ -522,20 +522,32 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
522
  switch (reqStatus) {
523
  case 200:
524
  console.log("请求成功");
 
 
525
  // 处理响应
526
  if (req.body.stream) {
527
  await ResponseHandler.handleStreamResponse(response, req.body.model, res);
528
  } else {
529
  await ResponseHandler.handleNormalResponse(response, req.body.model, res);
530
  }
531
- await tokenManager.updateTokens(response);
532
  return;
533
  case 202:
534
  console.log("请求受限,更新WAF");
535
  await tokenManager.updateTokens(response, true);
536
  break;
 
 
 
 
 
 
 
 
 
 
 
 
537
  default:
538
- console.log(await response.text());
539
  throw new Error(`请求失败! status: ${response.status}`);
540
  }
541
  console.log("请求失败,重试中...");
 
522
  switch (reqStatus) {
523
  case 200:
524
  console.log("请求成功");
525
+ // 异步更新token
526
+ tokenManager.updateTokens(response)
527
  // 处理响应
528
  if (req.body.stream) {
529
  await ResponseHandler.handleStreamResponse(response, req.body.model, res);
530
  } else {
531
  await ResponseHandler.handleNormalResponse(response, req.body.model, res);
532
  }
 
533
  return;
534
  case 202:
535
  console.log("请求受限,更新WAF");
536
  await tokenManager.updateTokens(response, true);
537
  break;
538
+ case 405:
539
+ console.log("人机验证");
540
+ await tokenManager.updateTokens(response, true);//常识获取waf,然后返回错误提示。
541
+ res.status(405).json({
542
+ error: {
543
+ message: "人机验证",
544
+ type: 'server_error',
545
+ param: null,
546
+ code: null
547
+ }
548
+ });
549
+ return;
550
  default:
 
551
  throw new Error(`请求失败! status: ${response.status}`);
552
  }
553
  console.log("请求失败,重试中...");