Update app.py

app.py

@@ -5,10 +5,10 @@ from api_usage import get_subscription, check_key_availability, check_key_ant_av
 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):    
+    if 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):
+    if re.match(re.compile("sk-ant-api03-[a-zA-Z0-9\-_]{93}AA"), _key) or (_key.startswith("sk-ant-") and len(_key) == 93) or (len(_key) == 89 and re.match(re.compile("sk-[a-zA-Z0-9]{86}"), _key)):
         return await get_key_ant_info(_key, rate_limit, claude_model)
         
     if _key.startswith("sk-"):    
@@ -226,10 +226,10 @@ def get_key_openrouter_info(key):
         info_dict['limit'] = key_avai[1]['limit']
         info_dict['limit_remaining'] = key_avai[1]['limit_remaining']
         info_dict['usage'] = f"${format(key_avai[1]['usage'], '.4f')}"
-        info_dict['balance'] = f"${format(models_info[0], '.4f')}"
+        info_dict['balance'] = f"${format(models_info[0], '.4f')}" if models_info[0] else f"${key_avai[2]/60} (estimated)" 
         info_dict['rate_limit_per_minite'] = key_avai[2]
-        info_dict['4_turbo_per_request_tokens_limit'] = models_info[1]['openai/gpt-4-turbo-preview']
-        info_dict['sonnet_per_request_tokens_limit'] = models_info[1]['anthropic/claude-3-sonnet:beta']
+        info_dict['4_turbo_per_request_tokens_limit'] = models_info[1]['openai/gpt-4o']
+        info_dict['sonnet_per_request_tokens_limit'] = models_info[1]['anthropic/claude-3.5-sonnet:beta']
         info_dict['opus_per_request_tokens_limit'] = models_info[1]['anthropic/claude-3-opus:beta']
     else:
         info_dict['usage'] = key_avai[1]
@@ -287,7 +287,6 @@ with gr.Blocks() as demo:
     # OpenAI/Anthropic/Gemini/Azure/Mistral/Replicate/AWS Claude/OpenRouter/Vertex AI(GCP Anthropic)/Groq/NovelAI/ElevenLabs API Key Status Checker
     
     *(Based on shaocongma, CncAnon1, su, Drago, kingbased key checkers)*
-
     AWS credential's format: AWS_ACCESS_KEY_ID:AWS_SECRET_ACCESS_KEY (root might not be accurate)
     
     Azure endpoint's format: YOUR_RESOURCE_NAME:YOUR_API_KEY or (https://)YOUR_RESOURCE_NAME.openai.azure.com;YOUR_API_KEY
@@ -300,6 +299,8 @@ with gr.Blocks() as demo:
     'claude-3-sonnet-20240229',
     'claude-3-opus-20240229',
     'claude-3-5-sonnet-20240620',
+    'claude-3-5-sonnet-20241022',
+    'claude-3-5-haiku-20241022'
     ]
     with gr.Row():
         with gr.Column():
@@ -310,8 +311,8 @@ with gr.Blocks() as demo:
                 clear_button = gr.Button("Clear")
                 submit_button = gr.Button("Submit", variant="primary")
         with gr.Column():
-            info = gr.JSON(label="API Key Information")
+            info = gr.JSON(label="API Key Information", open=True)
 
     clear_button.click(fn=clear_inputs, inputs=[key], outputs=[key])
     submit_button.click(fn=sort_key, inputs=[key, rate_limit, claude_model], outputs=[info], api_name="sort_key")
-demo.launch()
+demo.launch()