No public description

PiperOrigin-RevId: 814417092

Author: vertex-sdk-bot

vertexai/agent_engines/templates/adk.py

@@ -555,15 +555,19 @@ def set_up(self):
         elif "GOOGLE_CLOUD_AGENT_ENGINE_ID" in os.environ and is_version_sufficient(
             "1.5.0"
         ):
-            from google.adk.memory.vertex_ai_memory_bank_service import (
-                VertexAiMemoryBankService,
-            )
+            try:
+                from google.adk.memory.vertex_ai_memory_bank_service import (
+                    VertexAiMemoryBankService,
+                )
 
-            self._tmpl_attrs["memory_service"] = VertexAiMemoryBankService(
-                project=project,
-                location=location,
-                agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
-            )
+                self._tmpl_attrs["memory_service"] = VertexAiMemoryBankService(
+                    project=project,
+                    location=location,
+                    agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
+                )
+            except ImportError:
+                # TODO(ysian): Handle this via _g3 import for google3.
+                pass
         else:
             self._tmpl_attrs["memory_service"] = InMemoryMemoryService()
 

vertexai/preview/reasoning_engines/templates/adk.py

@@ -536,15 +536,19 @@ def set_up(self):
         elif "GOOGLE_CLOUD_AGENT_ENGINE_ID" in os.environ and is_version_sufficient(
             "1.5.0"
         ):
-            from google.adk.memory.vertex_ai_memory_bank_service import (
-                VertexAiMemoryBankService,
-            )
+            try:
+                from google.adk.memory.vertex_ai_memory_bank_service import (
+                    VertexAiMemoryBankService,
+                )
 
-            self._tmpl_attrs["memory_service"] = VertexAiMemoryBankService(
-                project=project,
-                location=location,
-                agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
-            )
+                self._tmpl_attrs["memory_service"] = VertexAiMemoryBankService(
+                    project=project,
+                    location=location,
+                    agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
+                )
+            except ImportError:
+                # TODO(ysian): Handle this via _g3 import for google3.
+                pass
         else:
             self._tmpl_attrs["memory_service"] = InMemoryMemoryService()