code-repository/maubot-llmplus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add: 增加切换ai平台命令

Browse Source
This commit is contained in:
taylor
2024-10-14 13:46:32 +08:00
parent 69fa0c0a50
commit 400d628e9f
2 changed files with 46 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
maubot_llmplus/plugin.py
View File

@@ -23,7 +23,8 @@ class AbsExtraConfigPlugin(Plugin):
class Config(BaseProxyConfig):
_cur_model: str
cur_model: str
cur_platform: str
def do_update(self, helper: ConfigUpdateHelper) -> None:
helper.copy("allowed_users")
@@ -35,4 +36,6 @@ class Config(BaseProxyConfig):
helper.copy("platforms")
helper.copy("additional_prompt")
self._cur_model = helper.base['platforms'][helper.base['use_platform']]['model']
self.cur_platform = helper.base['use_platform'] if helper.base['use_platform'] != 'local_ai' else \
f"{helper.base['use_platform']}#{helper.base['platforms'][helper.base['local_ai']['type']]}"
self.cur_model = helper.base['platforms'][helper.base['use_platform']]['model']