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:11:20 +08:00
parent 3ecca0b40a
commit eb05892eb1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
maubot_llmplus/platforms.py
View File

@@ -8,7 +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 AiBotPlugin import maubot_llmplus.aibot
""" """
AI响应对象 AI响应对象
@@ -59,7 +59,7 @@ class Platform:
async def get_context(plugin: AiBotPlugin, evt: MessageEvent) -> deque: async def get_context(plugin: maubot_llmplus.AiBotPlugin, evt: MessageEvent) -> deque:
# 创建系统提示词上下文 # 创建系统提示词上下文
system_context = deque() system_context = deque()
# 生成当前时间 # 生成当前时间
@@ -122,7 +122,7 @@ async def get_context(plugin: AiBotPlugin, evt: MessageEvent) -> deque:
async def generate_context_messages(plugin: AiBotPlugin, evt: MessageEvent) -> Generator[MessageEvent, None, None]: async def generate_context_messages(plugin: maubot_llmplus.AiBotPlugin, evt: MessageEvent) -> Generator[MessageEvent, None, None]:
yield evt yield evt
if plugin.config['reply_in_thread']: if plugin.config['reply_in_thread']:
while evt.content.relates_to.in_reply_to: while evt.content.relates_to.in_reply_to: