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 17:19:46 +08:00
parent 603a9fbde9
commit e75406e3d6
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
maubot_llmplus/aibot.py
View File

@@ -117,6 +117,7 @@ class AiBotPlugin(Plugin):
await self.client.set_typing(event.room_id, timeout=99999) await self.client.set_typing(event.room_id, timeout=99999)
platform = self.get_ai_platform() platform = self.get_ai_platform()
chat_completion = await platform.create_chat_completion(self, event) chat_completion = await platform.create_chat_completion(self, event)
self.log.debug(f"发送结果 {chat_completion.message}, {chat_completion.model}, {chat_completion.finish_reason}")
# ai gpt调用 # ai gpt调用
# 关闭typing提示 # 关闭typing提示
await self.client.set_typing(event.room_id, timeout=0) await self.client.set_typing(event.room_id, timeout=0)