From 80210cfb58cc4d67895f7a615b7b7b562f9fb9da Mon Sep 17 00:00:00 2001 From: minecraft1024a Date: Fri, 14 Nov 2025 13:01:05 +0800 Subject: [PATCH] =?UTF-8?q?feat(prompt):=20=E5=A2=9E=E5=BC=BA=20get=5Fcore?= =?UTF-8?q?=5Fprompt=5Fcontents=20=E6=96=B9=E6=B3=95=E4=BB=A5=E6=94=AF?= =?UTF-8?q?=E6=8C=81=E8=8E=B7=E5=8F=96=E5=8D=95=E4=B8=AA=E6=8F=90=E7=A4=BA?= =?UTF-8?q?=E8=AF=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 为 `get_core_prompt_contents` 方法添加了可选的 `prompt_name` 参数。 当指定 `prompt_name` 时，该方法现在只返回对应提示词模板的内容。如果未指定，则返回所有核心提示词。 BREAKING CHANGE: `get_core_prompt_contents` 方法的返回类型从 `dict[str, str]` 更改为 `list[list[str]]`，以统一处理单个和多个提示词的返回结果，确保接口行为的一致性。 --- src/chat/utils/prompt_component_manager.py | 25 ++++++++++++++++++---- 1 file changed, 21 insertions(+), 4 deletions(-) diff --git a/src/chat/utils/prompt_component_manager.py b/src/chat/utils/prompt_component_manager.py index de6d0689c..5cab12c02 100644 --- a/src/chat/utils/prompt_component_manager.py +++ b/src/chat/utils/prompt_component_manager.py @@ -281,10 +281,27 @@ class PromptComponentManager: from src.chat.utils.prompt import global_prompt_manager return list(global_prompt_manager._prompts.keys()) - def get_core_prompt_contents(self) -> dict[str, str]: - """获取所有核心提示词模板的原始内容。""" + def get_core_prompt_contents(self, prompt_name: str | None = None) -> list[list[str]]: + """ + 获取核心提示词模板的原始内容。 + + Args: + prompt_name (str | None, optional): + 如果指定，则只返回该名称对应的提示词模板。 + 如果为 None，则返回所有核心提示词模板。 + 默认为 None。 + + Returns: + list[list[str]]: 一个列表，每个子列表包含 [prompt_name, template_content]。 + 如果指定了 prompt_name 但未找到，则返回空列表。 + """ from src.chat.utils.prompt import global_prompt_manager - return {name: prompt.template for name, prompt in global_prompt_manager._prompts.items()} + + if prompt_name: + prompt = global_prompt_manager._prompts.get(prompt_name) + return [[prompt_name, prompt.template]] if prompt else [] + + return [[name, prompt.template] for name, prompt in global_prompt_manager._prompts.items()] def get_registered_prompt_component_info(self) -> list[PromptInfo]: """获取所有在 ComponentRegistry 中注册的 Prompt 组件信息。""" @@ -316,7 +333,7 @@ class PromptComponentManager: info_map = {} async with self._lock: all_targets = set(self._dynamic_rules.keys()) | set(self.get_core_prompts()) - + # 如果指定了目标，则只处理该目标 targets_to_process = [target_prompt] if target_prompt and target_prompt in all_targets else sorted(all_targets)