Fix missing usageMetadata in streamed aggregated response (#174) (#175)

google-gemini · Jun 17, 2024 · 03eb57b · 03eb57b
1 parent 40daf95
commit 03eb57b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/.changeset/quick-horses-retire.md b/.changeset/quick-horses-retire.md
@@ -0,0 +1,5 @@
+---
+"@google/generative-ai": patch
+---
+
+Fix missing usageMetadata in streamed aggregated response (#174)
diff --git a/packages/main/src/requests/stream-reader.ts b/packages/main/src/requests/stream-reader.ts
@@ -188,6 +188,9 @@ export function aggregateResponses(
         }
       }
     }
+    if (response.usageMetadata) {
+      aggregatedResponse.usageMetadata = response.usageMetadata;
+    }
   }
   return aggregatedResponse;
 }