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:46:13 +08:00
parent 2a8054b36a
commit 9199aacc69
4 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
maubot_llmplus/local_paltform.py
View File

@@ -1,6 +1,7 @@
import json
from aiohttp import ClientSession
from maubot import Plugin
from mautrix.types import MessageEvent
from mautrix.util.config import BaseProxyConfig
@@ -16,9 +17,9 @@ class Ollama(Platform):
super().__init__(config, http)
self.chat_api = '/api/chat'
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
full_context = []
context = maubot_llmplus.platforms.get_context(evt)
context = maubot_llmplus.platforms.get_context(plugin, evt)
full_context.extend(list(context))
endpoint = f"{self.url}/api/chat"
@@ -50,5 +51,5 @@ class LmStudio(Platform):
super().__init__(config, http)
pass
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
pass