mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 20:14:36 +00:00
209f16af76
* style: remove necessary imports * style: fix F841 * style: fix F401 * style: fix F811 * style: fix E402 * style: fix E721 * style: fix E722 * style: fix E722 * style: fix F541 * style: ruff format * style: all passed * style: add ruff in deps * style: more ignores in ruff.toml * style: add pre-commit
167 lines
7.0 KiB
Python
167 lines
7.0 KiB
Python
from __future__ import annotations
|
|
|
|
import typing
|
|
|
|
|
|
from ...core import entities as core_entities
|
|
from .. import entities
|
|
from .. import stage
|
|
from ...plugin import events
|
|
from ...platform.types import message as platform_message
|
|
|
|
|
|
@stage.stage_class('ResponseWrapper')
|
|
class ResponseWrapper(stage.PipelineStage):
|
|
"""回复包装阶段
|
|
|
|
把回复的 message 包装成人类识读的形式。
|
|
|
|
改写：
|
|
- resp_message_chain
|
|
"""
|
|
|
|
async def initialize(self, pipeline_config: dict):
|
|
pass
|
|
|
|
async def process(
|
|
self,
|
|
query: core_entities.Query,
|
|
stage_inst_name: str,
|
|
) -> typing.AsyncGenerator[entities.StageProcessResult, None]:
|
|
"""处理"""
|
|
|
|
# 如果 resp_messages[-1] 已经是 MessageChain 了
|
|
if isinstance(query.resp_messages[-1], platform_message.MessageChain):
|
|
query.resp_message_chain.append(query.resp_messages[-1])
|
|
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.CONTINUE, new_query=query
|
|
)
|
|
|
|
else:
|
|
if query.resp_messages[-1].role == 'command':
|
|
query.resp_message_chain.append(
|
|
query.resp_messages[-1].get_content_platform_message_chain(
|
|
prefix_text='[bot] '
|
|
)
|
|
)
|
|
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.CONTINUE, new_query=query
|
|
)
|
|
elif query.resp_messages[-1].role == 'plugin':
|
|
query.resp_message_chain.append(
|
|
query.resp_messages[-1].get_content_platform_message_chain()
|
|
)
|
|
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.CONTINUE, new_query=query
|
|
)
|
|
else:
|
|
if query.resp_messages[-1].role == 'assistant':
|
|
result = query.resp_messages[-1]
|
|
session = await self.ap.sess_mgr.get_session(query)
|
|
|
|
reply_text = ''
|
|
|
|
if result.content: # 有内容
|
|
reply_text = str(result.get_content_platform_message_chain())
|
|
|
|
# ============= 触发插件事件 ===============
|
|
event_ctx = await self.ap.plugin_mgr.emit_event(
|
|
event=events.NormalMessageResponded(
|
|
launcher_type=query.launcher_type.value,
|
|
launcher_id=query.launcher_id,
|
|
sender_id=query.sender_id,
|
|
session=session,
|
|
prefix='',
|
|
response_text=reply_text,
|
|
finish_reason='stop',
|
|
funcs_called=[
|
|
fc.function.name for fc in result.tool_calls
|
|
]
|
|
if result.tool_calls is not None
|
|
else [],
|
|
query=query,
|
|
)
|
|
)
|
|
if event_ctx.is_prevented_default():
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.INTERRUPT,
|
|
new_query=query,
|
|
)
|
|
else:
|
|
if event_ctx.event.reply is not None:
|
|
query.resp_message_chain.append(
|
|
platform_message.MessageChain(event_ctx.event.reply)
|
|
)
|
|
|
|
else:
|
|
query.resp_message_chain.append(
|
|
result.get_content_platform_message_chain()
|
|
)
|
|
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.CONTINUE,
|
|
new_query=query,
|
|
)
|
|
|
|
if (
|
|
result.tool_calls is not None and len(result.tool_calls) > 0
|
|
): # 有函数调用
|
|
function_names = [tc.function.name for tc in result.tool_calls]
|
|
|
|
reply_text = f'调用函数 {".".join(function_names)}...'
|
|
|
|
query.resp_message_chain.append(
|
|
platform_message.MessageChain(
|
|
[platform_message.Plain(reply_text)]
|
|
)
|
|
)
|
|
|
|
if query.pipeline_config['output']['misc'][
|
|
'track-function-calls'
|
|
]:
|
|
event_ctx = await self.ap.plugin_mgr.emit_event(
|
|
event=events.NormalMessageResponded(
|
|
launcher_type=query.launcher_type.value,
|
|
launcher_id=query.launcher_id,
|
|
sender_id=query.sender_id,
|
|
session=session,
|
|
prefix='',
|
|
response_text=reply_text,
|
|
finish_reason='stop',
|
|
funcs_called=[
|
|
fc.function.name for fc in result.tool_calls
|
|
]
|
|
if result.tool_calls is not None
|
|
else [],
|
|
query=query,
|
|
)
|
|
)
|
|
|
|
if event_ctx.is_prevented_default():
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.INTERRUPT,
|
|
new_query=query,
|
|
)
|
|
else:
|
|
if event_ctx.event.reply is not None:
|
|
query.resp_message_chain.append(
|
|
platform_message.MessageChain(
|
|
event_ctx.event.reply
|
|
)
|
|
)
|
|
|
|
else:
|
|
query.resp_message_chain.append(
|
|
platform_message.MessageChain(
|
|
[platform_message.Plain(reply_text)]
|
|
)
|
|
)
|
|
|
|
yield entities.StageProcessResult(
|
|
result_type=entities.ResultType.CONTINUE,
|
|
new_query=query,
|
|
)
|