From 35403678b2112481c695b6e9f77f344a21d1575f Mon Sep 17 00:00:00 2001 From: tt-P607 <68868379+tt-P607@users.noreply.github.com> Date: Fri, 12 Sep 2025 20:57:56 +0800 Subject: [PATCH] =?UTF-8?q?feat(prompt):=20=E6=B7=BB=E5=8A=A0=E5=AE=89?= =?UTF-8?q?=E5=85=A8=E5=87=86=E5=88=99=E5=8F=82=E6=95=B0=E5=B9=B6=E6=9B=B4?= =?UTF-8?q?=E6=96=B0=E5=B7=A5=E5=85=B7=E6=89=A7=E8=A1=8C=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在PromptParameters类中添加safety_guidelines_block参数 - 更新ToolExecutor初始化,传入chat_id参数用于上下文追踪 - 在所有参数准备方法中集成安全准则配置 - 增强prompt系统的安全性和可追溯性 --- src/chat/utils/prompt.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/chat/utils/prompt.py b/src/chat/utils/prompt.py index ae0c9c4b1..88584f51d 100644 --- a/src/chat/utils/prompt.py +++ b/src/chat/utils/prompt.py @@ -71,6 +71,7 @@ class PromptParameters: identity_block: str = "" schedule_block: str = "" moderation_prompt_block: str = "" + safety_guidelines_block: str = "" reply_target_block: str = "" mood_prompt: str = "" action_descriptions: str = "" @@ -641,7 +642,7 @@ class Prompt: ) # 创建工具执行器 - tool_executor = ToolExecutor() + tool_executor = ToolExecutor(chat_id=self.parameters.chat_id) # 执行工具获取信息 tool_results, _, _ = await tool_executor.execute_from_chat_message( @@ -768,6 +769,7 @@ class Prompt: "reply_style": global_config.personality.reply_style, "keywords_reaction_prompt": self.parameters.keywords_reaction_prompt or context_data.get("keywords_reaction_prompt", ""), "moderation_prompt": self.parameters.moderation_prompt_block or context_data.get("moderation_prompt", ""), + "safety_guidelines_block": self.parameters.safety_guidelines_block or context_data.get("safety_guidelines_block", ""), } def _prepare_normal_params(self, context_data: Dict[str, Any]) -> Dict[str, Any]: @@ -791,6 +793,7 @@ class Prompt: "mood_state": self.parameters.mood_prompt or context_data.get("mood_state", ""), "keywords_reaction_prompt": self.parameters.keywords_reaction_prompt or context_data.get("keywords_reaction_prompt", ""), "moderation_prompt": self.parameters.moderation_prompt_block or context_data.get("moderation_prompt", ""), + "safety_guidelines_block": self.parameters.safety_guidelines_block or context_data.get("safety_guidelines_block", ""), } def _prepare_default_params(self, context_data: Dict[str, Any]) -> Dict[str, Any]: @@ -810,6 +813,7 @@ class Prompt: "reply_style": global_config.personality.reply_style, "keywords_reaction_prompt": self.parameters.keywords_reaction_prompt or context_data.get("keywords_reaction_prompt", ""), "moderation_prompt": self.parameters.moderation_prompt_block or context_data.get("moderation_prompt", ""), + "safety_guidelines_block": self.parameters.safety_guidelines_block or context_data.get("safety_guidelines_block", ""), } def format(self, *args, **kwargs) -> str: