xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-25 11:37:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf: 超额时的提示信息

Browse Source
This commit is contained in:
Rock Chin
2023-01-08 20:43:37 +08:00
parent 5256d3c718
commit 9d31d8b071
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/qqbot/process.py
+6 -5
View File
@@ -307,18 +307,19 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
reply = ["[bot]err:调用API失败，请重试或联系作者，或等待修复"] reply = ["[bot]err:调用API失败，请重试或联系作者，或等待修复"]
except openai.error.RateLimitError as e: except openai.error.RateLimitError as e:
# 尝试切换api-key # 尝试切换api-key
current_tokens_amt = pkg.utils.context.get_openai_manager().key_mgr.get_fee( current_key_name = pkg.utils.context.get_openai_manager().key_mgr.get_key_name(
pkg.utils.context.get_openai_manager().key_mgr.get_using_key()) pkg.utils.context.get_openai_manager().key_mgr.using_key
)
pkg.utils.context.get_openai_manager().key_mgr.set_current_exceeded() pkg.utils.context.get_openai_manager().key_mgr.set_current_exceeded()
switched, name = pkg.utils.context.get_openai_manager().key_mgr.auto_switch() switched, name = pkg.utils.context.get_openai_manager().key_mgr.auto_switch()
if not switched: if not switched:
mgr.notify_admin("API调用额度超限({}),无可用api_key,请向OpenAI账户充值或在config.py中更换api_key".format( mgr.notify_admin("api-key调用额度超限({}),无可用api_key,请向OpenAI账户充值或在config.py中更换api_key".format(
current_tokens_amt)) current_key_name))
reply = ["[bot]err:API调用额度超额，请联系作者，或等待修复"] reply = ["[bot]err:API调用额度超额，请联系作者，或等待修复"]
else: else:
openai.api_key = pkg.utils.context.get_openai_manager().key_mgr.get_using_key() openai.api_key = pkg.utils.context.get_openai_manager().key_mgr.get_using_key()
mgr.notify_admin("API调用额度超限({}),接口报错,已切换到{}".format(current_tokens_amt, name)) mgr.notify_admin("api-key调用额度超限({}),接口报错,已切换到{}".format(current_key_name, name))
reply = ["[bot]err:API调用额度超额，已自动切换，请重新发送消息"] reply = ["[bot]err:API调用额度超额，已自动切换，请重新发送消息"]
continue continue
except openai.error.InvalidRequestError as e: except openai.error.InvalidRequestError as e: