gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(prompt): 添加安全准则参数并更新工具执行器

Browse Source 
- 在PromptParameters类中添加safety_guidelines_block参数
- 更新ToolExecutor初始化，传入chat_id参数用于上下文追踪
- 在所有参数准备方法中集成安全准则配置
- 增强prompt系统的安全性和可追溯性
This commit is contained in:
tt-P607
2025-09-12 20:57:56 +08:00
committed by Windpicker-owo
parent e607dcb36b
commit 25aa75a6db
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/chat/utils/prompt.py
View File

@@ -71,6 +71,7 @@ class PromptParameters:
identity_block: str = "" identity_block: str = ""
schedule_block: str = "" schedule_block: str = ""
moderation_prompt_block: str = "" moderation_prompt_block: str = ""
safety_guidelines_block: str = ""
reply_target_block: str = "" reply_target_block: str = ""
mood_prompt: str = "" mood_prompt: str = ""
action_descriptions: 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( tool_results, _, _ = await tool_executor.execute_from_chat_message(
@@ -768,6 +769,7 @@ class Prompt:
"reply_style": global_config.personality.reply_style, "reply_style": global_config.personality.reply_style,
"keywords_reaction_prompt": self.parameters.keywords_reaction_prompt or context_data.get("keywords_reaction_prompt", ""), "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", ""), "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]: 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", ""), "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", ""), "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", ""), "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]: 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, "reply_style": global_config.personality.reply_style,
"keywords_reaction_prompt": self.parameters.keywords_reaction_prompt or context_data.get("keywords_reaction_prompt", ""), "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", ""), "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: def format(self, *args, **kwargs) -> str: