fix: fix broken tests

Author: Cloud Shell

generative-ai/snippets/nonStreamingChat.js

@@ -37,20 +37,23 @@ async function createNonStreamingChat(
   console.log(`User: ${chatInput1}`);
 
   const result1 = await chat.sendMessage(chatInput1);
-  const response1 = result1.response.candidates[0].content.parts[0].text;
-  console.log('Chat bot: ', response1);
+  const response1 = await result1.response;
+  const text1 = response1.candidates[0].content.parts[0].text;
+  console.log('Chat bot: ', text1);
 
   const chatInput2 = 'Can you tell me a scientific fun fact?';
   console.log(`User: ${chatInput2}`);
   const result2 = await chat.sendMessage(chatInput2);
-  const response2 = result2.response.candidates[0].content.parts[0].text;
-  console.log('Chat bot: ', response2);
+  const response2 = await result2.response;
+  const text2 = response2.candidates[0].content.parts[0].text;
+  console.log('Chat bot: ', text2);
 
   const chatInput3 = 'How can I learn more about that?';
   console.log(`User: ${chatInput3}`);
   const result3 = await chat.sendMessage(chatInput3);
-  const response3 = result3.response.candidates[0].content.parts[0].text;
-  console.log('Chat bot: ', response3);
+  const response3 = await result3.response;
+  const text3 = response3.candidates[0].content.parts[0].text;
+  console.log('Chat bot: ', text3);
 }
 // [END aiplatform_gemini_multiturn_chat_nonstreaming]
 

generative-ai/snippets/nonStreamingMultipartContent.js

@@ -50,7 +50,7 @@ async function createNonStreamingMultipartContent(
   };
 
   console.log('Prompt Text:');
-  console.log(request.contents[0].parts[0].text);
+  console.log(request.contents[0].parts[1].text);
 
   console.log('Non-Streaming Response Text:');
   // Create the response stream

generative-ai/snippets/safetySettings.js

@@ -59,14 +59,15 @@ async function setSafetySettings(
   for await (const item of responseStream.stream) {
     if (item.candidates[0].finishReason === 'SAFETY') {
       console.log('This response stream terminated due to safety concerns.');
+      break;
     } else {
       process.stdout.write(item.candidates[0].content.parts[0].text);
     }
   }
 }
 // [END aiplatform_gemini_safety_settings]
 
-setSafetySettings(...process.argv.slice(3)).catch(err => {
+setSafetySettings(...process.argv.slice(2)).catch(err => {
   console.error(err.message);
   process.exitCode = 1;
 });

generative-ai/snippets/streamMultipartContent.js

@@ -50,7 +50,7 @@ async function createStreamMultipartContent(
   };
 
   console.log('Prompt Text:');
-  console.log(request.contents[0].parts[0].text);
+  console.log(request.contents[0].parts[1].text);
   console.log('Streaming Response Text:');
 
   // Create the response stream

generative-ai/snippets/test/sendMultiModalPromptWithImage.test.js

@@ -30,6 +30,6 @@ describe('Generative AI Stream MultiModal with Image', () => {
       `node ./sendMultiModalPromptWithImage.js ${project} ${location} ${model}`
     );
     // Ensure that the conversation is what we expect for these images
-    assert(output.match(/Paris/));
+    assert(output.match(/city: Rio de Janeiro, Landmark: Christ the Redeemer/));
   });
 });