From a624fe5015a0f80ab93602e10fc3b2275aa7f60d Mon Sep 17 00:00:00 2001 From: Yeesian Ng Date: Tue, 9 Jun 2026 10:50:31 -0700 Subject: [PATCH] chore: set GOOGLE_GENAI_USE_ENTERPRISE in the ADK template PiperOrigin-RevId: 929286381 --- vertexai/agent_engines/templates/adk.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vertexai/agent_engines/templates/adk.py b/vertexai/agent_engines/templates/adk.py index 43bc996bef..58c3025285 100644 --- a/vertexai/agent_engines/templates/adk.py +++ b/vertexai/agent_engines/templates/adk.py @@ -926,6 +926,7 @@ def set_up(self): from google.adk.memory.in_memory_memory_service import InMemoryMemoryService os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "1" + os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "1" project = self._tmpl_attrs.get("project") if project: os.environ["GOOGLE_CLOUD_PROJECT"] = project