Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat: 添加管理员命令以全局设置OpenAI的api_base、api_key和GPT模型 #2356

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

feat: 添加管理员命令以全局设置OpenAI的api_base、api_key和GPT模型 #2356

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 34 additions & 1 deletion plugins/godcmd/godcmd.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,20 @@
"alias": ["debug", "调试模式", "DEBUG"],
"desc": "开启机器调试日志",
},
"admin_set_openai_api_base": {
"alias": ["admin_set_openai_api_base"],
"args": ["api_key"],
"desc": "全局设置OpenAI的api_base",
},
"admin_set_openai_api_key": {
"alias": ["admin_set_openai_api_key"],
"args": ["api_key"],
"desc": "全局设置OpenAI的api_key",
},
"admin_set_gpt_model": {
"alias": ["admin_set_gpt_model"],
"desc": "全局设置模型",
},
}


Expand Down Expand Up @@ -336,7 +350,7 @@ def on_handle_context(self, e_context: EventContext):
ok, result = True, "服务已恢复"
elif cmd == "reconf":
load_config()
ok, result = True, "配置已重载"
ok, result = True, "配置已重载（私有or管理配置，需要重新设置）"
elif cmd == "resetall":
if bottype in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE, const.LINKAI,
const.BAIDU, const.XUNFEI, const.QWEN, const.GEMINI, const.ZHIPU_AI, const.MOONSHOT]:
Expand Down Expand Up @@ -418,6 +432,25 @@ def on_handle_context(self, e_context: EventContext):
ok, result = False, "请提供插件名"
else:
ok, result = PluginManager().update_plugin(args[0])
elif cmd == "admin_set_openai_api_base":
if len(args) == 1:
conf()["open_ai_api_base"] = args[0]
ok, result = True, "全局设置OpenAI的api_base为" + args[0]
else:
ok, result = False, "请提供一个api_base"
elif cmd == "admin_set_openai_api_key":
if len(args) == 1:
conf()["open_ai_api_key"] = args[0]
ok, result = True, "全局设置OpenAI的api_key为" + args[0]
else:
ok, result = False, "请提供一个api_key"
elif cmd == "admin_set_gpt_model":
if len(args) == 1:
conf()["model"] = self.model_mapping(args[0])
Bridge().reset_bot()
ok, result = True, "全局设置GPT模型为" + args[0]
else:
ok, result = False, "请提供一个GPT模型"
logger.debug("[Godcmd] admin command: %s by %s" % (cmd, user))
else:
ok, result = False, "需要管理员权限才能执行该指令"
Expand Down