kevin1207 commited on
Commit
41523ce
·
verified ·
1 Parent(s): 052808a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -268,6 +268,9 @@ class AuthTokenManager:
268
 
269
  if normalized_model not in self.token_model_map or not self.token_model_map[normalized_model]:
270
  return None
 
 
 
271
 
272
  token_entry = self.token_model_map[normalized_model][0]
273
  if is_return:
 
268
 
269
  if normalized_model not in self.token_model_map or not self.token_model_map[normalized_model]:
270
  return None
271
+ # 将第一个token移出 第二个作为作为当前token 第一个token放入self.token_model_map[normalized_model]末尾 实现token轮询
272
+ first_token_entry = self.token_model_map[normalized_model].pop(0)
273
+ self.token_model_map[normalized_model].append(first_token_entry)
274
 
275
  token_entry = self.token_model_map[normalized_model][0]
276
  if is_return: