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

add: 添加ollama调用AI chat逻辑

Browse Source
This commit is contained in:
taylor
2024-10-13 16:31:50 +08:00
parent 6786394b79
commit 03dc086d14
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
maubot_llmplus/aibot.py
View File

@@ -115,7 +115,7 @@ class AiBotPlugin(Plugin):
try: try:
await event.mark_read() await event.mark_read()
await self.client.set_typing(event.room_id, timeout=99999) await self.client.set_typing(event.room_id, timeout=99999)
platform = self.get_platform() platform = self.get_ai_platform()
chat_completion = platform.create_chat_completion(event) chat_completion = platform.create_chat_completion(event)
# ai gpt调用 # ai gpt调用
# 关闭typing提示 # 关闭typing提示
@@ -133,7 +133,7 @@ class AiBotPlugin(Plugin):
return None return None
async def get_platform(self) -> Platform: def get_ai_platform(self) -> Platform:
use_platform = self.config['use_platform'] use_platform = self.config['use_platform']
if use_platform == 'local_ai': if use_platform == 'local_ai':
type = self.config['platforms']['local_ai']['type'] type = self.config['platforms']['local_ai']['type']