diff --git a/metadata-ingestion/setup.py b/metadata-ingestion/setup.py
index edc8afdb220a8f..2a6f3fe013c5ac 100644
--- a/metadata-ingestion/setup.py
+++ b/metadata-ingestion/setup.py
@@ -675,6 +675,7 @@
             "sac",
             "cassandra",
             "neo4j",
+            "vertexai",
         ]
         if plugin
         for dependency in plugins[plugin]