encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

chore: Do not send api request when session does not have events

Browse Source 
PiperOrigin-RevId: 775423356
This commit is contained in:
Shangjie Chen
2025-06-24 16:15:23 -07:00
committed by Copybara-Service
parent 09f1269bf7
commit 88a4402d14
2 changed files with 25 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/memory/test_vertex_ai_memory_bank_service.py
+16
View File
@@ -48,6 +48,13 @@ MOCK_SESSION = Session(
],
)
MOCK_SESSION_WITH_EMPTY_EVENTS = Session(
app_name=MOCK_APP_NAME,
user_id=MOCK_USER_ID,
id='444',
last_update_time=22333,
)
RETRIEVE_MEMORIES_REGEX = r'^reasoningEngines/([^/]+)/memories:retrieve$'
GENERATE_MEMORIES_REGEX = r'^reasoningEngines/([^/]+)/memories:generate$'
@@ -136,6 +143,15 @@ async def test_add_session_to_memory(mock_get_api_client):
)
@pytest.mark.asyncio
@pytest.mark.usefixtures('mock_get_api_client')
async def test_add_empty_session_to_memory(mock_get_api_client):
memory_service = mock_vertex_ai_memory_bank_service()
await memory_service.add_session_to_memory(MOCK_SESSION_WITH_EMPTY_EVENTS)
mock_get_api_client.async_request.assert_not_called()
@pytest.mark.asyncio
@pytest.mark.usefixtures('mock_get_api_client')
async def test_search_memory(mock_get_api_client):