diff --git a/setup.py b/setup.py
index b382c66a41..8683843068 100644
--- a/setup.py
+++ b/setup.py
@@ -120,7 +120,14 @@
     "teachable": ["chromadb"],
     "lmm": ["replicate", "pillow"],
     "graph": ["networkx", "matplotlib"],
-    "gemini": ["google-generativeai>=0.5,<1", "google-cloud-aiplatform", "google-auth", "pillow", "pydantic"],
+    "gemini": [
+        "google-generativeai>=0.5,<1",
+        "google-cloud-aiplatform",
+        "google-auth",
+        "pillow",
+        "pydantic",
+        "jsonschema",
+    ],
     "together": ["together>=1.2"],
     "websurfer": ["beautifulsoup4", "markdownify", "pdfminer.six", "pathvalidate"],
     "redis": ["redis"],