ADK changes

PiperOrigin-RevId: 754131080

src/google/adk/memory/base_memory_service.py

@@ -51,7 +51,7 @@ class BaseMemoryService(abc.ABC):
   """
 
   @abc.abstractmethod
-  def add_session_to_memory(self, session: Session):
+  async def add_session_to_memory(self, session: Session):
     """Adds a session to the memory service.
 
     A session may be added multiple times during its lifetime.
@@ -61,7 +61,7 @@ class BaseMemoryService(abc.ABC):
     """
 
   @abc.abstractmethod
-  def search_memory(
+  async def search_memory(
       self, *, app_name: str, user_id: str, query: str
   ) -> SearchMemoryResponse:
     """Searches for sessions that match the query.

src/google/adk/memory/in_memory_memory_service.py

@@ -29,13 +29,13 @@ class InMemoryMemoryService(BaseMemoryService):
     self.session_events: dict[str, list[Event]] = {}
     """keys are app_name/user_id/session_id"""
 
-  def add_session_to_memory(self, session: Session):
+  async def add_session_to_memory(self, session: Session):
     key = f'{session.app_name}/{session.user_id}/{session.id}'
     self.session_events[key] = [
         event for event in session.events if event.content
     ]
 
-  def search_memory(
+  async def search_memory(
       self, *, app_name: str, user_id: str, query: str
   ) -> SearchMemoryResponse:
     """Prototyping purpose only."""

src/google/adk/memory/vertex_ai_rag_memory_service.py

@@ -54,7 +54,7 @@ class VertexAiRagMemoryService(BaseMemoryService):
     )
 
   @override
-  def add_session_to_memory(self, session: Session):
+  async def add_session_to_memory(self, session: Session):
     with tempfile.NamedTemporaryFile(
         mode="w", delete=False, suffix=".txt"
     ) as temp_file:
@@ -91,7 +91,7 @@ class VertexAiRagMemoryService(BaseMemoryService):
     os.remove(temp_file_path)
 
   @override
-  def search_memory(
+  async def search_memory(
       self, *, app_name: str, user_id: str, query: str
   ) -> SearchMemoryResponse:
     """Searches for sessions that match the query using rag.retrieval_query."""