Skip to content

Commit a624fe5

Browse files
yeesiancopybara-github
authored andcommitted
chore: set GOOGLE_GENAI_USE_ENTERPRISE in the ADK template
PiperOrigin-RevId: 929286381
1 parent 116b71c commit a624fe5

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

  • vertexai/agent_engines/templates

vertexai/agent_engines/templates/adk.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -926,6 +926,7 @@ def set_up(self):
926926
from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
927927

928928
os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "1"
929+
os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "1"
929930
project = self._tmpl_attrs.get("project")
930931
if project:
931932
os.environ["GOOGLE_CLOUD_PROJECT"] = project

0 commit comments

Comments
 (0)