diff --git a/easy-agents-store/easy-agents-store-milvus/src/main/java/com/easyagents/store/milvus/MilvusVectorStore.java b/easy-agents-store/easy-agents-store-milvus/src/main/java/com/easyagents/store/milvus/MilvusVectorStore.java index aadd67e..aed63f1 100644 --- a/easy-agents-store/easy-agents-store-milvus/src/main/java/com/easyagents/store/milvus/MilvusVectorStore.java +++ b/easy-agents-store/easy-agents-store-milvus/src/main/java/com/easyagents/store/milvus/MilvusVectorStore.java @@ -47,7 +47,7 @@ import java.util.*; /** * Milvus vector store based on Milvus Java SDK v2. */ -public class MilvusVectorStore extends DocumentStore { +public class MilvusVectorStore extends DocumentStore implements AutoCloseable { private static final Logger LOG = LoggerFactory.getLogger(MilvusVectorStore.class); private static final long LOAD_TIMEOUT_MS = 30_000L; @@ -546,4 +546,14 @@ public class MilvusVectorStore extends DocumentStore { return false; } } + + @Override + public void close() { + try { + client.close(1L); + } catch (InterruptedException e) { + Thread.currentThread().interrupt(); + LOG.warn("Interrupted while closing Milvus client. uri={}", config.getUri(), e); + } + } }