mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-26 03:44:58 +08:00
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
import random
|
|
import asyncio
|
|
|
|
|
|
from ...platform.types import events as platform_events
|
|
from ...platform.types import message as platform_message
|
|
|
|
from ...provider import entities as llm_entities
|
|
|
|
|
|
|
|
from .. import stage, entities
|
|
from ...core import entities as core_entities
|
|
|
|
|
|
@stage.stage_class('SendResponseBackStage')
|
|
class SendResponseBackStage(stage.PipelineStage):
|
|
"""发送响应消息"""
|
|
|
|
async def process(self, query: core_entities.Query, stage_inst_name: str) -> entities.StageProcessResult:
|
|
"""处理"""
|
|
|
|
random_range = (
|
|
query.pipeline_config['output']['force-delay']['min'],
|
|
query.pipeline_config['output']['force-delay']['max'],
|
|
)
|
|
|
|
random_delay = random.uniform(*random_range)
|
|
|
|
self.ap.logger.debug('根据规则强制延迟回复: %s s', random_delay)
|
|
|
|
await asyncio.sleep(random_delay)
|
|
|
|
if query.pipeline_config['output']['misc']['at-sender'] and isinstance(
|
|
query.message_event, platform_events.GroupMessage
|
|
):
|
|
query.resp_message_chain[-1].insert(0, platform_message.At(query.message_event.sender.id))
|
|
|
|
quote_origin = query.pipeline_config['output']['misc']['quote-origin']
|
|
|
|
has_chunks = any(isinstance(msg, llm_entities.MessageChunk) for msg in query.resp_messages)
|
|
# TODO 命令与流式的兼容性问题
|
|
if await query.adapter.is_stream_output_supported() and has_chunks:
|
|
is_final = [msg.is_final for msg in query.resp_messages][0]
|
|
await query.adapter.reply_message_chunk(
|
|
message_source=query.message_event,
|
|
bot_message=query.resp_messages[-1],
|
|
message=query.resp_message_chain[-1],
|
|
quote_origin=quote_origin,
|
|
is_final=is_final,
|
|
)
|
|
else:
|
|
await query.adapter.reply_message(
|
|
message_source=query.message_event,
|
|
message=query.resp_message_chain[-1],
|
|
quote_origin=quote_origin,
|
|
)
|
|
|
|
return entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
|