Update api_usage.py

api_usage.py

@@ -459,23 +459,22 @@ def check_policy(iam, username):
     except botocore.exceptions.ClientError as error:
         return False, error.response['Error']['Code']
 
-def invoke_claude(session, region):
+def invoke_claude(session, region, modelId):
     try:
         bedrock_runtime = session.client("bedrock-runtime", region_name=region)
         body = json.dumps({
             "prompt": "\n\nHuman:\n\nAssistant:",
             "max_tokens_to_sample": 0
         })
-        response = bedrock_runtime.invoke_model(body=body, modelId="anthropic.claude-v2")
+        response = bedrock_runtime.invoke_model(body=body, modelId=modelId)
     except bedrock_runtime.exceptions.ValidationException as error:
         #print(error.response['Error'])
         return region
     except bedrock_runtime.exceptions.AccessDeniedException as error:
         #print(error.response['Error'])
-        return
     except bedrock_runtime.exceptions.ResourceNotFoundException as error:
         #print(error.response['Error'])
-        return
+        return invoke_claude(session, region, "anthropic.claude-v2:1")
     except Exception as e:
         #print(e)
         return
@@ -484,7 +483,7 @@ def check_bedrock_invoke(session):
     regions = ['us-east-1', 'us-west-2', 'eu-central-1', 'ap-southeast-1', 'ap-northeast-1']
     enable_region = []
     with concurrent.futures.ThreadPoolExecutor() as executer:
-        futures = [executer.submit(invoke_claude, session, region) for region in regions]            
+        futures = [executer.submit(invoke_claude, session, region, "anthropic.claude-v2") for region in regions]            
         for future in concurrent.futures.as_completed(futures):
             if future.result():
                 enable_region.append(future.result())