Merge pull request #304 from ag2ai/missing_dependency_gemini

fix: adds missing import for gemini
ag2ai · Dec 27, 2024 · 6cc0a81 · 6cc0a81
2 parents 2411b22 + 4907943
commit 6cc0a81
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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"],