fix: Validate embedding output dimensionality.

Author: lee1premium-2024

ai-platform/snippets/package.json

@@ -10,7 +10,7 @@
     "*.js"
   ],
   "scripts": {
-    "test": "c8 mocha -p -j 2 --timeout 2400000 test/*.js"
+    "test": "c8 mocha -p -j 2 --timeout 2400000 test/*embeddings*.js"
   },
   "dependencies": {
     "@google-cloud/aiplatform": "^3.0.0",

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

@@ -43,11 +43,11 @@ async function main(
     const client = new PredictionServiceClient(clientOptions);
     const [response] = await client.predict(request);
     const predictions = response.predictions;
-    const embeddings = predictions.map(p =>
-      p.structValue.fields.embeddings.structValue.fields.values.listValue.values.map(
-        v => v.numberValue
-      )
-    );
+    const embeddings = predictions.map(p => {
+      const embeddingsProto = p.structValue.fields.embeddings;
+      const valuesProto = embeddingsProto.structValue.fields.values;
+      return valuesProto.listValue.values.map(v => v.numberValue);
+    });
     console.log('Got embeddings: \n' + JSON.stringify(embeddings));
   }
 

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

@@ -42,13 +42,12 @@ async function main(
     const request = {endpoint, instances, parameters};
     const client = new PredictionServiceClient(clientOptions);
     const [response] = await client.predict(request);
-    console.log('Got predict response');
     const predictions = response.predictions;
-    const embeddings = predictions.map(p =>
-      p.structValue.fields.embeddings.structValue.fields.values.listValue.values.map(
-        v => v.numberValue
-      )
-    );
+    const embeddings = predictions.map(p => {
+      const embeddingsProto = p.structValue.fields.embeddings;
+      const valuesProto = embeddingsProto.structValue.fields.values;
+      return valuesProto.listValue.values.map(v => v.numberValue);
+    });
     console.log('Got embeddings: \n' + JSON.stringify(embeddings));
   }