mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
39 lines
907 B
Python
39 lines
907 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import enum
|
||
|
|
import typing
|
||
|
|
|
||
|
|
import pydantic
|
||
|
|
import mirai
|
||
|
|
import mirai.models.message as mirai_message
|
||
|
|
|
||
|
|
from ..core import entities
|
||
|
|
|
||
|
|
|
||
|
|
class ResultType(enum.Enum):
|
||
|
|
|
||
|
|
CONTINUE = enum.auto()
|
||
|
|
"""继续流水线"""
|
||
|
|
|
||
|
|
INTERRUPT = enum.auto()
|
||
|
|
"""中断流水线"""
|
||
|
|
|
||
|
|
|
||
|
|
class StageProcessResult(pydantic.BaseModel):
|
||
|
|
|
||
|
|
result_type: ResultType
|
||
|
|
|
||
|
|
new_query: entities.Query
|
||
|
|
|
||
|
|
user_notice: typing.Optional[typing.Union[str, list[mirai_message.MessageComponent], mirai.MessageChain, None]] = []
|
||
|
|
"""只要设置了就会发送给用户"""
|
||
|
|
|
||
|
|
admin_notice: typing.Optional[typing.Union[str, list[mirai_message.MessageComponent], mirai.MessageChain, None]] = []
|
||
|
|
"""只要设置了就会发送给管理员"""
|
||
|
|
|
||
|
|
console_notice: typing.Optional[str] = ''
|
||
|
|
"""只要设置了就会输出到控制台"""
|
||
|
|
|
||
|
|
debug_notice: typing.Optional[str] = ''
|
||
|
|
|