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

fix: Process ListEvents response correctly when getSession is called

Browse Source 
When get session is being called on a session with more than 1 page of events (100+), the response of the subsequent listevents calls fails due to missing parsing logic. This change fixes the processing of the listevents responses.

PiperOrigin-RevId: 783166959
This commit is contained in:
Google Team Member
2025-07-14 21:21:51 -07:00
committed by Copybara-Service
parent 498ce906dd
commit a50419961c
2 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/sessions/test_vertex_ai_session_service.py
+7 -1
View File
@@ -201,7 +201,13 @@ class MockApiClient:
if match:
session_id = match.group(2)
if match.group(3):
return {'sessionEvents': MOCK_EVENT_JSON_3}
page_token = match.group(3)
if page_token == 'my_token':
response = {'sessionEvents': MOCK_EVENT_JSON_3}
response['nextPageToken'] = 'my_token2'
return response
else:
return {}
events_tuple = self.event_dict.get(session_id, ([], None))
response = {'sessionEvents': events_tuple[0]}
if events_tuple[1]: