diff --git a/pilot/helpers/AgentConvo.py b/pilot/helpers/AgentConvo.py index 16397851e..f8b9e1f5e 100644 --- a/pilot/helpers/AgentConvo.py +++ b/pilot/helpers/AgentConvo.py @@ -29,7 +29,7 @@ def __init__(self, agent): self.high_level_step = self.agent.project.current_step # add system message - system_message = get_sys_message(self.agent.role) + system_message = get_sys_message(self.agent.role,self.agent.project.args) logger.info('\n>>>>>>>>>> System Prompt >>>>>>>>>>\n%s\n>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>', system_message['content']) self.messages.append(system_message) diff --git a/pilot/utils/utils.py b/pilot/utils/utils.py index 03fbca81a..d3983c1c3 100644 --- a/pilot/utils/utils.py +++ b/pilot/utils/utils.py @@ -84,12 +84,12 @@ def get_prompt_components(): return prompts_components -def get_sys_message(role): +def get_sys_message(role,args=None): """ :param role: 'product_owner', 'architect', 'dev_ops', 'tech_lead', 'full_stack_developer', 'code_monkey' :return: { "role": "system", "content": "You are a {role}... You do..." } """ - content = get_prompt(f'system_messages/{role}.prompt') + content = get_prompt(f'system_messages/{role}.prompt',args) return { "role": "system",