Merge pull request #168 from Dhanushkumar-S-G/main

added args to get_sys_message
Pythagora-io · Oct 10, 2023 · 56b0947 · 56b0947
2 parents dba38e7 + 801c8c7
commit 56b0947
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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",