feat: Migrate vertex_ai_session_service to use Agent Engine SDK

PiperOrigin-RevId: 814967790

Author: vertex-sdk-bot

vertexai/agent_engines/templates/adk.py

@@ -537,20 +537,15 @@ def set_up(self):
         if session_service_builder:
             self._tmpl_attrs["session_service"] = session_service_builder()
         elif "GOOGLE_CLOUD_AGENT_ENGINE_ID" in os.environ:
-            try:
-                from google.adk.sessions.vertex_ai_session_service import (
-                    VertexAiSessionService,
-                )
-
-                self._tmpl_attrs["session_service"] = VertexAiSessionService(
-                    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
+            from google.adk.sessions.vertex_ai_session_service import (
+                VertexAiSessionService,
+            )
 
+            self._tmpl_attrs["session_service"] = VertexAiSessionService(
+                project=project,
+                location=location,
+                agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
+            )
         else:
             self._tmpl_attrs["session_service"] = InMemorySessionService()
 

vertexai/preview/reasoning_engines/templates/adk.py

@@ -512,26 +512,21 @@ def set_up(self):
         if session_service_builder:
             self._tmpl_attrs["session_service"] = session_service_builder()
         elif "GOOGLE_CLOUD_AGENT_ENGINE_ID" in os.environ:
-            try:
-                from google.adk.sessions.vertex_ai_session_service import (
-                    VertexAiSessionService,
-                )
-
-                if is_version_sufficient("1.5.0"):
-                    self._tmpl_attrs["session_service"] = VertexAiSessionService(
-                        project=project,
-                        location=location,
-                        agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
-                    )
-                else:
-                    self._tmpl_attrs["session_service"] = VertexAiSessionService(
-                        project=project,
-                        location=location,
-                    )
-            except ImportError:
-                # TODO(deanchen): Handle this via _g3 import for google3.
-                pass
+            from google.adk.sessions.vertex_ai_session_service import (
+                VertexAiSessionService,
+            )
 
+            if is_version_sufficient("1.5.0"):
+                self._tmpl_attrs["session_service"] = VertexAiSessionService(
+                    project=project,
+                    location=location,
+                    agent_engine_id=os.environ.get("GOOGLE_CLOUD_AGENT_ENGINE_ID"),
+                )
+            else:
+                self._tmpl_attrs["session_service"] = VertexAiSessionService(
+                    project=project,
+                    location=location,
+                )
         else:
             self._tmpl_attrs["session_service"] = InMemorySessionService()