mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
from __future__ import annotations
|
|
import abc
|
|
import typing
|
|
|
|
import mirai
|
|
from mirai.models.message import MessageComponent
|
|
|
|
from ...core import app
|
|
from ...core import entities as core_entities
|
|
|
|
|
|
preregistered_strategies: list[typing.Type[LongTextStrategy]] = []
|
|
|
|
|
|
def strategy_class(
|
|
name: str
|
|
) -> typing.Callable[[typing.Type[LongTextStrategy]], typing.Type[LongTextStrategy]]:
|
|
"""长文本处理策略类装饰器
|
|
|
|
Args:
|
|
name (str): 策略名称
|
|
|
|
Returns:
|
|
typing.Callable[[typing.Type[LongTextStrategy]], typing.Type[LongTextStrategy]]: 装饰器
|
|
"""
|
|
|
|
def decorator(cls: typing.Type[LongTextStrategy]) -> typing.Type[LongTextStrategy]:
|
|
assert issubclass(cls, LongTextStrategy)
|
|
|
|
cls.name = name
|
|
|
|
preregistered_strategies.append(cls)
|
|
|
|
return cls
|
|
|
|
return decorator
|
|
|
|
|
|
class LongTextStrategy(metaclass=abc.ABCMeta):
|
|
"""长文本处理策略抽象类
|
|
"""
|
|
|
|
name: str
|
|
|
|
ap: app.Application
|
|
|
|
def __init__(self, ap: app.Application):
|
|
self.ap = ap
|
|
|
|
async def initialize(self):
|
|
pass
|
|
|
|
@abc.abstractmethod
|
|
async def process(self, message: str, query: core_entities.Query) -> list[MessageComponent]:
|
|
"""处理长文本
|
|
|
|
在 platform.json 中配置 long-text-process 字段，只要 文本长度超过了 threshold 就会调用此方法
|
|
|
|
Args:
|
|
message (str): 消息
|
|
query (core_entities.Query): 此次请求的上下文对象
|
|
|
|
Returns:
|
|
list[mirai.models.messages.MessageComponent]: 转换后的 YiriMirai 消息组件列表
|
|
"""
|
|
return []
|