diff --git a/agent/agent_config.py b/agent/agent_config.py index d2e111e..5315abf 100644 --- a/agent/agent_config.py +++ b/agent/agent_config.py @@ -36,6 +36,7 @@ class AgentConfig: tool_response: bool = True preamble_text: Optional[str] = None messages: Optional[List] = field(default_factory=list) + _origin_messages: Optional[List] = field(default_factory=list) logging_handler: Optional['LoggingCallbackHandler'] = None @@ -96,6 +97,7 @@ class AgentConfig: generate_cfg=generate_cfg, logging_handler=LoggingCallbackHandler(), messages=messages, + _origin_messages=messages, preamble_text=preamble_text, ) config.safe_print() @@ -130,6 +132,7 @@ class AgentConfig: generate_cfg={}, # v2接口不传递额外的generate_cfg logging_handler=LoggingCallbackHandler(), messages=messages, + _origin_messages=messages, preamble_text=preamble_text, ) config.safe_print() diff --git a/agent/guideline_middleware.py b/agent/guideline_middleware.py index e76edea..6628d9a 100644 --- a/agent/guideline_middleware.py +++ b/agent/guideline_middleware.py @@ -34,6 +34,7 @@ class GuidelineMiddleware(AgentMiddleware): self.robot_type = config.robot_type self.terms_list = terms_list + self.messages = config._origin_messages if self.robot_type == "general_agent": if not self.guidelines: @@ -94,7 +95,7 @@ Action: Provide concise, friendly, and personified natural responses. if not self.guidelines: return None - guideline_prompt = self.get_guideline_prompt(convert_to_openai_messages(state['messages'])) + guideline_prompt = self.get_guideline_prompt(convert_to_openai_messages(self.messages)) # 准备完整的消息列表 messages = state['messages'].copy() @@ -123,7 +124,7 @@ Action: Provide concise, friendly, and personified natural responses. if not self.guidelines: return None - guideline_prompt = self.get_guideline_prompt(convert_to_openai_messages(state['messages'])) + guideline_prompt = self.get_guideline_prompt(convert_to_openai_messages(self.messages)) # 准备完整的消息列表 messages = state['messages'].copy()