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

add

Browse Source
This commit is contained in:
taylorxie
2026-03-09 21:34:14 +08:00
parent 04077c7f12
commit d5e818b334
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
maubot_llmplus/aibot.py
View File

@@ -10,7 +10,8 @@ from mautrix.util.config import BaseProxyConfig, ConfigUpdateHelper
from maubot_llmplus.local_paltform import Ollama, LmStudio from maubot_llmplus.local_paltform import Ollama, LmStudio
from maubot_llmplus.platforms import Platform from maubot_llmplus.platforms import Platform
from maubot_llmplus.plugin import AbsExtraConfigPlugin, Config from maubot_llmplus.plugin import AbsExtraConfigPlugin, Config
from maubot_llmplus.thrid_platform import OpenAi, Anthropic, XAi from maubot_llmplus.thrid_platform import OpenAi, Anthropic, XAi, Deepseek
class AiBotPlugin(AbsExtraConfigPlugin): class AiBotPlugin(AbsExtraConfigPlugin):
@@ -157,6 +158,8 @@ class AiBotPlugin(AbsExtraConfigPlugin):
return Anthropic(self.config, self.http) return Anthropic(self.config, self.http)
if use_platform == 'xai': if use_platform == 'xai':
return XAi(self.config, self.http) return XAi(self.config, self.http)
if use_platform == 'deepseek':
return Deepseek(self.config, self.http)
if use_platform == 'local_ai#ollama': if use_platform == 'local_ai#ollama':
return Ollama(self.config, self.http) return Ollama(self.config, self.http)
if use_platform == 'local_ai#lmstudio': if use_platform == 'local_ai#lmstudio':
@@ -295,7 +298,7 @@ class AiBotPlugin(AbsExtraConfigPlugin):
self.config.cur_model = self.config['platforms'][argus.split("#")[0]]['model'] self.config.cur_model = self.config['platforms'][argus.split("#")[0]]['model']
await event.react("✅") await event.react("✅")
# 如果是openai或者是claude # 如果是openai或者是claude
elif argus == 'openai' or argus == 'anthropic' or argus == 'xai': elif argus == 'openai' or argus == 'anthropic' or argus == 'xai' or argus == 'deepseek':
if argus == self.config.cur_platform: if argus == self.config.cur_platform:
await event.reply(f"current ai platform has be {argus}") await event.reply(f"current ai platform has be {argus}")
pass pass