sasaki-saku's picture
Update app.py
a6df1a0 verified
raw
history blame
6.28 kB
import gradio as gr
import anthropic
from api_usage import get_subscription, check_key_availability, check_key_ant_availability, check_key_gemini_availability, check_key_azure_availability, get_azure_status, get_azure_deploy, check_key_mistral_availability, check_mistral_quota
def sort_key(key):
_key = key.strip()
if _key.startswith("sk-ant-"):
return get_key_ant_info(_key)
elif _key.startswith("sk-"):
return get_key_oai_info(_key)
elif _key.startswith("AIzaSy"):
return get_key_gemini_info(_key)
elif len(_key.split(':')) == 2 and _key.split(':')[1].islower() and len(_key.split(':')[1]) == 32 and "openai.azure.com" not in _key.split(':')[1]:
endpoint = f"{_key.split(':')[0]}.openai.azure.com"
api_key = _key.split(':')[1]
return get_key_azure_info(endpoint, api_key)
elif "openai.azure.com" in _key.split(';')[0]:
endpoint = _key.split(';')[0]
api_key = _key.split(';')[1]
return get_key_azure_info(endpoint, api_key)
elif _key.startswith("AKIA") and len(_key.split(':')[0]) == 20 and _key.split(':')[0].isupper():
return get_key_aws_info(_key)
elif len(_key) == 32:
return get_key_mistral_info(_key)
else:
return not_supported(_key)
def get_key_oai_info(key):
# Return a dictionary containing key information
key_avai = check_key_availability(key)
info_dict = {#"account_name": "",
"key_type": "OpenAI",
"key_availability": True if key_avai else False,
"gpt4_availability": "",
"gpt4_32k_availability": "",
"default_org": "",
"org_description": "",
"organization": "",
"models": "",
"requests_per_minute": "",
"tokens_per_minute": "",
#"tokens_per_minute_left": "",
"quota": ""}
if key_avai:
info = get_subscription(key, key_avai)
info_dict["gpt4_availability"] = info["has_gpt4"]
info_dict["gpt4_32k_availability"] = info["has_gpt4_32k"]
info_dict["default_org"] = info["default_org"]
info_dict["org_description"] = info["org_description"]
info_dict["organization"] = info["organization"]
info_dict["models"] = info["models"]
info_dict["requests_per_minute"] = info["rpm"]
info_dict["tokens_per_minute"] = info["tpm"]
info_dict["quota"] = info["quota"]
return info_dict
def get_key_ant_info(key):
# Return a dictionary containing key information
ant = anthropic.Anthropic(api_key=key)
key_avai = check_key_ant_availability(ant)
info_dict = {#"account_name": "",
"key_type": "Anthropic Claude",
"key_availability": key_avai[0],
"status": key_avai[1],
"filter_response": key_avai[2]}
return info_dict
def get_key_gemini_info(key):
key_avai = check_key_gemini_availability(key)
info_dict = {#"account_name": "",
"key_type": "Google Gemini",
"key_availability": key_avai[0],
"models": key_avai[1]}
return info_dict
def get_key_gemini_info(key):
key_avai = check_key_gemini_availability(key)
info_dict = {#"account_name": "",
"key_type": "Google Gemini",
"key_availability": key_avai[0],
"models": key_avai[1]}
return info_dict
def get_key_azure_info(endpoint, api_key):
key_avai = check_key_azure_availability(endpoint, api_key)
info_dict = {#"account_name": "",
"key_type": "Microsoft Azure OpenAI",
"key_availability": key_avai[0],
"gpt35_availability": "",
"gpt4_availability": "",
"gpt4_turbo_availability": "",
"gpt4_32k_availability": "",
"moderation_status": "",
"models": "",
"deployments": ""}
if key_avai[0]:
azure_deploy = get_azure_deploy(endpoint, api_key)
status = get_azure_status(endpoint, api_key, azure_deploy)
info_dict["gpt35_availability"] = status[1]
info_dict["gpt4_availability"] = status[2]
info_dict["gpt4_turbo_availability"] = status[3]
info_dict["gpt4_32k_availability"] = status[4]
info_dict["moderation_status"] = status[0]
info_dict["models"] = key_avai[1]
info_dict["deployments"] = azure_deploy
return info_dict
def get_key_mistral_info(key):
key_avai = check_key_mistral_availability(key)
info_dict = {#"account_name": "",
"key_type": "Mistral AI",
"key_availability": key_avai,
"has_quota": ""}
if key_avai:
info_dict['has_quota'] = check_mistral_quota(key)
return info_dict
def get_key_aws_info(key):
info_dict = {#"account_name": "",
"key_type": "Amazon AWS Claude",
"status": " Soon™ "}
return info_dict
def not_supported(key):
info_dict = {#"account_name": "",
"key_type": "Not supported",
"status": ""}
return info_dict
def clear_inputs(text):
return ""
with gr.Blocks() as demo:
gr.Markdown('''
# API Key Status Checker
### Supported: OpenAI/Anthropic/Google Gemini/Microsoft Azure/Mistral AI
*(Based on shaocongma, CncAnon1 and Drago key checkers)*
*Azure endpoint's format:*
*(https://)YOUR_RESOURCE_NAME.openai.azure.com;YOUR_API_KEY*
*YOUR_RESOURCE_NAME:YOUR_API_KEY*
''')
with gr.Row():
with gr.Column():
key = gr.Textbox(lines=1, max_lines=1, label="API Key")
with gr.Row():
clear_button = gr.Button("Clear")
submit_button = gr.Button("Submit", variant="primary")
with gr.Column():
info = gr.JSON(label="OpenAI/Anthropic/Gemini/Azure API Key Information")
clear_button.click(fn=clear_inputs, inputs=[key], outputs=[key])
submit_button.click(fn=sort_key, inputs=[key], outputs=[info], api_name="sort_key")
demo.launch()