Removed context placeholder from prompts

antoninoLorenzo · Aug 3, 2024 · dc33343 · dc33343
1 parent 44c6d26
commit dc33343
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/src/agent/prompts.py b/src/agent/prompts.py
@@ -130,10 +130,10 @@
 * **Ask the target system (IP address or URL) if not provided.**
 * **You should plan a single Penetration Testing phase (Reconnaissance, Initial Access, Privilege Escalation).** 
 * **If you're unsure about the phase or need assistance with multiple phases, ask the user to select a single phase.**
-* **Only use the provided tools to create a plan.**
+* **Only use the provided tools to create a plan.**""")
 
-Additional Information:
-{context}""")
+# Additional Information:
+# {context}
         },
         'plan_conversion': {
             'system': textwrap.dedent("""You should extract the commands from the provided natural language plan and 
@@ -181,10 +181,10 @@
 * **Ask the target system (IP address or URL) if not provided.**
 * **You should plan a single Penetration Testing phase (Reconnaissance, Initial Access, Privilege Escalation).** 
 * **If you're unsure about the phase or need assistance with multiple phases, ask the user to select a single phase.**
-* **Only use the provided tools to create a plan.**
+* **Only use the provided tools to create a plan.**""")
 
-Additional Information:
-{context}""")
+# Additional Information:
+# {context}
         },
         'plan_conversion': {
             'system': textwrap.dedent("""You should extract the commands from the provided natural language plan and 
@@ -247,10 +247,10 @@
 * **Ask the target system (IP address or URL) if not provided.**
 * **The plan MUST include a single Penetration Testing phase (Reconnaissance, Initial Access, Privilege Escalation).**
 * **If you're unsure about the phase or need assistance with multiple phases, ask the user to select a single phase.**
-* **Only use the provided tools to create a plan.**
+* **Only use the provided tools to create a plan.**""")
 
-Additional Information:
-{context}""")
+# Additional Information:
+# {context}""")
         },
         'plan_conversion': {
             'system': textwrap.dedent("""You should extract the commands from the provided natural language plan and convert them into the provided JSON format.