diff --git a/metadata-ingestion/setup.py b/metadata-ingestion/setup.py
index fa7e0bb11f7a88..446fc445883fe5 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]