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

add: 添加切换模型命令

Browse Source
This commit is contained in:
taylor
2024-10-14 01:11:07 +08:00
parent 2081e78308
commit 3451e3591f
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
maubot_llmplus/aibot.py
View File

@@ -28,7 +28,8 @@ class Config(BaseProxyConfig):
helper.copy("system_prompt") helper.copy("system_prompt")
helper.copy("platforms") helper.copy("platforms")
helper.copy("additional_prompt") helper.copy("additional_prompt")
super._cur_model = self['platforms'][str(self['use_platform'])]['model']
super._cur_model = helper.base['platforms'][helper.base['use_platform']]['model']
class AiBotPlugin(AbsExtraConfigPlugin): class AiBotPlugin(AbsExtraConfigPlugin):

3
maubot_llmplus/platforms.py
View File

@@ -8,6 +8,7 @@ from maubot import Plugin
from mautrix.types import MessageEvent, EncryptedEvent from mautrix.types import MessageEvent, EncryptedEvent
from mautrix.util.config import BaseProxyConfig from mautrix.util.config import BaseProxyConfig
from maubot_llmplus.aibot import Config
from maubot_llmplus.plugin import AbsExtraConfigPlugin from maubot_llmplus.plugin import AbsExtraConfigPlugin
""" """
@@ -36,7 +37,7 @@ class Platform:
system_prompt: str system_prompt: str
max_context_messages: int max_context_messages: int
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: Config, http: ClientSession) -> None:
self.http = http self.http = http
self.config = config['platforms'][self.get_type()] self.config = config['platforms'][self.get_type()]
self.url = self.config['url'] self.url = self.config['url']