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:36:54 +08:00
parent 03dc086d14
commit 3fb79a12ad
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
maubot_llmplus/local_paltform.py
View File

@@ -1,5 +1,6 @@
import json
from aiohttp import ClientSession
from mautrix.types import MessageEvent
from mautrix.util.config import BaseProxyConfig
@@ -11,8 +12,8 @@ from maubot_llmplus.platforms import Platform, ChatCompletion
class Ollama(Platform):
chat_api: str
def __init__(self, config: BaseProxyConfig) -> None:
super().__init__(config)
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http)
self.chat_api = '/api/chat'
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:

9
maubot_llmplus/thrid_platform.py
View File

@@ -1,3 +1,4 @@
from aiohttp import ClientSession
from mautrix.types import MessageEvent
from mautrix.util.config import BaseProxyConfig
@@ -6,8 +7,8 @@ from maubot_llmplus.platforms import Platform, ChatCompletion
class OpenAi(Platform):
def __init__(self, config: BaseProxyConfig) -> None:
super().__init__(config)
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http)
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
# 获取系统提示词
@@ -21,8 +22,8 @@ class OpenAi(Platform):
class Anthropic(Platform):
def __init__(self, config: BaseProxyConfig) -> None:
super().__init__(config)
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http)
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
# 获取系统提示词