zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

guildline messages

Browse Source
This commit is contained in:
朱潮 2025-12-17 23:21:15 +08:00
parent 47f1d1c7eb
commit c1c51da77e
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
agent/agent_config.py
View File

@ -36,6 +36,7 @@ class AgentConfig:
tool_response: bool = True tool_response: bool = True
preamble_text: Optional[str] = None preamble_text: Optional[str] = None
messages: Optional[List] = field(default_factory=list) messages: Optional[List] = field(default_factory=list)
_origin_messages: Optional[List] = field(default_factory=list)
logging_handler: Optional['LoggingCallbackHandler'] = None logging_handler: Optional['LoggingCallbackHandler'] = None
@ -96,6 +97,7 @@ class AgentConfig:
generate_cfg=generate_cfg, generate_cfg=generate_cfg,
logging_handler=LoggingCallbackHandler(), logging_handler=LoggingCallbackHandler(),
messages=messages, messages=messages,
_origin_messages=messages,
preamble_text=preamble_text, preamble_text=preamble_text,
) )
config.safe_print() config.safe_print()
@ -130,6 +132,7 @@ class AgentConfig:
generate_cfg={}, # v2接口不传递额外的generate_cfg generate_cfg={}, # v2接口不传递额外的generate_cfg
logging_handler=LoggingCallbackHandler(), logging_handler=LoggingCallbackHandler(),
messages=messages, messages=messages,
_origin_messages=messages,
preamble_text=preamble_text, preamble_text=preamble_text,
) )
config.safe_print() config.safe_print()

5
agent/guideline_middleware.py
View File

@ -34,6 +34,7 @@ class GuidelineMiddleware(AgentMiddleware):
self.robot_type = config.robot_type self.robot_type = config.robot_type
self.terms_list = terms_list self.terms_list = terms_list
self.messages = config._origin_messages
if self.robot_type == "general_agent": if self.robot_type == "general_agent":
if not self.guidelines: if not self.guidelines:
@ -94,7 +95,7 @@ Action: Provide concise, friendly, and personified natural responses.
if not self.guidelines: if not self.guidelines:
return None 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() messages = state['messages'].copy()
@ -123,7 +124,7 @@ Action: Provide concise, friendly, and personified natural responses.
if not self.guidelines: if not self.guidelines:
return None 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() messages = state['messages'].copy()