Skip to content

Commit a021222

Browse files
DeanChensjcopybara-github
authored andcommitted
feat: Add custom_metadata to VertexAiSessionService event handling
PiperOrigin-RevId: 779255526
1 parent 4442167 commit a021222

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/google/adk/sessions/vertex_ai_session_service.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -381,6 +381,7 @@ def _convert_event_to_json(event: Event) -> Dict[str, Any]:
381381
'turn_complete': event.turn_complete,
382382
'interrupted': event.interrupted,
383383
'branch': event.branch,
384+
'custom_metadata': event.custom_metadata,
384385
'long_running_tool_ids': (
385386
list(event.long_running_tool_ids)
386387
if event.long_running_tool_ids
@@ -460,6 +461,9 @@ def _from_api_event(api_event: Dict[str, Any]) -> Event:
460461
event.turn_complete = api_event['eventMetadata'].get('turnComplete', None)
461462
event.interrupted = api_event['eventMetadata'].get('interrupted', None)
462463
event.branch = api_event['eventMetadata'].get('branch', None)
464+
event.custom_metadata = api_event['eventMetadata'].get(
465+
'customMetadata', None
466+
)
463467
event.grounding_metadata = _session_util.decode_grounding_metadata(
464468
api_event['eventMetadata'].get('groundingMetadata', None)
465469
)

0 commit comments

Comments
 (0)