guildline messages
This commit is contained in:
parent
47f1d1c7eb
commit
c1c51da77e
@ -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()
|
||||
|
||||
@ -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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user