diff --git a/templates/rag-google-cloud-vertexai-search/rag_google_cloud_vertexai_search/__init__.py b/templates/rag-google-cloud-vertexai-search/rag_google_cloud_vertexai_search/__init__.py
index e69de29bb2..7b31128b43 100644
--- a/templates/rag-google-cloud-vertexai-search/rag_google_cloud_vertexai_search/__init__.py
+++ b/templates/rag-google-cloud-vertexai-search/rag_google_cloud_vertexai_search/__init__.py
@@ -0,0 +1,3 @@
+from rag_google_cloud_vertexai_search.chain import chain
+
+__all__ = ["chain"]