diff --git a/pyproject.toml b/pyproject.toml
index fc418fe87..5cc39114c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -71,7 +71,6 @@ dependencies = [
   "urllib3>=2.0.0",
   "datasets",
   "colorlog>=6.9.0",
-  "codegen-sdk-pink>=0.1.0",
   "codegen-api-client",
 ]
 
@@ -115,6 +114,7 @@ keywords = [
 codegen = "codegen.cli.cli:main"
 
 [project.optional-dependencies]
+pink = ["codegen-sdk-pink>=0.1.0"]
 types = [
   "types-networkx>=3.2.1.20240918",
   "types-tabulate>=0.9.0.20240106",
diff --git a/src/codegen/configs/models/codebase.py b/src/codegen/configs/models/codebase.py
index e7ea1516b..3f5c0a4e3 100644
--- a/src/codegen/configs/models/codebase.py
+++ b/src/codegen/configs/models/codebase.py
@@ -8,9 +8,9 @@
 class PinkMode(IntEnum):
     # Use the python SDK for all files
     OFF = auto()
-    # Use the Rust SDK for all files
+    # Use the Rust SDK for all files. Make sure to install the pink extra
     ALL_FILES = auto()
-    # Use the Rust SDK for files the python SDK can't parse (non-source files)
+    # Use the Rust SDK for files the python SDK can't parse (non-source files). Make sure to install the pink extra
     NON_SOURCE_FILES = auto()