Merge pull request #3263 from GoogleCloudPlatform/b279595025-update

fix(aiplatform): update model parameters and instances
GoogleCloudPlatform · Jun 9, 2023 · 8b6aff8 · 8b6aff8
2 parents 5d14691 + 89b2d00
commit 8b6aff8
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 10 deletions.
diff --git a/ai-platform/snippets/predict-code-chat.js b/ai-platform/snippets/predict-code-chat.js
@@ -46,18 +46,20 @@ async function main(project, location = 'us-central1') {
     // Configure the parent resource
     const endpoint = `projects/${project}/locations/${location}/publishers/${publisher}/models/${model}`;
 
+    // Learn more about creating prompts to work with a code chat model at:
+    // https://cloud.google.com/vertex-ai/docs/generative-ai/code/code-chat-prompts
     const prompt = {
       messages: [
         {
-          author: 'content',
+          author: 'user',
           content: 'Hi, how are you?',
         },
         {
           author: 'system',
           content: 'I am doing good. What can I help you in the coding world?',
         },
         {
-          author: 'content',
+          author: 'user',
           content:
             'Please help write a function to calculate the min of two numbers',
         },
@@ -67,8 +69,8 @@ async function main(project, location = 'us-central1') {
     const instances = [instanceValue];
 
     const parameter = {
-      temperature: 0.2,
-      maxOutputTokens: 256,
+      temperature: 0.5,
+      maxOutputTokens: 1024,
     };
     const parameters = helpers.toValue(parameter);
 

diff --git a/ai-platform/snippets/predict-code-completion-comment.js b/ai-platform/snippets/predict-code-completion-comment.js
@@ -50,14 +50,14 @@ async function main(project, location = 'us-central1') {
       prefix:
         'def reverse_string(s): \
           return s[::-1] \
-        //This function',
+        #This function',
     };
     const instanceValue = helpers.toValue(prompt);
     const instances = [instanceValue];
 
     const parameter = {
       temperature: 0.2,
-      maxOutputTokens: 256,
+      maxOutputTokens: 64,
     };
     const parameters = helpers.toValue(parameter);
 

diff --git a/ai-platform/snippets/predict-code-completion-test-function.js b/ai-platform/snippets/predict-code-completion-test-function.js
@@ -57,7 +57,7 @@ async function main(project, location = 'us-central1') {
 
     const parameter = {
       temperature: 0.2,
-      maxOutputTokens: 256,
+      maxOutputTokens: 64,
     };
     const parameters = helpers.toValue(parameter);
 

diff --git a/ai-platform/snippets/predict-code-generation-function.js b/ai-platform/snippets/predict-code-generation-function.js
@@ -53,7 +53,7 @@ async function main(project, location = 'us-central1') {
     const instances = [instanceValue];
 
     const parameter = {
-      temperature: 0.2,
+      temperature: 0.5,
       maxOutputTokens: 256,
     };
     const parameters = helpers.toValue(parameter);

diff --git a/ai-platform/snippets/predict-code-generation-unittest.js b/ai-platform/snippets/predict-code-generation-unittest.js
@@ -65,8 +65,8 @@ async function main(project, location = 'us-central1') {
     const instances = [instanceValue];
 
     const parameter = {
-      temperature: 0.2,
-      maxOutputTokens: 512,
+      temperature: 0.5,
+      maxOutputTokens: 256,
     };
     const parameters = helpers.toValue(parameter);