Skip to content

Commit 5967189

Browse files
committed
fix: remove duplicate usage_metadata handling after rebase
Upstream main already handles usage_metadata round-trip via _USAGE_METADATA_CUSTOM_METADATA_KEY in custom_metadata. Remove the redundant event_metadata-based usage_metadata storage/retrieval that was added in the initial commit, which caused a SyntaxError (repeated keyword argument) when merged with upstream.
1 parent 2c2c869 commit 5967189

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

src/google/adk/sessions/vertex_ai_session_service.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -304,10 +304,6 @@ async def append_event(self, session: Session, event: Event) -> Event:
304304
else None
305305
),
306306
}
307-
if event.usage_metadata:
308-
metadata_dict['usage_metadata'] = event.usage_metadata.model_dump(
309-
exclude_none=True, mode='json'
310-
)
311307
if event.cache_metadata:
312308
metadata_dict['cache_metadata'] = event.cache_metadata.model_dump(
313309
exclude_none=True, mode='json'
@@ -432,10 +428,6 @@ def _from_api_event(api_event_obj: vertexai.types.SessionEvent) -> Event:
432428
getattr(event_metadata, 'grounding_metadata', None),
433429
types.GroundingMetadata,
434430
)
435-
usage_metadata = _session_util.decode_model(
436-
getattr(event_metadata, 'usage_metadata', None),
437-
types.GenerateContentResponseUsageMetadata,
438-
)
439431
cache_metadata = _session_util.decode_model(
440432
getattr(event_metadata, 'cache_metadata', None),
441433
CacheMetadata,
@@ -450,7 +442,6 @@ def _from_api_event(api_event_obj: vertexai.types.SessionEvent) -> Event:
450442
compaction_data = None
451443
usage_metadata_data = None
452444
grounding_metadata = None
453-
usage_metadata = None
454445
cache_metadata = None
455446

456447
if actions:
@@ -493,7 +484,6 @@ def _from_api_event(api_event_obj: vertexai.types.SessionEvent) -> Event:
493484
branch=branch,
494485
custom_metadata=custom_metadata,
495486
grounding_metadata=grounding_metadata,
496-
usage_metadata=usage_metadata,
497487
cache_metadata=cache_metadata,
498488
long_running_tool_ids=long_running_tool_ids,
499489
usage_metadata=usage_metadata,

0 commit comments

Comments
 (0)