make format recipe.

Author: Felipe Campos Penha

initiatives/genai_red_team_handbook/exploitation/example/Makefile

@@ -4,17 +4,23 @@ SANDBOX_DIR := ../../sandboxes/llm_local
 
 # Default target
 help:
-	@echo "Red Team Example - Available Commands:"
-	@echo ""
-	@echo "  make setup    - Build and start the local LLM sandbox"
-	@echo "  make attack   - Run the adversarial attack script"
-	@echo "  make stop     - Stop and remove the sandbox container"
-	@echo "  make all      - Run setup, attack, and stop in sequence"
-	@echo ""
-	@echo "Environment:"
-	@echo "  - Sandbox Directory: $(SANDBOX_DIR)"
+	@echo "Red Team Example - Available Commands:"\
+	@echo ""\
+	@echo "  make setup    - Build and start the local LLM sandbox"\
+	@echo "  make attack   - Run the adversarial attack script"\
+	@echo "  make stop     - Stop and remove the sandbox container"\
+	@echo "  make all      - Run setup, attack, and stop in sequence"\
+	@echo "  make format   - Run code formatting (black, isort, mypy)"\
+	@echo ""\
+	@echo "Environment:"\
+	@echo "  - Sandbox Directory: $(SANDBOX_DIR)"\
 	@echo ""
 
+format:
+	uv run black .
+	uv run isort .
+	uv run mypy .
+
 setup:
 	@echo "🚀 Setting up Red Team environment..."
 	$(MAKE) -C $(SANDBOX_DIR) build up

initiatives/genai_red_team_handbook/exploitation/example/attack.py

@@ -1,15 +1,16 @@
 import json
-import urllib.request
 import sys
 import tomllib
+import urllib.request
+
 
 def attack():
     url = "http://localhost:8000/v1/chat/completions"
     headers = {
         "Content-Type": "application/json",
-        "Authorization": "Bearer sk-mock-key"
+        "Authorization": "Bearer sk-mock-key",
     }
-    
+
     # Load prompt from configuration
     try:
         with open("config.toml", "rb") as f:
@@ -21,35 +22,31 @@ def attack():
     except Exception as e:
         print(f"[!] Error loading config: {e}")
         sys.exit(1)
-    
+
     data = {
         "model": "gpt-oss:20b",
-        "messages": [
-            {"role": "user", "content": prompt}
-        ],
-        "temperature": 0.7
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.7,
     }
-    
+
     try:
         req = urllib.request.Request(
-            url, 
-            data=json.dumps(data).encode('utf-8'), 
-            headers=headers, 
-            method='POST'
+            url, data=json.dumps(data).encode("utf-8"), headers=headers, method="POST"
         )
-        
+
         print(f"[*] Sending adversarial prompt: {prompt}")
         with urllib.request.urlopen(req) as response:
-            result = json.loads(response.read().decode('utf-8'))
-            content = result['choices'][0]['message']['content']
+            result = json.loads(response.read().decode("utf-8"))
+            content = result["choices"][0]["message"]["content"]
             print(f"[*] Response received:\n{content}")
-            
+
     except urllib.error.URLError as e:
         print(f"[!] Error communicating with API: {e}")
         sys.exit(1)
     except Exception as e:
         print(f"[!] Unexpected error: {e}")
         sys.exit(1)
 
+
 if __name__ == "__main__":
     attack()