Skip to content

Commit 67a2e96

Browse files
yyyu-googlecopybara-github
authored andcommitted
fix: unary api should only need to await once.
PiperOrigin-RevId: 607141894
1 parent 6782431 commit 67a2e96

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

src/functions/post_fetch_processing.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -237,10 +237,10 @@ export async function processNonStream(
237237
): Promise<GenerateContentResult> {
238238
if (response !== undefined) {
239239
// ts-ignore
240-
const responseJson = response.json();
241-
return {
240+
const responseJson = await response.json();
241+
return Promise.resolve({
242242
response: responseJson,
243-
};
243+
});
244244
}
245245

246246
return Promise.resolve({

system_test/end_to_end_sample_test.ts

+4-4
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,7 @@ describe('generateContent', () => {
390390
it('should return the aggregated response', async () => {
391391
const response = await generativeTextModel.generateContent(TEXT_REQUEST);
392392

393-
const aggregatedResp = await response.response;
393+
const aggregatedResp = response.response;
394394
expect(aggregatedResp.candidates[0]).toBeTruthy(
395395
`sys test failure on generateContentStream for aggregated response: ${aggregatedResp}`
396396
);
@@ -399,7 +399,7 @@ describe('generateContent', () => {
399399
const response =
400400
await generativeTextModelPreview.generateContent(TEXT_REQUEST);
401401

402-
const aggregatedResp = await response.response;
402+
const aggregatedResp = response.response;
403403
expect(aggregatedResp.candidates[0]).toBeTruthy(
404404
`sys test failure on generateContentStream in preview for aggregated response: ${aggregatedResp}`
405405
);
@@ -414,7 +414,7 @@ describe('sendMessage', () => {
414414
const chat = generativeTextModel.startChat();
415415
const chatInput1 = 'How can I learn more about Node.js?';
416416
const result1 = await chat.sendMessage(chatInput1);
417-
const response1 = await result1.response;
417+
const response1 = result1.response;
418418
expect(response1.candidates[0]).toBeTruthy(
419419
`sys test failure on sendMessage for aggregated response: ${response1}`
420420
);
@@ -424,7 +424,7 @@ describe('sendMessage', () => {
424424
const chat = generativeTextModelPreview.startChat();
425425
const chatInput1 = 'How can I learn more about Node.js?';
426426
const result1 = await chat.sendMessage(chatInput1);
427-
const response1 = await result1.response;
427+
const response1 = result1.response;
428428
expect(response1.candidates[0]).toBeTruthy(
429429
`sys test failure on sendMessage in preview for aggregated response: ${response1}`
430430
);

0 commit comments

Comments
 (0)