fix: Specify task_type properly. (#9404)

lee1premium-2024 · web-flow · commit dfa93d92f5dc · 2024-07-09T08:20:54.000-04:00
diff --git a/aiplatform/src/main/java/aiplatform/PredictTextEmbeddingsSample.java b/aiplatform/src/main/java/aiplatform/PredictTextEmbeddingsSample.java
@@ -84,7 +84,7 @@ public static List<List<Float>> predictTextEmbeddings(
                 .setStructValue(
                     Struct.newBuilder()
                         .putFields("content", valueOf(texts.get(i)))
-                        .putFields("taskType", valueOf(task))
+                        .putFields("task_type", valueOf(task))
                         .build()));
       }
       PredictResponse response = client.predict(request.build());
diff --git a/aiplatform/src/main/java/aiplatform/PredictTextEmbeddingsSamplePreview.java b/aiplatform/src/main/java/aiplatform/PredictTextEmbeddingsSamplePreview.java
@@ -85,7 +85,7 @@ public static List<List<Float>> predictTextEmbeddings(
                 .setStructValue(
                     Struct.newBuilder()
                         .putFields("content", valueOf(texts.get(i)))
-                        .putFields("taskType", valueOf(task))
+                        .putFields("task_type", valueOf(task))
                         .build()));
       }
       PredictResponse response = client.predict(request.build());

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ public static List<List<Float>> predictTextEmbeddings(`
`84`	`84`	`.setStructValue(`
`85`	`85`	`Struct.newBuilder()`
`86`	`86`	`.putFields("content", valueOf(texts.get(i)))`
`87`		`- .putFields("taskType", valueOf(task))`
	`87`	`+ .putFields("task_type", valueOf(task))`
`88`	`88`	`.build()));`
`89`	`89`	`}`
`90`	`90`	`PredictResponse response = client.predict(request.build());`
Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ public static List<List<Float>> predictTextEmbeddings(`
`85`	`85`	`.setStructValue(`
`86`	`86`	`Struct.newBuilder()`
`87`	`87`	`.putFields("content", valueOf(texts.get(i)))`
`88`		`- .putFields("taskType", valueOf(task))`
	`88`	`+ .putFields("task_type", valueOf(task))`
`89`	`89`	`.build()));`
`90`	`90`	`}`
`91`	`91`	`PredictResponse response = client.predict(request.build());`