File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2626from google .cloud .firestore_v1 .base_query import FieldFilter
2727from typing_extensions import override
2828
29- from google .genai .types import Content
29+ from google .genai .types import Content , GroundingMetadata
3030
3131from ..events .event import Event
3232from ..events .event_actions import EventActions
@@ -336,9 +336,9 @@ def _from_firestore_doc_to_event(doc: firestore.DocumentSnapshot) -> Event:
336336 event .turn_complete = metadata .get ("turn_complete" , None )
337337 event .interrupted = metadata .get ("interrupted" , None )
338338 event .branch = metadata .get ("branch" , None )
339- # grounding_metadata_dict = metadata.get("grounding_metadata", None)
340- # if grounding_metadata_dict:
341- # event.grounding_metadata = GroundingMetadata(**grounding_metadata_dict)
339+ grounding_metadata_dict = metadata .get ("grounding_metadata" , None )
340+ if grounding_metadata_dict :
341+ event .grounding_metadata = GroundingMetadata (** grounding_metadata_dict )
342342 event .long_running_tool_ids = (
343343 set (long_running_tool_ids_list ) if long_running_tool_ids_list else None
344344 )
You can’t perform that action at this time.
0 commit comments