From 60ed484b67957624a21275aee4f365f9f0183e31 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 10 Dec 2024 16:44:03 +0000 Subject: [PATCH] Use Cosine similarity --- .../trustgraph/storage/graph_embeddings/qdrant/write.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py b/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py index e27c2516..47b53979 100755 --- a/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py +++ b/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py @@ -62,7 +62,7 @@ class Processor(Consumer): self.client.create_collection( collection_name=collection, vectors_config=VectorParams( - size=dim, distance=Distance.DOT + size=dim, distance=Distance.COSINE ), ) except Exception as e: