refactor: 独立出预处理阶段

2026-08-30 14:17:14 +00:00 · 2024-02-01 16:35:00 +08:00
parent 976a9de39c
commit 532a713355
9 changed files with 156 additions and 90 deletions
@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+from .. import stage, entities, stagemgr
+from ...core import entities as core_entities
+from ...provider import entities as llm_entities
+from ...plugin import events
+
+
+@stage.stage_class("PreProcessor")
+class PreProcessor(stage.PipelineStage):
+    """预处理器
+    """
+
+    async def process(
+        self,
+        query: core_entities.Query,
+        stage_inst_name: str,
+    ) -> entities.StageProcessResult:
+        """处理
+        """
+        session = await self.ap.sess_mgr.get_session(query)
+
+        conversation = await self.ap.sess_mgr.get_conversation(session)
+
+        # 从会话取出消息和情景预设到query
+        query.session = session
+        query.prompt = conversation.prompt.copy()
+        query.messages = conversation.messages.copy()
+
+        query.user_message = llm_entities.Message(
+            role='user',
+            content=str(query.message_chain).strip()
+        )
+
+        query.use_model = conversation.use_model
+
+        query.use_funcs = conversation.use_funcs
+
+        # =========== 触发事件 PromptPreProcessing
+        session = query.session
+
+        event_ctx = await self.ap.plugin_mgr.emit_event(
+            event=events.PromptPreProcessing(
+                session_name=f'{session.launcher_type.value}_{session.launcher_id}',
+                default_prompt=query.prompt.messages,
+                prompt=query.messages,
+                query=query
+            )
+        )
+
+        query.prompt.messages = event_ctx.event.default_prompt
+        query.messages = event_ctx.event.prompt
+
+        # 根据模型max_tokens剪裁
+
+        return entities.StageProcessResult(
+            result_type=entities.ResultType.CONTINUE,
+            new_query=query
+        )