chore: make sure memory service is setup in streaming_agent_run_with_response

PiperOrigin-RevId: 853909960

Author: Tongzhou-Jiang

vertexai/agent_engines/templates/adk.py

@@ -1162,6 +1162,10 @@ async def streaming_agent_run_with_events(self, request_json: str):
             self.set_up()
         if not self._tmpl_attrs.get("session_service"):
             self.set_up()
+        if not self._tmpl_attrs.get("in_memory_memory_service"):
+            self.set_up()
+        if not self._tmpl_attrs.get("memory_service"):
+            self.set_up()
         app = self._tmpl_attrs.get("app")
 
         # Try to get the session, if it doesn't exist, create a new one.

vertexai/preview/reasoning_engines/templates/adk.py

@@ -1002,6 +1002,10 @@ async def _invoke_agent_async():
                 self.set_up()
             if not self._tmpl_attrs.get("session_service"):
                 self.set_up()
+            if not self._tmpl_attrs.get("in_memory_memory_service"):
+                self.set_up()
+            if not self._tmpl_attrs.get("memory_service"):
+                self.set_up()
 
             # Try to get the session, if it doesn't exist, create a new one.
             if request.session_id: