Merge pull request #3598 from GoogleCloudPlatform/feature/add-model-name

add model to each function

Author: gericdong

generative-ai/snippets/countTokens.js

@@ -12,22 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_token_count]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function countTokens(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-  // [START aiplatform_gemini_token_count]
-
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertex_ai = new VertexAI({project: projectId, location: location});
 
@@ -42,9 +37,8 @@ async function countTokens(
 
   const countTokensResp = await generativeModel.countTokens(req);
   console.log('count tokens response: ', countTokensResp);
-
-  // [END aiplatform_gemini_token_count]
 }
+// [END aiplatform_gemini_token_count]
 
 countTokens(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/nonStreamingChat.js

@@ -12,21 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_multiturn_chat_nonstreaming]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function createNonStreamingChat(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-  // [START aiplatform_gemini_multiturn_chat_nonstreaming]
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -55,9 +51,8 @@ async function createNonStreamingChat(
   const result3 = await chat.sendMessage(chatInput3);
   const response3 = result3.response.candidates[0].content.parts[0].text;
   console.log('Chat bot: ', response3);
-
-  // [END aiplatform_gemini_multiturn_chat_nonstreaming]
 }
+// [END aiplatform_gemini_multiturn_chat_nonstreaming]
 
 createNonStreamingChat(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/nonStreamingContent.js

@@ -12,22 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_content_nonstreaming]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function createNonStreamingContent(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-  // [START aiplatform_gemini_content_nonstreaming]
-
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -55,9 +50,8 @@ async function createNonStreamingContent(
     aggregatedResponse.candidates[0].content.parts[0].text;
 
   console.log(fullTextResponse);
-
-  // [END aiplatform_gemini_content_nonstreaming]
 }
+// [END aiplatform_gemini_content_nonstreaming]
 
 createNonStreamingContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/nonStreamingMultipartContent.js

@@ -12,25 +12,19 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_get_started]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function createNonStreamingMultipartContent(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL',
+  location = 'us-central1',
+  model = 'gemini-pro-vision',
   image = 'gs://generativeai-downloads/images/scones.jpg',
   mimeType = 'image/jpeg'
 ) {
-  // [START aiplatform_gemini_get_started]
-
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const image = 'gs://generativeai-downloads/images/scones.jpg'; // Google Cloud Storage image
-  // const mimeType = 'image/jpeg';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -71,9 +65,8 @@ async function createNonStreamingMultipartContent(
     aggregatedResponse.candidates[0].content.parts[0].text;
 
   console.log(fullTextResponse);
-
-  // [END aiplatform_gemini_get_started]
 }
+// [END aiplatform_gemini_get_started]
 
 createNonStreamingMultipartContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/safetySettings.js

@@ -12,25 +12,21 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_safety_settings]
 const {
   VertexAI,
   HarmCategory,
   HarmBlockThreshold,
 } = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function setSafetySettings(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-  // [START aiplatform_gemini_safety_settings]
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -67,8 +63,8 @@ async function setSafetySettings(
       process.stdout.write(item.candidates[0].content.parts[0].text);
     }
   }
-  // [END aiplatform_gemini_safety_settings]
 }
+// [END aiplatform_gemini_safety_settings]
 
 setSafetySettings(...process.argv.slice(3)).catch(err => {
   console.error(err.message);

generative-ai/snippets/sendMultiModalPromptWithImage.js

@@ -26,8 +26,8 @@ async function getBase64(url) {
  */
 async function sendMultiModalPromptWithImage(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro-vision'
 ) {
   // For images, the SDK supports base64 strings
   const landmarkImage1 = await getBase64(

generative-ai/snippets/sendMultiModalPromptWithVideo.js

@@ -12,21 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_single_turn_video]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function sendMultiModalPromptWithVideo(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro-vision'
 ) {
-  // [START aiplatform_gemini_single_turn_video]
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -63,9 +59,8 @@ async function sendMultiModalPromptWithVideo(
     aggregatedResponse.candidates[0].content.parts[0].text;
 
   console.log(fullTextResponse);
-
-  // [END aiplatform_gemini_single_turn_video]
 }
+// [END aiplatform_gemini_single_turn_video]
 
 sendMultiModalPromptWithVideo(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/streamChat.js

@@ -12,21 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_multiturn_chat]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function createStreamChat(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-  // [START aiplatform_gemini_multiturn_chat]
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -44,9 +40,8 @@ async function createStreamChat(
   for await (const item of result1.stream) {
     console.log(item.candidates[0].content.parts[0].text);
   }
-
-  // [END aiplatform_gemini_multiturn_chat]
 }
+// [END aiplatform_gemini_multiturn_chat]
 
 createStreamChat(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/streamContent.js

@@ -12,22 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_content]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function createStreamContent(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL'
+  location = 'us-central1',
+  model = 'gemini-pro'
 ) {
-  // [START aiplatform_gemini_content]
-
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -51,9 +46,8 @@ async function createStreamContent(
   for await (const item of responseStream.stream) {
     process.stdout.write(item.candidates[0].content.parts[0].text);
   }
-
-  // [END aiplatform_gemini_content]
 }
+// [END aiplatform_gemini_content]
 
 createStreamContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);

generative-ai/snippets/streamMultipartContent.js

@@ -12,26 +12,19 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+// [START aiplatform_gemini_get_started]
 const {VertexAI} = require('@google-cloud/vertexai');
 
+/**
+ * TODO(developer): Update these variables before running the sample.
+ */
 async function createStreamMultipartContent(
   projectId = 'PROJECT_ID',
-  location = 'LOCATION_ID',
-  model = 'MODEL',
+  location = 'us-central1',
+  model = 'gemini-pro-vision',
   image = 'gs://generativeai-downloads/images/scones.jpg',
   mimeType = 'image/jpeg'
 ) {
-  // [START aiplatform_gemini_get_started]
-
-  /**
-   * TODO(developer): Uncomment these variables before running the sample.
-   */
-  // const projectId = 'your-project-id';
-  // const location = 'us-central1';
-  // const model = 'chosen-genai-model';
-  // const image = 'gs://generativeai-downloads/images/scones.jpg'; // Google Cloud Storage image
-  // const mimeType = 'image/jpeg';
-
   // Initialize Vertex with your Cloud project and location
   const vertexAI = new VertexAI({project: projectId, location: location});
 
@@ -68,9 +61,8 @@ async function createStreamMultipartContent(
   for await (const item of responseStream.stream) {
     process.stdout.write(item.candidates[0].content.parts[0].text);
   }
-
-  // [END aiplatform_gemini_get_started]
 }
+// [END aiplatform_gemini_get_started]
 
 createStreamMultipartContent(...process.argv.slice(2)).catch(err => {
   console.error(err.message);