From df3f05ba8ff1385a48576ae2703124de86041d0e Mon Sep 17 00:00:00 2001
From: Thejas-bhat <thejas.orkombu@couchbase.com>
Date: Thu, 21 May 2026 13:49:58 -0700
Subject: [PATCH] use training params for fast merge

---
 vector.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/vector.go b/vector.go
index a19f463..70ef231 100644
--- a/vector.go
+++ b/vector.go
@@ -95,6 +95,10 @@ func OptimizationRequiresBinaryIndex(optimization string) bool {
 const TrainedIndexFileName = "trained_index"
 const TrainingKey = "_training"
 
+type TrainingParams struct {
+	NumCentroids int
+}
+
 const TrainedIndexCallback = "_trained_index_callback"
 
 type TrainedIndexCallbackFn func(string) (interface{}, error)