QChatGPT/pkg/core/entities.py

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.requester 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
    """适配器对象"""

    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]] = []
    """由provider生成的回复消息对象列表"""

    resp_message_chain: typing.Optional[mirai.MessageChain] = None
    """回复消息链，从resp_messages包装而得"""

    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}_{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