feat: update the v1 service definition to add the embedding_id field in MatchRequest.

vertex-sdk-bot · copybara-github · commit 5a1146e02ec9 · 2023-03-22T13:33:49.000-07:00
PiperOrigin-RevId: 518658960
diff --git a/google/cloud/aiplatform/matching_engine/_protos/match_service.proto b/google/cloud/aiplatform/matching_engine/_protos/match_service.proto
@@ -68,6 +68,13 @@ message MatchRequest {
   // embedding. Set this value may impact the query performance (e.g, increase
   // query latency, etc).
   bool embedding_enabled = 8;
+
+  // The embedding id.
+  // If embedding values (float_val) is set, will use embedding values do
+  // nearest neighbor search. If embedding values isn't set, will first uses
+  // embedding_id to lookup embedding values from dataset, if embedding exist in
+  // the dataset, do nearest neighbor search.
+  string embedding_id = 10 [(datapol.semantic_type) = ST_IDENTIFYING_ID];
 }
 
 // Embedding on query result.
