Update app.py

app.py

@@ -8,7 +8,7 @@ async def sort_key(key, rate_limit, claude_model):
     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):
+    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 "No info"
         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]
@@ -300,6 +300,7 @@ with gr.Blocks() as demo:
     'claude-3-sonnet-20240229',
     'claude-3-opus-20240229',
     'claude-3-5-sonnet-20240620',
+    'claude-3-5-sonnet-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()