From 3681af89dacff731408ef025e10754b681ecc8d6 Mon Sep 17 00:00:00 2001
From: Thejas-bhat <35959007+Thejas-bhat@users.noreply.github.com>
Date: Fri, 2 Feb 2024 21:19:03 +0530
Subject: [PATCH] introducing context for the vectorOptimize API (#47)

---
 vector_optimize.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/vector_optimize.go b/vector_optimize.go
index 5ad0941..e8f0055 100644
--- a/vector_optimize.go
+++ b/vector_optimize.go
@@ -17,6 +17,8 @@
 
 package index
 
+import "context"
+
 // VectorOptimizable represents an optional interface that implementable by
 // optimizable resources (e.g., VectorReaders, Searchers).  These
 // optimizable resources are provided the same OptimizableContext
@@ -25,7 +27,7 @@ package index
 // To avoid KNNSearchers' OptimizableContext being casted to ones used for
 // TFRs, term searchers, etc.
 type VectorOptimizable interface {
-	VectorOptimize(octx VectorOptimizableContext) (VectorOptimizableContext, error)
+	VectorOptimize(ctx context.Context, octx VectorOptimizableContext) (VectorOptimizableContext, error)
 }
 
 type VectorOptimizableContext interface {