add: 添加切换模型命令
This commit is contained in:
@@ -18,6 +18,7 @@ from maubot_llmplus.thrid_platform import OpenAi, Anthropic
|
||||
|
||||
|
||||
class Config(BaseProxyConfig):
|
||||
|
||||
def do_update(self, helper: ConfigUpdateHelper) -> None:
|
||||
helper.copy("allowed_users")
|
||||
helper.copy("use_platform")
|
||||
@@ -27,6 +28,7 @@ class Config(BaseProxyConfig):
|
||||
helper.copy("system_prompt")
|
||||
helper.copy("platforms")
|
||||
helper.copy("additional_prompt")
|
||||
super._cur_model = self['platforms'][self['use_platform']]['model']
|
||||
|
||||
|
||||
class AiBotPlugin(AbsExtraConfigPlugin):
|
||||
@@ -182,7 +184,7 @@ class AiBotPlugin(AbsExtraConfigPlugin):
|
||||
models = platform.list_models()
|
||||
if f"- {arg_elements[1]}" in models:
|
||||
self.log.debug(f"switch model: {arg_elements[1]}")
|
||||
self.model = arg_elements[1]
|
||||
self.config._cur_model = arg_elements[1]
|
||||
await event.react("✅")
|
||||
else:
|
||||
await event.reply("not found valid model")
|
||||
|
||||
@@ -40,7 +40,7 @@ class Platform:
|
||||
self.http = http
|
||||
self.config = config['platforms'][self.get_type()]
|
||||
self.url = self.config['url']
|
||||
self.model = self.config['model']
|
||||
self.model = config._cur_model
|
||||
self.max_words = self.config['max_words']
|
||||
self.api_key = self.config['api_key']
|
||||
self.max_context_messages = self.config['max_context_messages']
|
||||
|
||||
@@ -5,18 +5,13 @@ class AbsExtraConfigPlugin(Plugin):
|
||||
default_username: str
|
||||
user_id: str
|
||||
|
||||
model: str;
|
||||
|
||||
async def start(self) -> None:
|
||||
await super().start()
|
||||
self.default_username = await self.client.get_displayname(self.client.mxid)
|
||||
self.user_id = self.client.parse_user_id(self.client.mxid)[0]
|
||||
self.model = self.config['platforms'][self.config['use_platform']]['model']
|
||||
|
||||
|
||||
def get_bot_name(self) -> str:
|
||||
return self.config['name'] or \
|
||||
self.default_username or \
|
||||
self.user_id
|
||||
|
||||
def get_use_model(self) -> str:
|
||||
return self.model
|
||||
|
||||
Reference in New Issue
Block a user