diff --git a/code/ai_models/pyproject.toml b/code/ai_models/pyproject.toml
index a286861..30af747 100644
--- a/code/ai_models/pyproject.toml
+++ b/code/ai_models/pyproject.toml
@@ -10,7 +10,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-dev = ["pytest>=8", "httpx>=0.27", "ruff>=0.6"]
+dev = ["pytest>=8", "httpx>=0.27", "ruff>=0.17.4"]
 
 [tool.setuptools.packages.find]
 include = ["ballast_ai*"]
diff --git a/code/ai_models/requirements-dev.txt b/code/ai_models/requirements-dev.txt
index 38b0acf..3beebeb 100644
--- a/code/ai_models/requirements-dev.txt
+++ b/code/ai_models/requirements-dev.txt
@@ -1,4 +1,4 @@
 -r requirements.txt
 pytest>=8
 httpx>=0.27
-ruff>=0.6
+ruff>=0.17.4