Skip to content

Commit 9dd3a59

Browse files
committed
Fix generateContentStream returning wrong inferenceSource.
1 parent 5c35f51 commit 9dd3a59

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

.changeset/gorgeous-rice-carry.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
'@firebase/ai': patch
3+
---
4+
5+
Fix `generateContentStream` returning wrong `inferenceSource`.

packages/ai/src/methods/generate-content.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,11 @@ export async function generateContentStream(
7070
() =>
7171
generateContentStreamOnCloud(apiSettings, model, params, requestOptions)
7272
);
73-
return processStream(callResult.response, apiSettings); // TODO: Map streaming responses
73+
return processStream(
74+
callResult.response,
75+
apiSettings,
76+
callResult.inferenceSource
77+
); // TODO: Map streaming responses
7478
}
7579

7680
async function generateContentOnCloud(

0 commit comments

Comments
 (0)