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

feat: Add custom_metadata to VertexAiSessionService event handling

Browse Source 
PiperOrigin-RevId: 779255526
This commit is contained in:
Shangjie Chen
2025-07-04 12:38:15 -07:00
committed by Copybara-Service
parent 4442167dd5
commit a021222077
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/sessions/vertex_ai_session_service.py
+4
View File
@@ -381,6 +381,7 @@ def _convert_event_to_json(event: Event) -> Dict[str, Any]:
'turn_complete': event.turn_complete,
'interrupted': event.interrupted,
'branch': event.branch,
'custom_metadata': event.custom_metadata,
'long_running_tool_ids': (
list(event.long_running_tool_ids)
if event.long_running_tool_ids
@@ -460,6 +461,9 @@ def _from_api_event(api_event: Dict[str, Any]) -> Event:
event.turn_complete = api_event['eventMetadata'].get('turnComplete', None)
event.interrupted = api_event['eventMetadata'].get('interrupted', None)
event.branch = api_event['eventMetadata'].get('branch', None)
event.custom_metadata = api_event['eventMetadata'].get(
'customMetadata', None
)
event.grounding_metadata = _session_util.decode_grounding_metadata(
api_event['eventMetadata'].get('groundingMetadata', None)
)