diff --git a/rag_agent/__init__.py b/rag_agent/__init__.py
index 02c597e..855906f 100644
--- a/rag_agent/__init__.py
+++ b/rag_agent/__init__.py
@@ -1 +1,10 @@
-from . import agent
+"""Package export for the ADK root agent."""
+
+import os
+
+# Ensure the Google GenAI SDK talks to Vertex AI instead of the public Gemini API.
+os.environ.setdefault("GOOGLE_GENAI_USE_VERTEXAI", "true")
+
+from .agent import root_agent
+
+__all__ = ["root_agent"]