kevin commited on
Commit
eaf1a7f
·
1 Parent(s): 38f0a99

出现429时移除token

Browse files
Files changed (1) hide show
  1. core/utils.py +11 -0
core/utils.py CHANGED
@@ -274,6 +274,17 @@ async def process_streaming_response(request: ChatRequest, app_secret: str, curr
274
  )
275
  except httpx.HTTPStatusError as e:
276
  # 这里需要处理401错误
 
 
 
 
 
 
 
 
 
 
 
277
  logger.error(f"HTTP error occurred: {e}")
278
  raise HTTPException(status_code=e.response.status_code, detail=str(e))
279
  except httpx.RequestError as e:
 
274
  )
275
  except httpx.HTTPStatusError as e:
276
  # 这里需要处理401错误
277
+
278
+ # 处理429错误
279
+ if e.response.status_code == 429:
280
+ token_str = os.getenv('TOKEN', '')
281
+ token_array = token_str.split(',')
282
+ token_array.pop(current_index)
283
+ os.environ["TOKEN"] = ','.join(token_array)
284
+ refresh_token_str = os.getenv('REFRESH_TOKEN', '')
285
+ refresh_token_array = refresh_token_str.split(',')
286
+ refresh_token_array.pop(current_index)
287
+ os.environ["REFRESH_TOKEN"] = ','.join(refresh_token_array)
288
  logger.error(f"HTTP error occurred: {e}")
289
  raise HTTPException(status_code=e.response.status_code, detail=str(e))
290
  except httpx.RequestError as e: