feat: GetItemAndFetchVectorsBatch API (#235)

momentohq · Nov 13, 2023 · 72cb995 · 72cb995
1 parent 50b7cf3
commit 72cb995
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/proto/vectorindex.proto b/proto/vectorindex.proto
@@ -8,6 +8,7 @@ service VectorIndex {
     rpc Search(_SearchRequest) returns (_SearchResponse) {}
     rpc SearchAndFetchVectors(_SearchAndFetchVectorsRequest) returns (_SearchAndFetchVectorsResponse) {}
     rpc GetItemBatch(_GetItemBatchRequest) returns (_GetItemBatchResponse) {}
+    rpc GetItemAndFetchVectorsBatch(_GetItemAndFetchVectorsBatchRequest) returns (_GetItemAndFetchVectorsBatchResponse) {}
 }
 
 message _Item {
@@ -131,3 +132,26 @@ message _ItemResponse {
 message _GetItemBatchResponse {
   repeated _ItemResponse item_response = 1;
 }
+
+message _GetItemAndFetchVectorsBatchRequest {
+    string index_name = 1;
+    repeated string ids = 2;
+    _MetadataRequest metadata_fields = 3;
+}
+
+message _ItemWithVectorResponse {
+    message _Miss {}
+    message _Hit {
+        string id = 1;
+        _Vector vector = 2;
+        repeated _Metadata metadata = 3;
+    }
+    oneof response {
+        _Miss miss = 1;
+        _Hit hit = 2;
+    }
+}
+
+message _GetItemAndFetchVectorsBatchResponse {
+    repeated _ItemWithVectorResponse item_with_vector_response = 1;
+}