added openai and google-cloud-aiplatform to all-steps dependencies

AstraZeneca · Sep 9, 2024 · 664fdd2 · 664fdd2
1 parent 4d36786
commit 664fdd2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -89,8 +89,9 @@ all-steps = [
   "py4j>=0.10.9",
   "rdkit>=2023.3.1",
   "stanza>=1.0.0,<1.6.0", # We're seeing IT test failures on linux with stanza 1.6.0
-  "gliner==0.1.7" # GLINER is under active development, so pinning version until API stabilises
-
+  "gliner==0.1.7", # GLINER is under active development, so pinning version until API stabilises
+  "google-cloud-aiplatform", # Required for vertex API
+  "openai" # Required for OpenAI API
 ]
 dev = [
   "kazu[webserver,typed,model-training,all-steps,model-training]",