From 3f21a5285317add63c69349329670a305e4ec9df Mon Sep 17 00:00:00 2001
From: "Wei Sun (Jack)" <weisun@google.com>
Date: Mon, 19 May 2025 13:08:26 -0700
Subject: [PATCH] chore: Fixes the main.py for memory/agent.py sample.

PiperOrigin-RevId: 760741263
---
 contributing/samples/memory/{asyncio_run.py => main.py} | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)
 rename contributing/samples/memory/{asyncio_run.py => main.py} (94%)

diff --git a/contributing/samples/memory/asyncio_run.py b/contributing/samples/memory/main.py
similarity index 94%
rename from contributing/samples/memory/asyncio_run.py
rename to contributing/samples/memory/main.py
index 9cdea4e..be9627d 100755
--- a/contributing/samples/memory/asyncio_run.py
+++ b/contributing/samples/memory/main.py
@@ -16,7 +16,6 @@ import asyncio
 from datetime import datetime
 from datetime import timedelta
 from typing import cast
-import warnings
 
 import agent
 from dotenv import load_dotenv
@@ -26,7 +25,6 @@ from google.adk.sessions import Session
 from google.genai import types
 
 load_dotenv(override=True)
-warnings.filterwarnings('ignore', category=UserWarning)
 logs.log_to_tmp_folder()
 
 
@@ -67,12 +65,12 @@ async def main():
 
     return cast(
         Session,
-        runner.session_service.get_session(
+        await runner.session_service.get_session(
             app_name=app_name, user_id=user_id_1, session_id=session.id
         ),
     )
 
-  session_1 = runner.session_service.create_session(
+  session_1 = await runner.session_service.create_session(
       app_name=app_name, user_id=user_id_1
   )
 
@@ -93,7 +91,7 @@ async def main():
     await runner.memory_service.add_session_to_memory(session_1)
   print('-------------------------------------------------------------------')
 
-  session_2 = runner.session_service.create_session(
+  session_2 = await runner.session_service.create_session(
       app_name=app_name, user_id=user_id_1
   )
   print(f'----Session to use memory: {session_2.id} ----------------------')