fix: specify outputDimensionality. (#3732)

Author: lee1premium-2024

ai-platform/snippets/predict-text-embeddings-preview.js

@@ -22,7 +22,7 @@ async function main(
   model = 'text-embedding-preview-0409',
   texts = 'banana bread?;banana muffins?',
   task = 'QUESTION_ANSWERING',
-  outputDimensionality = 256,
+  dimensionality = 256,
   apiEndpoint = 'us-central1-aiplatform.googleapis.com'
 ) {
   const aiplatform = require('@google-cloud/aiplatform');
@@ -31,7 +31,9 @@ async function main(
   const clientOptions = {apiEndpoint: apiEndpoint};
   const location = 'us-central1';
   const endpoint = `projects/${project}/locations/${location}/publishers/google/models/${model}`;
-  const parameters = helpers.toValue(outputDimensionality);
+  const parameters = helpers.toValue({
+    outputDimensionality: parseInt(dimensionality),
+  });
 
   async function callPredict() {
     const instances = texts

ai-platform/snippets/predict-text-embeddings.js

@@ -22,7 +22,7 @@ async function main(
   model = 'text-embedding-004',
   texts = 'banana bread?;banana muffins?',
   task = 'QUESTION_ANSWERING',
-  outputDimensionality = 0,
+  dimensionality = 0,
   apiEndpoint = 'us-central1-aiplatform.googleapis.com'
 ) {
   const aiplatform = require('@google-cloud/aiplatform');
@@ -31,15 +31,14 @@ async function main(
   const clientOptions = {apiEndpoint: apiEndpoint};
   const location = 'us-central1';
   const endpoint = `projects/${project}/locations/${location}/publishers/google/models/${model}`;
-  const parameters =
-    outputDimensionality > 0
-      ? helpers.toValue(outputDimensionality)
-      : helpers.toValue(256);
 
   async function callPredict() {
     const instances = texts
       .split(';')
       .map(e => helpers.toValue({content: e, task_type: task}));
+    const parameters = helpers.toValue(
+      dimensionality > 0 ? {outputDimensionality: parseInt(dimensionality)} : {}
+    );
     const request = {endpoint, instances, parameters};
     const client = new PredictionServiceClient(clientOptions);
     const [response] = await client.predict(request);