added changes

darkcoderrises · darkcoderrises · commit 4fa1f8338996 · 2025-06-26T07:21:04.000+05:30
diff --git a/tok/kmeans/kmeans.go b/tok/kmeans/kmeans.go
@@ -33,7 +33,7 @@ func (km *Kmeans[T]) AddVector(vec []T) error {
 }
 
 func (km *Kmeans[T]) FindIndexForSearch(vec []T) ([]int, error) {
-	res := make([]int, len(km.centroids.centroids))
+	res := make([]int, km.NumSeedVectors())
 	for i := range res {
 		res[i] = i
 	}
diff --git a/tok/partitioned_hnsw/partitioned_hnsw.go b/tok/partitioned_hnsw/partitioned_hnsw.go
@@ -169,7 +169,7 @@ func (ph *partitionedHNSW[T]) Search(ctx context.Context, txn index.CacheType, q
 		}(index)
 	}
 	wg.Wait()
-	fmt.Println("Result:", res)
+	fmt.Println("Result:", res, indexes)
 	return ph.clusterMap[0].MergeResults(ctx, txn, res, query, maxResults, filter)
 }
 

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@ func (km *Kmeans[T]) AddVector(vec []T) error {`
`33`	`33`	`}`
`34`	`34`
`35`	`35`	`func (km *Kmeans[T]) FindIndexForSearch(vec []T) ([]int, error) {`
`36`		`- res := make([]int, len(km.centroids.centroids))`
	`36`	`+ res := make([]int, km.NumSeedVectors())`
`37`	`37`	`for i := range res {`
`38`	`38`	`res[i] = i`
`39`	`39`	`}`
Original file line number	Diff line number	Diff line change
`@@ -169,7 +169,7 @@ func (ph *partitionedHNSW[T]) Search(ctx context.Context, txn index.CacheType, q`
`169`	`169`	`}(index)`
`170`	`170`	`}`
`171`	`171`	`wg.Wait()`
`172`		`- fmt.Println("Result:", res)`
	`172`	`+ fmt.Println("Result:", res, indexes)`
`173`	`173`	`return ph.clusterMap[0].MergeResults(ctx, txn, res, query, maxResults, filter)`
`174`	`174`	`}`
`175`	`175`