refactor(prompt): 优化Prompt组件注入逻辑并简化代码

将Prompt组件的注入逻辑从`Prompt.format`方法前置到`PromptManager.get_prompt_async`和`create_prompt_async`中。这使得注入时机更早，逻辑更清晰，并允许在获取Prompt时就能动态传入参数以影响注入内容。主要变更： - `PromptManager`: `get_prompt_async`现在负责处理组件注入，并接收可选的`parameters`参数。`format_prompt`相应地传递参数。 - `create_prompt_async`: 现在也支持在创建时进行动态注入。 - `Prompt.format`: 移除了原有的组件注入逻辑，简化了方法实现。 - `PromptComponentManager`: 重构为直接从全局`component_registry`获取组件，移除了自身的注册和存储逻辑，减少了状态管理的复杂性。 - `plan_filter.py`: 删除了大量冗余和重复的代码块，包括主动聊天模式的独立逻辑和旧的历史消息构建方式。
2025-10-19 13:32:49 +08:00
parent 50a6c2de58
commit 73383f8246
3 changed files with 104 additions and 141 deletions
--- a/src/chat/utils/prompt.py
+++ b/src/chat/utils/prompt.py
@@ -58,7 +58,7 @@ class PromptContext:
                context_id = None

            previous_context = self._current_context
-            token = self._current_context_var.set(context_id) if context_id else None
+            token = self._current_context_var.set(context_id) if context_id else None # type: ignore
        else:
            previous_context = self._current_context
            token = None
@@ -111,16 +111,42 @@ class PromptManager:
        async with self._context.async_scope(message_id):
            yield self

-    async def get_prompt_async(self, name: str) -> "Prompt":
-        """异步获取提示模板"""
+    async def get_prompt_async(self, name: str, parameters: PromptParameters | None = None) -> "Prompt":
+        """
+        异步获取提示模板，并动态注入插件内容
+        """
+        original_prompt = None
        context_prompt = await self._context.get_prompt_async(name)
        if context_prompt is not None:
            logger.debug(f"从上下文中获取提示词: {name} {context_prompt}")
-            return context_prompt
-
-        if name not in self._prompts:
+            original_prompt = context_prompt
+        elif name in self._prompts:
+            original_prompt = self._prompts[name]
+        else:
            raise KeyError(f"Prompt '{name}' not found")
-        return self._prompts[name]
+
+        # 动态注入插件内容
+        if original_prompt.name:
+            # 确保我们有有效的parameters实例
+            params_for_injection = parameters or original_prompt.parameters
+
+            components_prefix = await prompt_component_manager.execute_components_for(
+                injection_point=original_prompt.name, params=params_for_injection
+            )
+            logger.info(components_prefix)
+            if components_prefix:
+                logger.info(f"为'{name}'注入插件内容: \n{components_prefix}")
+                # 创建一个新的临时Prompt实例，不进行注册
+                new_template = f"{components_prefix}\n\n{original_prompt.template}"
+                temp_prompt = Prompt(
+                    template=new_template,
+                    name=original_prompt.name,
+                    parameters=original_prompt.parameters,
+                    should_register=False,  # 确保不重新注册
+                )
+                return temp_prompt
+
+        return original_prompt

    def generate_name(self, template: str) -> str:
        """为未命名的prompt生成名称"""
@@ -142,7 +168,9 @@ class PromptManager:

    async def format_prompt(self, name: str, **kwargs) -> str:
        """格式化提示模板"""
-        prompt = await self.get_prompt_async(name)
+        # 提取parameters用于注入
+        parameters = kwargs.get("parameters")
+        prompt = await self.get_prompt_async(name, parameters=parameters)
        result = prompt.format(**kwargs)
        return result

@@ -230,23 +258,14 @@ class Prompt:

        start_time = time.time()
        try:
-            # 1. 从组件管理器获取注入内容
-            components_prefix = ""
-            if self.name:
-                components_prefix = await prompt_component_manager.execute_components_for(
-                    injection_point=self.name, params=self.parameters
-                )
-
-            # 2. 构建核心上下文数据
+            # 1. 构建核心上下文数据
            context_data = await self._build_context_data()

-            # 3. 格式化主模板
+            # 2. 格式化主模板
            main_formatted_prompt = await self._format_with_context(context_data)

-            # 4. 拼接组件内容和主模板内容
+            # 3. 拼接组件内容和主模板内容 (逻辑已前置到 get_prompt_async)
            result = main_formatted_prompt
-            if components_prefix:
-                result = f"{components_prefix}\n\n{main_formatted_prompt}"

            total_time = time.time() - start_time
            logger.debug(f"Prompt构建完成，模式: {self.parameters.prompt_mode}, 耗时: {total_time:.2f}s")
@@ -406,9 +425,13 @@ class Prompt:
        if not self.parameters.message_list_before_now_long:
            return

+        target_user_id = ""
+        if self.parameters.target_user_info:
+            target_user_id = self.parameters.target_user_info.get("user_id") or ""
+
        read_history_prompt, unread_history_prompt = await self._build_s4u_chat_history_prompts(
            self.parameters.message_list_before_now_long,
-            self.parameters.target_user_info.get("user_id") if self.parameters.target_user_info else "",
+            target_user_id,
            self.parameters.sender,
            self.parameters.chat_id,
        )
@@ -434,11 +457,14 @@ class Prompt:

            # 创建临时生成器实例来使用其方法
            temp_generator = await get_replyer(None, chat_id, request_type="prompt_building")
-            return await temp_generator.build_s4u_chat_history_prompts(
-                message_list_before_now, target_user_id, sender, chat_id
-            )
+            if temp_generator:
+                return await temp_generator.build_s4u_chat_history_prompts(
+                    message_list_before_now, target_user_id, sender, chat_id
+                )
+            return "", ""
        except Exception as e:
            logger.error(f"构建S4U历史消息prompt失败: {e}")
+            return "", ""

    async def _build_expression_habits(self) -> dict[str, Any]:
        """构建表达习惯"""
@@ -525,10 +551,10 @@ class Prompt:
                running_memories, instant_memory = await asyncio.gather(*memory_tasks, return_exceptions=True)

                # 处理可能的异常结果
-                if isinstance(running_memories, Exception):
+                if isinstance(running_memories, BaseException):
                    logger.warning(f"长期记忆查询失败: {running_memories}")
                    running_memories = []
-                if isinstance(instant_memory, Exception):
+                if isinstance(instant_memory, BaseException):
                    logger.warning(f"即时记忆查询失败: {instant_memory}")
                    instant_memory = None

@@ -1042,8 +1068,24 @@ def create_prompt(
 async def create_prompt_async(
    template: str, name: str | None = None, parameters: PromptParameters | None = None, **kwargs
 ) -> Prompt:
-    """异步创建Prompt实例"""
-    prompt = create_prompt(template, name, parameters, **kwargs)
+    """异步创建Prompt实例，并动态注入插件内容"""
+    # 确保有可用的parameters实例
+    final_params = parameters or PromptParameters(**kwargs)
+
+    # 动态注入插件内容
+    if name:
+        components_prefix = await prompt_component_manager.execute_components_for(
+            injection_point=name, params=final_params
+        )
+        if components_prefix:
+            logger.debug(f"为'{name}'注入插件内容: \n{components_prefix}")
+            template = f"{components_prefix}\n\n{template}"
+
+    # 使用可能已修改的模板创建实例
+    prompt = create_prompt(template, name, final_params)
+
+    # 如果在特定上下文中，则异步注册
    if global_prompt_manager._context._current_context:
        await global_prompt_manager._context.register_async(prompt)
+
    return prompt