Skip to content

Commit 51a760f

Browse files
lingyinwcopybara-github
authored andcommitted
feat: update the v1 service definition to add embedding_metadata.
PiperOrigin-RevId: 856673502
1 parent feeb54d commit 51a760f

1 file changed

Lines changed: 11 additions & 11 deletions

File tree

google/cloud/aiplatform/matching_engine/_protos/match_service.proto

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,14 @@ service MatchService {
2121
}
2222

2323
// Feature embedding vector for sparse index. An array of numbers whose values
24-
// are located in the specified dimensions.
25-
message SparseEmbedding {
24+
// are located in the specified dimensions.
25+
message SparseEmbedding {
26+
// The list of embedding values of the sparse vector.
27+
repeated float float_val = 1;
2628

27-
// The list of embedding values of the sparse vector.
28-
repeated float float_val = 1;
29-
30-
// The list of indexes for the embedding values of the sparse vector.
31-
repeated int64 dimension = 2;
32-
}
29+
// The list of indexes for the embedding values of the sparse vector.
30+
repeated int64 dimension = 2;
31+
}
3332

3433
// Parameters for a match query.
3534
message MatchRequest {
@@ -50,7 +49,6 @@ message MatchRequest {
5049

5150
// Parameters for RRF algorithm that combines search results.
5251
message RRF {
53-
5452
// Users can provide an alpha value to give more weight to sparse vs dense.
5553
// For example, if the alpha is 0, we don't return dense at all, if it's 1,
5654
// we don't return sparse at all.
@@ -70,7 +68,7 @@ message MatchRequest {
7068
// The list of restricts.
7169
repeated Namespace restricts = 4;
7270

73-
//The list of numeric restricts.
71+
// The list of numeric restricts.
7472
repeated NumericNamespace numeric_restricts = 11;
7573

7674
// Crowding is a constraint on a neighbor list produced by nearest neighbor
@@ -128,6 +126,9 @@ message Embedding {
128126
// to return per crowding attribute value
129127
// (per_crowding_attribute_num_neighbors) is configured per-query.
130128
int64 crowding_attribute = 4;
129+
130+
// The key-value map of additional metadata for the datapoint.
131+
google.protobuf.Struct embedding_metadata = 7;
131132
}
132133

133134
// Response of a match query.
@@ -238,7 +239,6 @@ message Namespace {
238239
// eligible for each matching query, overall query is an AND across namespaces.
239240
// This uses numeric comparisons.
240241
message NumericNamespace {
241-
242242
// The string name of the namespace that this proto is specifying,
243243
// such as "size" or "cost".
244244
string name = 1;

0 commit comments

Comments
 (0)