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 22:24:09 +08:00
parent d5e818b334
commit 22cb30bde0
3 changed files with 91 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
maubot_llmplus/aibot.py
View File

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