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

add: 添加chatgpt chat api逻辑

Browse Source
This commit is contained in:
taylor
2024-10-14 00:16:19 +08:00
parent 8c16faa34a
commit 0f7a2c4b33
4 changed files with 59 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
maubot_llmplus/local_paltform.py
View File

@@ -1,14 +1,15 @@
import json
from typing import List
from aiohttp import ClientSession
from maubot import Plugin
from mautrix.types import MessageEvent
from mautrix.util.config import BaseProxyConfig
import maubot_llmplus
import maubot_llmplus.platforms
from maubot_llmplus.platforms import Platform, ChatCompletion
from maubot_llmplus.plugin import AbsExtraConfigPlugin
class Ollama(Platform):
@@ -18,7 +19,7 @@ class Ollama(Platform):
super().__init__(config, http)
self.chat_api = '/api/chat'
async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
async def create_chat_completion(self, plugin: AbsExtraConfigPlugin, evt: MessageEvent) -> ChatCompletion:
full_context = []
context = await maubot_llmplus.platforms.get_context(plugin, self, evt)
full_context.extend(list(context))
@@ -59,5 +60,5 @@ class LmStudio(Platform):
super().__init__(config, http)
pass
async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
async def create_chat_completion(self, plugin: AbsExtraConfigPlugin, evt: MessageEvent) -> ChatCompletion:
pass