Skip to content

Commit 1bcec76

Browse files
committed
fix(llma): fix mypy issue
1 parent 97c9048 commit 1bcec76

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

posthog/ai/gemini/gemini_converter.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -472,8 +472,7 @@ def extract_gemini_usage_from_chunk(chunk: Any) -> TokenUsage:
472472
usage_from_metadata = _extract_usage_from_metadata(chunk.usage_metadata)
473473

474474
# Merge the usage from metadata with any web search count we found
475-
for key, value in usage_from_metadata.items():
476-
usage[key] = value
475+
usage.update(usage_from_metadata)
477476

478477
return usage
479478

0 commit comments

Comments
 (0)