From 6b360dfc55ee06ae150525c0d1e47a98756c404e Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 30 Sep 2025 23:02:18 +0100
Subject: [PATCH] Get Milvus to use ANN

---
 .../trustgraph/direct/milvus_doc_embeddings.py      | 13 ++++---------
 .../trustgraph/direct/milvus_graph_embeddings.py    | 11 ++---------
 2 files changed, 6 insertions(+), 18 deletions(-)

diff --git a/trustgraph-flow/trustgraph/direct/milvus_doc_embeddings.py b/trustgraph-flow/trustgraph/direct/milvus_doc_embeddings.py
index 131c114a..08b7b56f 100644
--- a/trustgraph-flow/trustgraph/direct/milvus_doc_embeddings.py
+++ b/trustgraph-flow/trustgraph/direct/milvus_doc_embeddings.py
@@ -120,6 +120,8 @@ class DocVectors:
 
         dim = len(embeds)
 
+        print("INSERT VEC", dim, flush=True)
+
         if (dim, user, collection) not in self.collections:
             self.init_collection(dim, user, collection)
     
@@ -144,14 +146,6 @@ class DocVectors:
 
         coll = self.collections[(dim, user, collection)]
 
-        search_params = {
-            "metric_type": "COSINE",
-            "params": {
-                "radius": 0.1,
-                "range_filter": 0.8
-            }
-        }
-
         logger.debug("Loading...")
         self.client.load_collection(
             collection_name=coll,
@@ -161,10 +155,11 @@ class DocVectors:
 
         res = self.client.search(
             collection_name=coll,
+            anns_field="vector",
             data=[embeds],
             limit=limit,
             output_fields=fields,
-            search_params=search_params,
+            search_params={ "metric_type": "COSINE" },
         )[0]
 
 
diff --git a/trustgraph-flow/trustgraph/direct/milvus_graph_embeddings.py b/trustgraph-flow/trustgraph/direct/milvus_graph_embeddings.py
index 7c2cb55b..b3ed2a9f 100644
--- a/trustgraph-flow/trustgraph/direct/milvus_graph_embeddings.py
+++ b/trustgraph-flow/trustgraph/direct/milvus_graph_embeddings.py
@@ -144,14 +144,6 @@ class EntityVectors:
 
         coll = self.collections[(dim, user, collection)]
 
-        search_params = {
-            "metric_type": "COSINE",
-            "params": {
-                "radius": 0.1,
-                "range_filter": 0.8
-            }
-        }
-
         logger.debug("Loading...")
         self.client.load_collection(
             collection_name=coll,
@@ -161,10 +153,11 @@ class EntityVectors:
 
         res = self.client.search(
             collection_name=coll,
+            anns_field="vector",
             data=[embeds],
             limit=limit,
             output_fields=fields,
-            search_params=search_params,
+            search_params={ "metric_type": "COSINE" },
         )[0]