mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-15 19:22:24 +08:00
122 lines
3.8 KiB
Python
122 lines
3.8 KiB
Python
from __future__ import annotations
|
||
|
||
import enum
|
||
import typing
|
||
import datetime
|
||
import asyncio
|
||
|
||
import pydantic
|
||
import mirai
|
||
|
||
from ..provider import entities as llm_entities
|
||
from ..provider.modelmgr import entities
|
||
from ..provider.sysprompt import entities as sysprompt_entities
|
||
from ..provider.tools import entities as tools_entities
|
||
from ..platform import adapter as msadapter
|
||
|
||
|
||
class LauncherTypes(enum.Enum):
|
||
"""一个请求的发起者类型"""
|
||
|
||
PERSON = 'person'
|
||
"""私聊"""
|
||
|
||
GROUP = 'group'
|
||
"""群聊"""
|
||
|
||
|
||
class Query(pydantic.BaseModel):
|
||
"""一次请求的信息封装"""
|
||
|
||
query_id: int
|
||
"""请求ID，添加进请求池时生成"""
|
||
|
||
launcher_type: LauncherTypes
|
||
"""会话类型，platform处理阶段设置"""
|
||
|
||
launcher_id: int
|
||
"""会话ID，platform处理阶段设置"""
|
||
|
||
sender_id: int
|
||
"""发送者ID，platform处理阶段设置"""
|
||
|
||
message_event: mirai.MessageEvent
|
||
"""事件，platform收到的原始事件"""
|
||
|
||
message_chain: mirai.MessageChain
|
||
"""消息链，platform收到的原始消息链"""
|
||
|
||
adapter: msadapter.MessageSourceAdapter
|
||
"""消息平台适配器对象，单个app中可能启用了多个消息平台适配器，此对象表明发起此query的适配器"""
|
||
|
||
session: typing.Optional[Session] = None
|
||
"""会话对象，由前置处理器阶段设置"""
|
||
|
||
messages: typing.Optional[list[llm_entities.Message]] = []
|
||
"""历史消息列表，由前置处理器阶段设置"""
|
||
|
||
prompt: typing.Optional[sysprompt_entities.Prompt] = None
|
||
"""情景预设内容，由前置处理器阶段设置"""
|
||
|
||
user_message: typing.Optional[llm_entities.Message] = None
|
||
"""此次请求的用户消息对象，由前置处理器阶段设置"""
|
||
|
||
use_model: typing.Optional[entities.LLMModelInfo] = None
|
||
"""使用的模型，由前置处理器阶段设置"""
|
||
|
||
use_funcs: typing.Optional[list[tools_entities.LLMFunction]] = None
|
||
"""使用的函数，由前置处理器阶段设置"""
|
||
|
||
resp_messages: typing.Optional[list[llm_entities.Message]] | typing.Optional[list[mirai.MessageChain]] = []
|
||
"""由Process阶段生成的回复消息对象列表"""
|
||
|
||
resp_message_chain: typing.Optional[list[mirai.MessageChain]] = None
|
||
"""回复消息链，从resp_messages包装而得"""
|
||
|
||
# ======= 内部保留 =======
|
||
current_stage: "pkg.pipeline.stagemgr.StageInstContainer" = None
|
||
|
||
class Config:
|
||
arbitrary_types_allowed = True
|
||
|
||
|
||
class Conversation(pydantic.BaseModel):
|
||
"""对话，包含于 Session 中，一个 Session 可以有多个历史 Conversation，但只有一个当前使用的 Conversation"""
|
||
|
||
prompt: sysprompt_entities.Prompt
|
||
|
||
messages: list[llm_entities.Message]
|
||
|
||
create_time: typing.Optional[datetime.datetime] = pydantic.Field(default_factory=datetime.datetime.now)
|
||
|
||
update_time: typing.Optional[datetime.datetime] = pydantic.Field(default_factory=datetime.datetime.now)
|
||
|
||
use_model: entities.LLMModelInfo
|
||
|
||
use_funcs: typing.Optional[list[tools_entities.LLMFunction]]
|
||
|
||
|
||
class Session(pydantic.BaseModel):
|
||
"""会话，一个 Session 对应一个 {launcher_type.value}_{launcher_id}"""
|
||
launcher_type: LauncherTypes
|
||
|
||
launcher_id: int
|
||
|
||
sender_id: typing.Optional[int] = 0
|
||
|
||
use_prompt_name: typing.Optional[str] = 'default'
|
||
|
||
using_conversation: typing.Optional[Conversation] = None
|
||
|
||
conversations: typing.Optional[list[Conversation]] = []
|
||
|
||
create_time: typing.Optional[datetime.datetime] = pydantic.Field(default_factory=datetime.datetime.now)
|
||
|
||
update_time: typing.Optional[datetime.datetime] = pydantic.Field(default_factory=datetime.datetime.now)
|
||
|
||
semaphore: typing.Optional[asyncio.Semaphore] = None
|
||
"""当前会话的信号量，用于限制并发"""
|
||
|
||
class Config:
|
||
arbitrary_types_allowed = True
|