Spaces:
Running
Running
| 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() |