Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 import requests, re
 from api_usage import get_subscription, check_key_availability, check_key_ant_availability, check_ant_rate_limit, check_key_gemini_availability, check_key_azure_availability, get_azure_status, get_azure_deploy, check_key_mistral_availability, check_mistral_quota, check_key_replicate_availability, check_key_aws_availability, check_key_or_availability, check_key_or_limits, check_gcp_anthropic, check_groq_status, check_nai_status, check_elevenlabs_status
 
-async def sort_key(key, rate_limit, claude_opus):
+async def sort_key(key, rate_limit, claude_model):
     _key = key.strip()
     
     if _key.startswith("sk-or-v1-") and re.match(re.compile("sk-or-v1-[a-z0-9]{64}"), _key):    
         return get_key_openrouter_info(_key)
         
     if _key.startswith("sk-ant-") and re.match(re.compile("sk-ant-api03-[a-zA-Z0-9\-_]{93}AA"), _key):
-        return await get_key_ant_info(_key, rate_limit, claude_opus)
+        return await get_key_ant_info(_key, rate_limit, claude_model)
         
     if _key.startswith("sk-"):    
         return get_key_oai_info(_key)
@@ -88,9 +88,9 @@ def get_key_oai_info(key):
         info_dict["quota"] = info["quota"]
     return info_dict
 
-async def get_key_ant_info(key, rate_limit, claude_opus):
+async def get_key_ant_info(key, rate_limit, claude_model):
     # Return a dictionary containing key information
-    key_avai = check_key_ant_availability(key, claude_opus)
+    key_avai = await check_key_ant_availability(key, claude_model)
     info_dict = {#"account_name": "",
                  "key_type": "Anthropic Claude",
                  "key_availability": key_avai[0],
@@ -108,7 +108,7 @@ async def get_key_ant_info(key, rate_limit, claude_opus):
     info_dict["tier"] = key_avai[7]
         
     if rate_limit:
-        rate = await check_ant_rate_limit(key)
+        rate = await check_ant_rate_limit(key, claude_model)
         info_dict["concurrent_rate_limit"] = rate
         
     return info_dict
@@ -295,12 +295,17 @@ with gr.Blocks() as demo:
     GCP format: PROJECT_ID:CLIENT_EMAIL:PRIVATE_KEY (including \\n)
      | or refresh token: PROJECT_ID:CLIENT_ID:CLIENT_SECRET:REFRESH_TOKEN
     ''')
-
+    claude_options = [
+    'claude-3-haiku-20240307',
+    'claude-3-sonnet-20240229',
+    'claude-3-opus-20240229',
+    'claude-3-5-sonnet-20240620',
+    ]
     with gr.Row():
         with gr.Column():
             key = gr.Textbox(lines=1, max_lines=5, label="API Key")
-            rate_limit = gr.Checkbox(label="Check concurrent rate limit (API Claude)")
-            claude_opus = gr.Checkbox(label="Use claude-3-opus for filter_response check? (default: claude-3-haiku)")
+            claude_model = gr.Dropdown(claude_options, value="claude-3-haiku-20240307", label="Claude API model", info="model for filter_response and concurrent check")
+            rate_limit = gr.Checkbox(label="Check concurrent rate limit (API Claude, experimental)")
             with gr.Row():
                 clear_button = gr.Button("Clear")
                 submit_button = gr.Button("Submit", variant="primary")
@@ -308,5 +313,5 @@ with gr.Blocks() as demo:
             info = gr.JSON(label="API Key Information")
 
     clear_button.click(fn=clear_inputs, inputs=[key], outputs=[key])
-    submit_button.click(fn=sort_key, inputs=[key, rate_limit, claude_opus], outputs=[info], api_name="sort_key")
-demo.launch()
+    submit_button.click(fn=sort_key, inputs=[key, rate_limit, claude_model], outputs=[info], api_name="sort_key")
+demo.launch()