update dependencies

alan-turing-institute · Jan 30, 2025 · ff794dc · ff794dc
1 parent 504cef2
commit ff794dc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,6 +34,7 @@ cli-test-helpers = { version = "^4.0.0", optional = true }
 vertexai = { version ="^1.71.1", optional = true }
 google-cloud-aiplatform = { version = "^1.71.1", optional = true }
 google-generativeai = { version = "^0.8.4", optional = true }
+google-genai = { version = "^0.7.0", optional = true }
 openai = { version = "^1.60.0", optional = true }
 pillow = { version = "^11.1.0", optional = true }
 ollama = { version = "^0.4.7", optional = true }
@@ -66,6 +67,7 @@ all = [
     "vertexai",
     "google-cloud-aiplatform",
     "google-generativeai",
+    "google-genai",
     "openai",
     "pillow",
     "ollama",
@@ -95,8 +97,8 @@ dev = [
     "mkdocs-jupyter",
     "cli-test-helpers",
 ]
-gemini = ["vertexai", "google-cloud-aiplatform", "google-generativeai", "pillow"]
-vertexai = ["vertexai", "google-cloud-aiplatform", "google-generativeai", "pillow"]
+gemini = ["vertexai", "google-cloud-aiplatform", "google-generativeai", "google-genai", "pillow"]
+vertexai = ["vertexai", "google-cloud-aiplatform", "google-generativeai", "google-genai", "pillow"]
 azure_openai = ["openai", "pillow"]
 openai = ["openai", "pillow"]
 ollama = ["ollama"]