From 8d326d34b3230c0c6ce4f684c52f59f6de401a0d Mon Sep 17 00:00:00 2001 From: cybermaggedon Date: Tue, 10 Dec 2024 22:15:42 +0000 Subject: [PATCH] Use Cosine similarity (#209) --- .../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: