feat: adapt more events

This commit is contained in:
Junyan Qin
2025-07-02 11:04:03 +08:00
parent c246470b37
commit ee3da8aa17
8 changed files with 37 additions and 14 deletions

View File

@@ -4,9 +4,10 @@ import datetime
from .. import stage, entities
from langbot_plugin.api.entities.builtin.provider import message as provider_message
from ...plugin import events
import langbot_plugin.api.entities.events as events
import langbot_plugin.api.entities.builtin.platform.message as platform_message
import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query
import langbot_plugin.api.entities.context as event_context
@stage.stage_class('PreProcessor')
@@ -108,7 +109,7 @@ class PreProcessor(stage.PipelineStage):
query.user_message = provider_message.Message(role='user', content=content_list)
# =========== 触发事件 PromptPreProcessing
event_ctx = await self.ap.plugin_mgr.emit_event(
event_ctx = event_context.EventContext(
event=events.PromptPreProcessing(
session_name=f'{query.session.launcher_type.value}_{query.session.launcher_id}',
default_prompt=query.prompt.messages,
@@ -117,6 +118,12 @@ class PreProcessor(stage.PipelineStage):
)
)
event_ctx_result = await self.ap.plugin_connector.handler.emit_event(
event_ctx.model_dump(serialize_as_any=True)
)
event_ctx = event_context.EventContext.parse_from_dict(event_ctx_result['event_context'])
query.prompt.messages = event_ctx.event.default_prompt
query.messages = event_ctx.event.prompt