From dad0879b96f433247d4ff0cdfa9f8d049be3c455 Mon Sep 17 00:00:00 2001
From: Rachit Chaudhary <65501028+Rachit-Chaudhary11@users.noreply.github.com>
Date: Fri, 21 Feb 2025 19:53:22 +0530
Subject: [PATCH 1/2] Add: GPU_BRUTE_FORCE index

---
 libs/milvus/langchain_milvus/vectorstores/milvus.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/libs/milvus/langchain_milvus/vectorstores/milvus.py b/libs/milvus/langchain_milvus/vectorstores/milvus.py
index 33ab4c4..a994b76 100644
--- a/libs/milvus/langchain_milvus/vectorstores/milvus.py
+++ b/libs/milvus/langchain_milvus/vectorstores/milvus.py
@@ -325,6 +325,7 @@ def __init__(
             },
             "GPU_IVF_FLAT": {"metric_type": "L2", "params": {"nprobe": 10}},
             "GPU_IVF_PQ": {"metric_type": "L2", "params": {"nprobe": 10}},
+            "GPU_BRUTE_FORCE": {"metric_type": "L2", "params": {"nprobe": 10}},
             "SPARSE_INVERTED_INDEX": {
                 "metric_type": "IP",
                 "params": {"drop_ratio_build": 0.2},

From 99d67c815de20f7bb33b614ef1adf794a4429b55 Mon Sep 17 00:00:00 2001
From: Rachit Chaudhary <65501028+Rachit-Chaudhary11@users.noreply.github.com>
Date: Wed, 26 Feb 2025 15:14:08 +0530
Subject: [PATCH 2/2] Enable on-disk indexing algorithm: DiskANN

---
 libs/milvus/langchain_milvus/vectorstores/milvus.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/libs/milvus/langchain_milvus/vectorstores/milvus.py b/libs/milvus/langchain_milvus/vectorstores/milvus.py
index a994b76..8b52ed2 100644
--- a/libs/milvus/langchain_milvus/vectorstores/milvus.py
+++ b/libs/milvus/langchain_milvus/vectorstores/milvus.py
@@ -313,6 +313,7 @@ def __init__(
             "ANNOY": {"metric_type": "L2", "params": {"search_k": 10}},
             "SCANN": {"metric_type": "L2", "params": {"search_k": 10}},
             "AUTOINDEX": {"metric_type": "L2", "params": {}},
+            "DISKANN":  {"metric_type": "L2", "params": {"search_list": 16}},
             "GPU_CAGRA": {
                 "metric_type": "L2",
                 "params": {