Skip to content

Commit 6096927

Browse files
committed
grounding metadata uncommented
1 parent 5e84dde commit 6096927

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/google/adk/sessions/firestore_session_service.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from google.cloud.firestore_v1.base_query import FieldFilter
2727
from typing_extensions import override
2828

29-
from google.genai.types import Content
29+
from google.genai.types import Content, GroundingMetadata
3030

3131
from ..events.event import Event
3232
from ..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
)

0 commit comments

Comments
 (0)