exploitation/example now runs in uv env.

Author: Felipe Campos Penha

initiatives/genai_red_team_handbook/exploitation/example/Makefile

@@ -11,11 +11,19 @@ help:
 	@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 "  make sync     - Sync dependencies with uv"\
+	@echo "  make lock     - Lock dependencies with uv"\
 	@echo ""\
 	@echo "Environment:"\
 	@echo "  - Sandbox Directory: $(SANDBOX_DIR)"\
 	@echo ""
 
+sync:
+	uv sync
+
+lock:
+	uv lock
+
 format:
 	uv run black .
 	uv run isort .
@@ -28,9 +36,9 @@ setup:
 	@sleep 5
 	@echo "✅ Environment ready!"
 
-attack:
+attack: sync lock
 	@echo "⚔️  Launching Red Team attack..."
-	python3 attack.py
+	uv run attack.py
 
 stop:
 	@echo "🧹 Tearing down Red Team environment..."

initiatives/genai_red_team_handbook/exploitation/example/attack.py

@@ -1,16 +1,15 @@
 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:
@@ -22,31 +21,35 @@ 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()

initiatives/genai_red_team_handbook/exploitation/example/main.py

@@ -0,0 +1,6 @@
+def main():
+    print("Hello from example!")
+
+
+if __name__ == "__main__":
+    main()

initiatives/genai_red_team_handbook/exploitation/example/pyproject.toml

@@ -0,0 +1,7 @@
+[project]
+name = "example"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12,<3.13"
+dependencies = []