帮我写一个chatgpt api key查询余额的python脚本脚本支持多key查询key文件在rootkeytxt

当然可以！以下是一个实现这个功能的Python脚本：

import openai

def query_balance(api_key):
    openai.api_key = api_key
    response = openai.Account.retrieve()
    balance = response['data'][0]['attributes']['balance']
    return balance

# 读取key文件
with open('/root/key.txt', 'r') as file:
    api_keys = file.readlines()

# 去除每个key的换行符
api_keys = [key.strip() for key in api_keys]

# 查询余额
for key in api_keys:
    balance = query_balance(key)
    print(f"API Key: {key}\nBalance: {balance}\n")

首先，确保你已经安装了OpenAI的Python库（openai）。可以使用以下命令安装它：

pip install openai

然后，将上述代码保存到一个名为query_balance.py的文件中。确保key.txt文件位于/root/key.txt路径下，并在该文件中按行存储API密钥。

运行脚本后，它将逐个查询每个API密钥的余额，并将结果打印出来