From a178481428d34ae9d3ee810463effda4acbd1f00 Mon Sep 17 00:00:00 2001 From: Maria Rigaki Date: Wed, 3 Apr 2024 11:49:06 +0200 Subject: [PATCH] Fix prompt --- agents/interactive_tui/assistant.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/agents/interactive_tui/assistant.py b/agents/interactive_tui/assistant.py index 83c8775..d01aecb 100644 --- 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.