diff --git a/.codegen.json b/.codegen.json
index c5c1a82f3..2ef68a1a1 100644
--- a/.codegen.json
+++ b/.codegen.json
@@ -5,7 +5,7 @@
     "databricks/sdk/version.py": "__version__ = \"$VERSION\""
   },
   "toolchain": {
-    "required": ["python3.12"],
+    "required": ["python3.12", "uv"],
     "pre_setup": [
       "python3.12 -m venv .databricks"
     ],