removed double await, fix linter

Author: saraford

generative-ai/snippets/functionCallingStreamChat.js

@@ -13,84 +13,87 @@
 // limitations under the License.
 
 // [START aiplatform_gemini_function_calling_chat]
-const { VertexAI, FunctionDeclarationSchemaType } = require('@google-cloud/vertexai');
+const {
+    VertexAI,
+    FunctionDeclarationSchemaType
+} = require('@google-cloud/vertexai');
 
 const functionDeclarations = [
-    {
-      function_declarations: [
-        {
-          name: "get_current_weather",
-          description: 'get weather in a given location',
-          parameters: {
-            type: FunctionDeclarationSchemaType.OBJECT,
-            properties: {
-              location: {type: FunctionDeclarationSchemaType.STRING},
-              unit: {
-                type: FunctionDeclarationSchemaType.STRING,
-                enum: ['celsius', 'fahrenheit'],
-              },
+  {
+    function_declarations: [
+      {
+        name: 'get_current_weather',
+        description: 'get weather in a given location',
+        parameters: {
+          type: FunctionDeclarationSchemaType.OBJECT,
+          properties: {
+            location: { type: FunctionDeclarationSchemaType.STRING },
+            unit: {
+              type: FunctionDeclarationSchemaType.STRING,
+              enum: ['celsius', 'fahrenheit'],
             },
-            required: ['location'],
           },
+          required: ['location'],
         },
-      ],
-    },
-  ];
-  
-  const functionResponseParts = [
-    {
-      functionResponse: {
-        name: "get_current_weather",
-        response:
-            {name: "get_current_weather", content: {weather: "super nice"}},
       },
+    ],
+  },
+];
+
+const functionResponseParts = [
+  {
+    functionResponse: {
+      name: 'get_current_weather',
+      response:
+        { name: 'get_current_weather', content: { weather: 'super nice' } },
     },
-  ];
+  },
+];
 
 /**
  * TODO(developer): Update these variables before running the sample.
  */
 async function functionCallingStreamChat(
-    projectId = 'PROJECT_ID',
-    location = 'us-central1',
-    model = 'gemini-pro'
+  projectId = 'PROJECT_ID',
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-    // Initialize Vertex with your Cloud project and location
-    const vertexAI = new VertexAI({ project: projectId, location: location });
+  // Initialize Vertex with your Cloud project and location
+  const vertexAI = new VertexAI({ project: projectId, location: location });
 
-    // Instantiate the model
-    const generativeModel = vertexAI.preview.getGenerativeModel({
-        model: model,
-    });
+  // Instantiate the model
+  const generativeModel = vertexAI.preview.getGenerativeModel({
+    model: model,
+  });
 
-    // Create a chat session and pass your function declarations
-    const chat = generativeModel.startChat({
-        tools: functionDeclarations,
-    });
+  // Create a chat session and pass your function declarations
+  const chat = generativeModel.startChat({
+    tools: functionDeclarations,
+  });
 
-    const chatInput1 = 'What is the weather in Boston?';
+  const chatInput1 = 'What is the weather in Boston?';
 
-    // This should include a functionCall response from the model
-    const result1 = await chat.sendMessageStream(chatInput1);
-    for await (const item of result1.stream) {
-        console.log(item.candidates[0]);
-    }
-    const response1 = await result1.response;
+  // This should include a functionCall response from the model
+  const result1 = await chat.sendMessageStream(chatInput1);
+  for await (const item of result1.stream) {
+    console.log(item.candidates[0]);
+  }
+  const response1 = await result1.response;
 
-    // Send a follow up message with a FunctionResponse
-    const result2 = await chat.sendMessageStream(functionResponseParts);
-    for await (const item of result2.stream) {
-        console.log(item.candidates[0]);
-    }
+  // Send a follow up message with a FunctionResponse
+  const result2 = await chat.sendMessageStream(functionResponseParts);
+  for await (const item of result2.stream) {
+    console.log(item.candidates[0]);
+  }
 
-    // This should include a text response from the model using the response content
-    // provided above
-    const response2 = await result2.response;
-    console.log(response2.candidates[0].content.parts[0].text);    
+  // This should include a text response from the model using the response content
+  // provided above
+  const response2 = await result2.response;
+  console.log(response2.candidates[0].content.parts[0].text);
 }
 // [END aiplatform_gemini_function_calling_chat]
 
 functionCallingStreamChat(...process.argv.slice(2)).catch(err => {
-    console.error(err.message);
-    process.exitCode = 1;
+  console.error(err.message);
+  process.exitCode = 1;
 });

generative-ai/snippets/functionCallingStreamContent.js

@@ -19,7 +19,7 @@ const functionDeclarations = [
     {
       function_declarations: [
         {
-          name: "get_current_weather",
+          name: 'get_current_weather',
           description: 'get weather in a given location',
           parameters: {
             type: FunctionDeclarationSchemaType.OBJECT,
@@ -40,9 +40,9 @@ const functionDeclarations = [
   const functionResponseParts = [
     {
       functionResponse: {
-        name: "get_current_weather",
+        name: 'get_current_weather',
         response:
-            {name: "get_current_weather", content: {weather: "super nice"}},
+            {name: 'get_current_weather', content: {weather: 'super nice'}},
       },
     },
   ];

generative-ai/snippets/nonStreamingChat.js

@@ -37,7 +37,7 @@ async function createNonStreamingChat(
   console.log(`User: ${chatInput1}`);
 
   const result1 = await chat.sendMessage(chatInput1);
-  const response1 = await result1.response;
+  const response1 = result1.response;
   const text1 = response1.candidates[0].content.parts[0].text;
   console.log('Chat bot: ', text1);