NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 23:17:35 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: Fixes the main.py for memory/agent.py sample.

Browse Source 
PiperOrigin-RevId: 760741263
This commit is contained in:
Wei Sun (Jack) 2025-05-19 13:08:26 -07:00 committed by Copybara-Service
parent 74454170a3
commit 3f21a52853
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
contributing/samples/memory/asyncio_run.py → contributing/samples/memory/main.py
View File

@ -16,7 +16,6 @@ import asyncio
from datetime import datetime from datetime import datetime
from datetime import timedelta from datetime import timedelta
from typing import cast from typing import cast
import warnings
import agent import agent
from dotenv import load_dotenv from dotenv import load_dotenv
@ -26,7 +25,6 @@ from google.adk.sessions import Session
from google.genai import types from google.genai import types
load_dotenv(override=True) load_dotenv(override=True)
warnings.filterwarnings('ignore', category=UserWarning)
logs.log_to_tmp_folder() logs.log_to_tmp_folder()
@ -67,12 +65,12 @@ async def main():
return cast( return cast(
Session, 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 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 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) await runner.memory_service.add_session_to_memory(session_1)
print('-------------------------------------------------------------------') 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 app_name=app_name, user_id=user_id_1
) )
print(f'----Session to use memory: {session_2.id} ----------------------') print(f'----Session to use memory: {session_2.id} ----------------------')