add: 添加ollama调用AI chat逻辑

This commit is contained in:
taylor
2024-10-13 15:37:12 +08:00
parent 78acb679ee
commit e5d4e52bc0
8 changed files with 275 additions and 13 deletions

View File

@@ -0,0 +1,39 @@
from collections import deque
from typing import List
from maubot import Plugin
from mautrix.types import MessageEvent
from mautrix.util.config import BaseProxyConfig
from maubot_llmplus import AiBotPlugin
from maubot_llmplus.llm.platforms import Platform, ChatCompletion
class OpenAi(Platform):
def __init__(self, config: BaseProxyConfig) -> None:
super().__init__(config)
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
# 获取系统提示词
# 获取额外的其他角色的提示词: role: user role: system
pass
def get_type(self) -> str:
return "openai"
class Anthropic(Platform):
def __init__(self, config: BaseProxyConfig) -> None:
super().__init__(config)
async def create_chat_completion(self, evt: MessageEvent) -> ChatCompletion:
# 获取系统提示词
# 获取额外的其他角色的提示词: role: user role: system
pass
def get_type(self) -> str:
return "anthropic"