From 98d23b80e460e5f19003ce9440a0a1cb264df1f0 Mon Sep 17 00:00:00 2001
From: A8080816 <angel.juarez.vazquez@banorte.com>
Date: Fri, 20 Feb 2026 23:24:26 +0000
Subject: [PATCH] dev_fix: update __init__.py to set environment variable for
 Google GenAI SDK to use Vertex AI

---
 rag_agent/__init__.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

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"]