Merge pull request #53 from stratosphereips/mari_tui_prettify5

Fix prompt
stratosphereips · Apr 3, 2024 · 13d6c56 · 13d6c56
2 parents ae9adb9 + a178481
commit 13d6c56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agents/interactive_tui/assistant.py b/agents/interactive_tui/assistant.py
@@ -39,8 +39,8 @@
 COT_PROMPT = """
 Here are some examples of actions:
 Action: {"action":"ScanNetwork", "parameters": {"target_network": "1.1.1.0/24", 'source_host': '2.2.2.2'}}
-Action: {"action":"ScanServices", "parameters":{"target_host":"2.2.2.3"}, 'source_host': '2.2.2.2'}}
-Action: {"action":"ExploitService", "parameters":{"target_host":"1.1.1.1", "target_service":"openssh"}, 'source_host': '1.1.1.2'}}
+Action: {"action":"ScanServices", "parameters":{"target_host":"2.2.2.3", 'source_host': '2.2.2.2'}}
+Action: {"action":"ExploitService", "parameters":{"target_host":"1.1.1.1", "target_service":"openssh", 'source_host': '1.1.1.2'}}
 Action: {"action":"FindData", "parameters":{"target_host":"1.1.1.1", 'source_host': '1.1.1.2'}}
 Action: {"action":"ExfiltrateData", "parameters": {"target_host": "2.2.2.2", "data": {"owner":"User1", "id":"WebData"}, "source_host": "1.1.1.2"}}
 End of examples.