37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from aiohttp import ClientSession
|
|
from maubot import Plugin
|
|
from mautrix.types import MessageEvent
|
|
from mautrix.util.config import BaseProxyConfig
|
|
|
|
from maubot_llmplus.platforms import Platform, ChatCompletion
|
|
|
|
|
|
class OpenAi(Platform):
|
|
|
|
def __init__(self, config: BaseProxyConfig, name: str, http: ClientSession) -> None:
|
|
super().__init__(config, name, http)
|
|
|
|
async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
|
|
# 获取系统提示词
|
|
# 获取额外的其他角色的提示词： role: user role: system
|
|
|
|
pass
|
|
|
|
def get_type(self) -> str:
|
|
return "openai"
|
|
|
|
|
|
class Anthropic(Platform):
|
|
|
|
def __init__(self, config: BaseProxyConfig, name: str, http: ClientSession) -> None:
|
|
super().__init__(config, name, http)
|
|
|
|
async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
|
|
# 获取系统提示词
|
|
# 获取额外的其他角色的提示词： role: user role: system
|
|
|
|
pass
|
|
|
|
def get_type(self) -> str:
|
|
return "anthropic"
|