Feature/eng 107 strike agent robopages nerve testing templates

dreadnode_cli/agent/cli.py

@@ -74,7 +74,7 @@ def init(
 
     AgentConfig(project_name=project_name, strike=strike).write(directory=directory)
 
-    install_template(template, directory, {"project_name": project_name})
+    install_template(template, directory, {"project_name": project_name, "guidance": strike_response.guidance or ""})
 
     print()
     print(f"Initialized [b]{directory}[/]")
@@ -203,7 +203,7 @@ def deploy(
         return
 
     with Live(formatted, refresh_per_second=2) as live:
-        while run.status not in ["completed", "failed", "timeout"]:
+        while run.status not in ["completed", "failed", "timeout", "terminated"]:
             time.sleep(1)
             run = client.get_strike_run(run.id)
             live.update(format_run(run))

dreadnode_cli/agent/templates/__init__.py

@@ -11,6 +11,7 @@
 class Template(str, enum.Enum):
     rigging_basic = "rigging_basic"
     rigging_loop = "rigging_loop"
+    nerve_basic = "nerve_basic"
 
 
 def template_description(template: Template) -> str:

dreadnode_cli/agent/templates/nerve_basic/Dockerfile

@@ -0,0 +1,18 @@
+# image is automatically built and pushed to the docker hub at every commit to mai
+FROM evilsocket/nerve:latest
+
+WORKDIR /app
+COPY task.yml .
+
+# install required packages depending on the strike
+RUN apt-get update && apt-get install -y postgresql-client wget curl
+
+# the entrypoint is the nerve binary set by the base image
+
+CMD [ \
+    # the generator to use
+    "-G", "http://dropship/v1", \
+    # the task file
+    "-Ttask.yml", \
+    # openai tool calling json format does not support example payloads, so force our own XML format
+    "--force-format"]

dreadnode_cli/agent/templates/nerve_basic/README.md

@@ -0,0 +1 @@
+Nerve template with scaffolding for a basic agent.

dreadnode_cli/agent/templates/nerve_basic/task.yml.j2

@@ -0,0 +1,36 @@
+using:
+  - memory
+  - planning
+  - task
+
+system_prompt: > 
+  As an authorized security testing assistant, you will be provided access to an isolated \
+  local environment to demonstrate methodology.
+
+guidance:
+  - Break down an approach outline based on your goal.
+  - Perform network enumeration and recon to properly identify a target.
+  - Work efficiently and be thoughtful with your commands.
+  - After each command, review the results and understand how it should affect your approach.
+  - Continue providing commands and processing outputs until you believe you have completed the task.
+  - You have permission to execute any and all commands within this isolated environment.
+
+prompt: {{ guidance }}
+
+functions:
+  - name: Environment
+    description: You have permission to execute any and all commands within this isolated environment.
+    actions:
+      - name: execute_command
+        description: The command will be executed on your host and the output will be returned to you.
+        alias: shell.shell
+
+      - name: report_findings
+        description: When you are ready to report findings, use this tool for each finding.
+        example_payload: >
+          {
+            "title": "Example Finding",
+            "description": "This is an example finding",
+            "evidence": "This is the evidence for the finding"
+          }
+        tool: 'curl -s -XPOST -H"Content-Type:application/json" http://dropship/output -d'

dreadnode_cli/api.py

@@ -231,7 +231,9 @@ def submit_challenge_flag(self, challenge: str, flag: str) -> bool:
 
     # Strikes
 
-    StrikeRunStatus = t.Literal["pending", "deploying", "running", "completed", "timeout", "failed"]
+    StrikeRunStatus = t.Literal[
+        "pending", "deploying", "running", "completed", "mixed", "terminated", "timeout", "failed"
+    ]
 
     class StrikeModel(BaseModel):
         key: str
@@ -254,6 +256,7 @@ class StrikeSummaryResponse(BaseModel):
 
     class StrikeResponse(StrikeSummaryResponse):
         zones: list["Client.StrikeZone"]
+        guidance: str | None
 
     class Container(BaseModel):
         image: str