fix: Set the app_name in AdkApp template to GOOGLE_CLOUD_AGENT_ENGINE_ID if it exists.

yeesian · copybara-github · commit 1cbe028e045f · 2025-05-01T13:29:35.000-07:00
PiperOrigin-RevId: 753705341
diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py
@@ -397,6 +397,11 @@ def set_up(self):
             )
         for key, value in self._tmpl_attrs.get("env_vars").items():
             os.environ[key] = value
+        if "GOOGLE_CLOUD_AGENT_ENGINE_ID" in os.environ:
+            self._tmpl_attrs["app_name"] = os.environ.get(
+                "GOOGLE_CLOUD_AGENT_ENGINE_ID",
+                self._tmpl_attrs.get("app_name"),
+            )
 
         artifact_service_builder = self._tmpl_attrs.get("artifact_service_builder")
         if artifact_service_builder:
@@ -416,10 +421,6 @@ def set_up(self):
                 project=project,
                 location=location,
             )
-            self._tmpl_attrs["app_name"] = os.environ.get(
-                "GOOGLE_CLOUD_AGENT_ENGINE_ID",
-                self._tmpl_attrs.get("app_name"),
-            )
         else:
             self._tmpl_attrs["session_service"] = InMemorySessionService()
 
