MB-66396: Augment VectorIndex interface

abhinavdangeti · abhinavdangeti · commit 629cd765d8b2 · 2025-08-18T12:15:53.000-06:00
Requires: blevesearch/bleve_index_api#71
diff --git a/go.mod b/go.mod
@@ -4,7 +4,7 @@ go 1.21
 
 require (
 	github.com/RoaringBitmap/roaring/v2 v2.4.5
-	github.com/blevesearch/bleve_index_api v1.2.9
+	github.com/blevesearch/bleve_index_api v1.2.10-0.20250818181143-42826d84beea
 )
 
 require (
diff --git a/go.sum b/go.sum
@@ -2,8 +2,8 @@ github.com/RoaringBitmap/roaring/v2 v2.4.5 h1:uGrrMreGjvAtTBobc0g5IrW1D5ldxDQYe2
 github.com/RoaringBitmap/roaring/v2 v2.4.5/go.mod h1:FiJcsfkGje/nZBZgCu0ZxCPOKD/hVXDS2dXi7/eUFE0=
 github.com/bits-and-blooms/bitset v1.12.0 h1:U/q1fAF7xXRhFCrhROzIfffYnu+dlS38vCZtmFVPHmA=
 github.com/bits-and-blooms/bitset v1.12.0/go.mod h1:7hO7Gc7Pp1vODcmWvKMRA9BNmbv6a/7QIWpPxHddWR8=
-github.com/blevesearch/bleve_index_api v1.2.9 h1:WqD3kvYwnlYLv8sTdH+AF7n/L4v969Cek68+wZnYj4Q=
-github.com/blevesearch/bleve_index_api v1.2.9/go.mod h1:rKQDl4u51uwafZxFrPD1R7xFOwKnzZW7s/LSeK4lgo0=
+github.com/blevesearch/bleve_index_api v1.2.10-0.20250818181143-42826d84beea h1:CyyHwFaoPBkhmUX3+Y+NV6F2PuYg5nH/ptgDlrvzFvc=
+github.com/blevesearch/bleve_index_api v1.2.10-0.20250818181143-42826d84beea/go.mod h1:rKQDl4u51uwafZxFrPD1R7xFOwKnzZW7s/LSeK4lgo0=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
diff --git a/segment_vector.go b/segment_vector.go
@@ -20,6 +20,7 @@ package segment
 import (
 	"encoding/json"
 
+	index "github.com/blevesearch/bleve_index_api"
 	"github.com/RoaringBitmap/roaring/v2"
 )
 
@@ -64,6 +65,8 @@ type VectorIndex interface {
 		params json.RawMessage) (VecPostingsList, error)
 	Close()
 	Size() uint64
+
+	ObtainTopKCentroidCardinalitiesFromIVFIndex(limit int) ([]index.CentroidCardinality, error)
 }
 
 type VectorSegment interface {

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ go 1.21`
`4`	`4`
`5`	`5`	`require (`
`6`	`6`	`github.com/RoaringBitmap/roaring/v2 v2.4.5`
`7`		`- github.com/blevesearch/bleve_index_api v1.2.9`
	`7`	`+ github.com/blevesearch/bleve_index_api v1.2.10-0.20250818181143-42826d84beea`
`8`	`8`	`)`
`9`	`9`
`10`	`10`	`require (`
Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,7 @@ package segment`
`20`	`20`	`import (`
`21`	`21`	`"encoding/json"`
`22`	`22`
	`23`	`+ index "github.com/blevesearch/bleve_index_api"`
`23`	`24`	`"github.com/RoaringBitmap/roaring/v2"`
`24`	`25`	`)`
`25`	`26`
`@@ -64,6 +65,8 @@ type VectorIndex interface {`
`64`	`65`	`params json.RawMessage) (VecPostingsList, error)`
`65`	`66`	`Close()`
`66`	`67`	`Size() uint64`
	`68`	`+`
	`69`	`+ ObtainTopKCentroidCardinalitiesFromIVFIndex(limit int) ([]index.CentroidCardinality, error)`
`67`	`70`	`}`
`68`	`71`
`69`	`72`	`type VectorSegment interface {`