mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
d9fa1cbb06
* add cmd config && fix bugs * perf: use `get` * update bansess fix block match rule * perf: comment for access-control session str --------- Co-authored-by: Junyan Qin <rockchinq@gmail.com>
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from ...core import entities as core_entities
|
|
from . import handler
|
|
from .handlers import chat, command
|
|
from .. import entities
|
|
from .. import stage
|
|
|
|
|
|
@stage.stage_class('MessageProcessor')
|
|
class Processor(stage.PipelineStage):
|
|
"""请求实际处理阶段
|
|
|
|
通过命令处理器和聊天处理器处理消息。
|
|
|
|
改写：
|
|
- resp_messages
|
|
"""
|
|
|
|
cmd_handler: handler.MessageHandler
|
|
|
|
chat_handler: handler.MessageHandler
|
|
|
|
async def initialize(self, pipeline_config: dict):
|
|
self.cmd_handler = command.CommandHandler(self.ap)
|
|
self.chat_handler = chat.ChatMessageHandler(self.ap)
|
|
|
|
await self.cmd_handler.initialize()
|
|
await self.chat_handler.initialize()
|
|
|
|
async def process(
|
|
self,
|
|
query: core_entities.Query,
|
|
stage_inst_name: str,
|
|
) -> entities.StageProcessResult:
|
|
"""Process"""
|
|
message_text = str(query.message_chain).strip()
|
|
|
|
self.ap.logger.info(
|
|
f'Processing request from {query.launcher_type.value}_{query.launcher_id} ({query.query_id}): {message_text}'
|
|
)
|
|
|
|
async def generator():
|
|
cmd_prefix = self.ap.instance_config.data['command']['prefix']
|
|
cmd_enable = self.ap.instance_config.data['command'].get('enable', True)
|
|
|
|
if cmd_enable and any(message_text.startswith(prefix) for prefix in cmd_prefix):
|
|
handler_to_use = self.cmd_handler
|
|
else:
|
|
handler_to_use = self.chat_handler
|
|
|
|
async for result in handler_to_use.handle(query):
|
|
yield result
|
|
|
|
return generator()
|