xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-09-16 14:57:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(pipeline): apply returned plugin event contexts

Browse Source
This commit is contained in:
RockChinQ
2026-09-07 23:51:34 +08:00
parent 5d0657b1db
commit 812eb09ee4
8 changed files with 329 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/langbot/pkg/pipeline/pipelinemgr.py
+11 -19
View File
@@ -122,20 +122,14 @@ class RuntimePipeline:
self.placement_generation = self.execution_context.placement_generation
# Extract bound plugins and MCP servers from extensions_preferences
extensions_prefs = normalize_extension_preferences(
pipeline_entity.extensions_preferences
)
extensions_prefs = normalize_extension_preferences(pipeline_entity.extensions_preferences)
self.enable_all_plugins = extensions_prefs['enable_all_plugins'] is True
self.enable_all_mcp_servers = (
extensions_prefs['enable_all_mcp_servers'] is True
)
self.enable_all_mcp_servers = extensions_prefs['enable_all_mcp_servers'] is True
pipeline_config = pipeline_entity.config or {}
runner_config: dict[str, typing.Any] = {}
runner_id = RunnerConfigResolver.resolve_runner_id(pipeline_config)
if runner_id:
resolved = RunnerConfigResolver.resolve_runner_config(
pipeline_config, runner_id
)
resolved = RunnerConfigResolver.resolve_runner_config(pipeline_config, runner_id)
if isinstance(resolved, dict):
runner_config = resolved
self.mcp_resource_attachments = runner_config.get(
@@ -154,10 +148,7 @@ class RuntimePipeline:
# None indicates to use all available plugins
self.bound_plugins = None
else:
self.bound_plugins = [
f'{plugin["author"]}/{plugin["name"]}'
for plugin in extensions_prefs['plugins']
]
self.bound_plugins = [f'{plugin["author"]}/{plugin["name"]}' for plugin in extensions_prefs['plugins']]
if self.enable_all_mcp_servers:
# None indicates to use all available MCP servers
@@ -385,9 +376,7 @@ class RuntimePipeline:
# Get runner name from pipeline config
runner_name = None
if query.pipeline_config:
runner_name = RunnerConfigResolver.resolve_runner_id(
query.pipeline_config
)
runner_name = RunnerConfigResolver.resolve_runner_id(query.pipeline_config)
# Record query start and store message_id
message_id = ''
@@ -442,6 +431,11 @@ class RuntimePipeline:
)
return
# The Runtime returns a deserialized event, not the original Query.
# Carry plugin message edits into the following Pipeline stages.
query.message_chain = event_ctx.event.message_chain
query.message_event.message_chain = query.message_chain
self.ap.logger.debug(f'Processing query {query.query_id}')
await self._execute_from_stage(0, query)
@@ -669,9 +663,7 @@ class PipelineManager:
stage_containers: list[StageInstContainer] = []
for stage_name in pipeline_entity.stages:
if stage_name not in self.stage_dict:
self.ap.logger.warning(
f'Pipeline stage {stage_name} is not registered; skipping'
)
self.ap.logger.warning(f'Pipeline stage {stage_name} is not registered; skipping')
continue
stage_containers.append(StageInstContainer(inst_name=stage_name, inst=self.stage_dict[stage_name](self.ap)))
src/langbot/pkg/plugin/connector.py
+4
View File
@@ -2114,6 +2114,10 @@ class PluginRuntimeConnector(ManagedRuntimeConnector):
event_ctx = context.EventContext.model_validate(result['event_context'])
emitted_plugins.extend(result.get('emitted_plugins', []))
response_sources.extend(result.get('response_sources', []))
if event_ctx.is_prevented_postorder():
break
if query is not None:
event_ctx.event.query = query
event_ctx._emitted_plugins = emitted_plugins
event_ctx._response_sources = response_sources