chore: upgrade dreadnode sdk to 1.17.1 and add org/workspace params

README.md

@@ -87,7 +87,45 @@ uv run -m airtbench --help
 ### Basic Usage
 
 ```bash
-uv run -m airtbench --model $MODEL --project $PROJECT --platform-api-key $DREADNODE_TOKEN --token $DREADNODE_TOKEN --server https://platform.dreadnode.io --max-steps 100 --inference_timeout 240 --enable-cache --no-give-up --challenges bear1 bear2
+uv run -m airtbench \
+  --model $MODEL \
+  --project $PROJECT \
+  --platform-api-key $DREADNODE_TOKEN \
+  --token $DREADNODE_TOKEN \
+  --server https://platform.dreadnode.io \
+  --organization $ORGANIZATION \
+  --max-steps 100 \
+  --inference-timeout 240 \
+  --enable-cache \
+  --no-give-up \
+  --challenges bear1 bear2
+```
+
+**Organization and Workspace Parameters**
+
+If you are part of multiple organizations, you must specify which organization to use:
+
+```bash
+--organization "dreadnode"
+```
+
+Optionally, you can also specify a workspace within the organization:
+
+```bash
+--organization "dreadnode" --workspace "my-workspace"
+```
+
+**Example with organization:**
+
+```bash
+uv run -m airtbench \
+  --model openai/gpt-4o \
+  --project airtbench \
+  --platform-api-key $DREADNODE_TOKEN \
+  --token $DREADNODE_TOKEN \
+  --server https://platform.dreadnode.io \
+  --organization "dreadnode" \
+  --challenges bear1
 ```
 
 ### Challenge Filtering

airtbench/main.py

@@ -84,6 +84,10 @@ class DreadnodeArgs:
     """Local directory to store data in"""
     project: str = "airtbench"
     """Project name"""
+    organization: str | None = None
+    """Organization name"""
+    workspace: str | None = None
+    """Workspace name"""
 
 
 # Models
@@ -709,6 +713,8 @@ async def main(
         token=dn_args.token,
         local_dir=dn_args.local_dir or False,
         project=dn_args.project,
+        organization=dn_args.organization,
+        workspace=dn_args.workspace,
         console=True,
     )
 

pyproject.toml

@@ -7,7 +7,7 @@ dependencies = [
     "backoff>=2.2.1",
     "cyclopts>=3.12.0",
     "docker>=7.1.0",
-    "dreadnode==1.14.1",
+    "dreadnode==1.17.1",
     "ipykernel>=6.29.5",
     "mypy>=1.15.0",
     "pythonnet>=3.0.5",