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: Add usage_metadata and citation_metadata to DatabaseSessionService

Browse Source 
PiperOrigin-RevId: 819900773
This commit is contained in:
Shangjie Chen
2025-10-16 13:05:52 -07:00
committed by Copybara-Service
parent 2424d6a3b1
commit 6ab1498aa0
4 changed files with 44 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/sessions/test_session_service.py
+5
View File
@@ -390,6 +390,11 @@ async def test_append_event_complete(service_type):
error_code='error_code',
error_message='error_message',
interrupted=True,
usage_metadata=types.GenerateContentResponseUsageMetadata(
prompt_token_count=1, candidates_token_count=1, total_token_count=2
),
citation_metadata=types.CitationMetadata(),
custom_metadata={'custom_key': 'custom_value'},
)
await session_service.append_event(session=session, event=event)