superdup95 commited on
Commit
9d7fea1
·
verified ·
1 Parent(s): c62888d

Update api_usage.py

Browse files
Files changed (1) hide show
  1. api_usage.py +3 -2
api_usage.py CHANGED
@@ -685,8 +685,9 @@ async def check_bedrock_claude_status(session, key, secret):
685
  if region and model_name:
686
  if msg == "Maybe":
687
  invoke_info = await send_signed_request_bedrock(session, payload, f"anthropic.{model_name}", key, secret, region)
688
- if 'messages.0' in invoke_info.get('message') or 'many requests' in invoke_info.get('message') or 'equal to 1' in invoke_info.get('message'):
689
- models[model_name].append(f'{region}')
 
690
  else:
691
  models[model_name].append(region)
692
  elif form_info.get('message') == "Operation not allowed" and "Operation not allowed" not in models[model_name]:
 
685
  if region and model_name:
686
  if msg == "Maybe":
687
  invoke_info = await send_signed_request_bedrock(session, payload, f"anthropic.{model_name}", key, secret, region)
688
+ if 'message' in invoke_info:
689
+ if 'messages.0' in invoke_info.get('message') or 'many requests' in invoke_info.get('message') or 'equal to 1' in invoke_info.get('message'):
690
+ models[model_name].append(f'{region}')
691
  else:
692
  models[model_name].append(region)
693
  elif form_info.get('message') == "Operation not allowed" and "Operation not allowed" not in models[model_name]: