yym68686 commited on
Commit
f4d6dda
·
1 Parent(s): 5c2c740

🐛 Bug: Fix the bug where httpx.RemoteProtocolError was not caught

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -228,7 +228,7 @@ async def process_request(request: Union[RequestModel, ImageGenerationRequest],
228
  app.middleware_stack.app.channel_success_counts[provider['provider']] += 1
229
 
230
  return response
231
- except (Exception, HTTPException, asyncio.CancelledError, httpx.ReadError) as e:
232
  # 更新失败计数
233
  async with app.middleware_stack.app.lock:
234
  app.middleware_stack.app.channel_failure_counts[provider['provider']] += 1
@@ -388,7 +388,7 @@ class ModelRequestHandler:
388
  continue
389
  else:
390
  raise HTTPException(status_code=500, detail=f"Error: Current provider response failed: {error_message}")
391
- except (Exception, asyncio.CancelledError, httpx.ReadError) as e:
392
  logger.error(f"Error with provider {provider['provider']}: {str(e)}")
393
  error_message = str(e)
394
  if auto_retry:
 
228
  app.middleware_stack.app.channel_success_counts[provider['provider']] += 1
229
 
230
  return response
231
+ except (Exception, HTTPException, asyncio.CancelledError, httpx.ReadError, httpx.RemoteProtocolError) as e:
232
  # 更新失败计数
233
  async with app.middleware_stack.app.lock:
234
  app.middleware_stack.app.channel_failure_counts[provider['provider']] += 1
 
388
  continue
389
  else:
390
  raise HTTPException(status_code=500, detail=f"Error: Current provider response failed: {error_message}")
391
+ except (Exception, asyncio.CancelledError, httpx.ReadError, httpx.RemoteProtocolError) as e:
392
  logger.error(f"Error with provider {provider['provider']}: {str(e)}")
393
  error_message = str(e)
394
  if auto_retry: