refactor(agent-runner): remove host context windowing

src/langbot/pkg/pipeline/msgtrun/msgtrun.py

@@ -1,39 +0,0 @@
-from __future__ import annotations
-
-from .. import stage, entities
-from . import truncator
-from ...utils import importutil
-from ...agent.runner.config_migration import ConfigMigration
-import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query
-from . import truncators
-
-importutil.import_modules_in_pkg(truncators)
-
-
-@stage.stage_class('ConversationMessageTruncator')
-class ConversationMessageTruncator(stage.PipelineStage):
-    """Conversation message truncator
-
-    Used to truncate the conversation message chain to adapt to the LLM message length limit.
-    """
-
-    trun: truncator.Truncator
-
-    async def initialize(self, pipeline_config: dict):
-        use_method = 'round'
-
-        for trun in truncator.preregistered_truncators:
-            if trun.name == use_method:
-                self.trun = trun(self.ap)
-                break
-        else:
-            raise ValueError(f'Unknown truncator: {use_method}')
-
-    async def process(self, query: pipeline_query.Query, stage_inst_name: str) -> entities.StageProcessResult:
-        """处理"""
-        if ConfigMigration.resolve_runner_id(query.pipeline_config):
-            return entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)
-
-        query = await self.trun.truncate(query)
-
-        return entities.StageProcessResult(result_type=entities.ResultType.CONTINUE, new_query=query)

src/langbot/pkg/pipeline/msgtrun/round_policy.py

@@ -1,34 +0,0 @@
-"""Shared max-round message window helpers for Pipeline behavior."""
-from __future__ import annotations
-
-import typing
-
-
-DEFAULT_MAX_ROUND = 10
-
-
-def get_max_round(config: dict[str, typing.Any]) -> typing.Any:
-    """Return the configured Pipeline max-round value."""
-    return config.get('max-round', DEFAULT_MAX_ROUND)
-
-
-def select_max_round_messages(
-    messages: list[typing.Any] | None,
-    max_round: typing.Any,
-) -> list[typing.Any]:
-    """Select a bounded recent message window by user-round count."""
-    if not messages:
-        return []
-
-    temp_messages: list[typing.Any] = []
-    current_round = 0
-
-    for msg in messages[::-1]:
-        if current_round < max_round:
-            temp_messages.append(msg)
-            if getattr(msg, 'role', None) == 'user':
-                current_round += 1
-        else:
-            break
-
-    return temp_messages[::-1]

src/langbot/pkg/pipeline/msgtrun/truncator.py

@@ -1,56 +0,0 @@
-from __future__ import annotations
-
-import typing
-import abc
-
-from ...core import app
-import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query
-
-preregistered_truncators: list[typing.Type[Truncator]] = []
-
-
-def truncator_class(
-    name: str,
-) -> typing.Callable[[typing.Type[Truncator]], typing.Type[Truncator]]:
-    """截断器类装饰器
-
-    Args:
-        name (str): 截断器名称
-
-    Returns:
-        typing.Callable[[typing.Type[Truncator]], typing.Type[Truncator]]: 装饰器
-    """
-
-    def decorator(cls: typing.Type[Truncator]) -> typing.Type[Truncator]:
-        assert issubclass(cls, Truncator)
-
-        cls.name = name
-
-        preregistered_truncators.append(cls)
-
-        return cls
-
-    return decorator
-
-
-class Truncator(abc.ABC):
-    """消息截断器基类"""
-
-    name: str
-
-    ap: app.Application
-
-    def __init__(self, ap: app.Application):
-        self.ap = ap
-
-    async def initialize(self):
-        pass
-
-    @abc.abstractmethod
-    async def truncate(self, query: pipeline_query.Query) -> pipeline_query.Query:
-        """截断
-
-        一般只需要操作query.messages，也可以扩展操作query.prompt, query.user_message。
-        请勿操作其他字段。
-        """
-        pass

src/langbot/pkg/pipeline/msgtrun/truncators/round.py

@@ -1,29 +0,0 @@
-from __future__ import annotations
-
-from .. import truncator
-import langbot_plugin.api.entities.builtin.pipeline.query as pipeline_query
-from ....agent.runner.config_migration import ConfigMigration
-from ..round_policy import (
-    get_max_round,
-    select_max_round_messages,
-)
-
-
-@truncator.truncator_class('round')
-class RoundTruncator(truncator.Truncator):
-    """Truncate the conversation message chain to adapt to the LLM message length limit."""
-
-    async def truncate(self, query: pipeline_query.Query) -> pipeline_query.Query:
-        """截断"""
-        runner_id = ConfigMigration.resolve_runner_id(query.pipeline_config)
-        if runner_id:
-            runner_config = ConfigMigration.resolve_runner_config(query.pipeline_config, runner_id)
-        else:
-            runner_config = query.pipeline_config.get('msg-truncate', {}).get('round', {})
-
-        query.messages = select_max_round_messages(
-            query.messages,
-            get_max_round(runner_config),
-        )
-
-        return query

src/langbot/pkg/pipeline/pipelinemgr.py

@@ -28,7 +28,6 @@ from . import (
     wrapper,
     preproc,
     ratelimit,
-    msgtrun,
 )
 
 importutil.import_modules_in_pkgs(
@@ -42,7 +41,6 @@ importutil.import_modules_in_pkgs(
         wrapper,
         preproc,
         ratelimit,
-        msgtrun,
     ]
 )
 
@@ -438,6 +436,9 @@ class PipelineManager:
         # initialize stage containers according to pipeline_entity.stages
         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')
+                continue
             stage_containers.append(StageInstContainer(inst_name=stage_name, inst=self.stage_dict[stage_name](self.ap)))
 
         for stage_container in stage_containers: