Update api_usage.py

api_usage.py

@@ -570,19 +570,23 @@ def is_model_working(form_info, model_info):
         entitlementAvai = model_info['entitlementAvailability']
         
         if 'formData' in form_status and agreement_status == 'AVAILABLE' and auth_status == 'AUTHORIZED' and entitlementAvai == 'AVAILABLE':
-            return True
-        return False
+            return "Yes"
+        if agreement_status == "ERROR":
+            return model_info['agreementAvailability']['errorMessage']
+        return "No"
     except:
         #print(form_status)
-        return False
+        return "No"
         
 async def get_model_status(session, key, secret, region, model_name, form_info):
     model_info = await bedrock_model_available(session, key, secret, region, f"anthropic.{model_name}")
     model_status = is_model_working(form_info, model_info)
-    if model_status:
-        return region, model_name
+    if model_status == "Yes":
+        return region, model_name, ""
+    elif model_status == "No":
+        return None, model_name, ""
     else:
-        return None, None
+        return None, model_name, model_status
         
 async def check_bedrock_claude_status(key, secret):
     regions = ['us-east-1', 'us-west-2', 'eu-central-1', 'eu-west-3', 'ap-northeast-1', 'ap-southeast-2'] # currently these regions aren't "gated" nor having only "low context" models
@@ -601,10 +605,13 @@ async def check_bedrock_claude_status(key, secret):
             for model in models:
                 tasks.append(get_model_status(session, key, secret, region, model, form_info))
         results = await asyncio.gather(*tasks)
-        for region, model_name in results:
+        for region, model_name, msg in results:
             if region and model_name:
                 models[model_name].append(region)
-
+            elif form_info.get('message') == "Operation not allowed" and "Operation not allowed" not in models[model_name]:
+                models[model_name].append('Operation not allowed')
+            elif msg and msg not in models[model_name]:
+                models[model_name].append(msg)
     return models
 
 def check_aws_billing(session):