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:17:27 +08:00
parent eb05892eb1
commit 342eae6464
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
maubot_llmplus/local_paltform.py
View File

@@ -3,7 +3,8 @@ import json
from mautrix.types import MessageEvent from mautrix.types import MessageEvent
from mautrix.util.config import BaseProxyConfig from mautrix.util.config import BaseProxyConfig
from maubot_llmplus import platforms import maubot_llmplus
import maubot_llmplus.platforms
from maubot_llmplus.platforms import Platform, ChatCompletion from maubot_llmplus.platforms import Platform, ChatCompletion
@@ -16,7 +17,7 @@ class Ollama(Platform):
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion: async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
full_context = [] full_context = []
context = platforms.get_context(evt) context = maubot_llmplus.platforms.get_context(evt)
full_context.extend(list(context)) full_context.extend(list(context))
endpoint = f"{self.url}/api/chat" endpoint = f"{self.url}/api/chat"