From 7189c9ceb287463cad2ce944edc98f7b11e9cd93 Mon Sep 17 00:00:00 2001 From: Anton Bossenbroek Date: Fri, 27 Dec 2024 09:43:20 -0800 Subject: [PATCH 1/2] fix: adds missing import for gemini --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b382c66a41..7021802c74 100644 --- a/setup.py +++ b/setup.py @@ -120,7 +120,7 @@ "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"], From 49079433d69aae7004fa850b413fb5330033e9da Mon Sep 17 00:00:00 2001 From: Anton Bossenbroek Date: Fri, 27 Dec 2024 11:02:22 -0800 Subject: [PATCH 2/2] ci: fixes formatting --- setup.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 7021802c74..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", "jsonschema"], + "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"],