diff --git a/pyproject.toml b/pyproject.toml
index 246f3d1..a79ce7a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "celeste-ai"
-version = "0.2.1"
+version = "0.2.2"
 description = "Open source, type-safe primitives for multi-modal AI. All capabilities, all providers, one interface"
 authors = [{name = "Kamilbenkirane", email = "kamil@withceleste.ai"}]
 readme = "README.md"
@@ -35,7 +35,12 @@ Issues = "https://github.com/withceleste/celeste-python/issues"
 [project.optional-dependencies]
 text-generation = ["celeste-text-generation>=0.2.1"]
 image-generation = ["celeste-image-generation>=0.2.1"]
-all = ["celeste-text-generation>=0.2.1", "celeste-image-generation>=0.2.1"]
+image-intelligence = ["celeste-image-intelligence>=0.2.1"]
+all = [
+    "celeste-text-generation>=0.2.1",
+    "celeste-image-generation>=0.2.1",
+    "celeste-image-intelligence>=0.2.1",
+]
 
 [dependency-groups]
 dev = [