From 0a0054f361eee2bd438859eb44082589ad272d06 Mon Sep 17 00:00:00 2001 From: onewesong Date: Thu, 28 Nov 2024 10:52:21 +0800 Subject: [PATCH] fix: pg_similarity_search error: Relevance scores must be between 0 and 1 #5023 --- .../chatchat/server/knowledge_base/kb_service/pg_kb_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/chatchat-server/chatchat/server/knowledge_base/kb_service/pg_kb_service.py b/libs/chatchat-server/chatchat/server/knowledge_base/kb_service/pg_kb_service.py index 2f40f13ef6..bdff11b532 100644 --- a/libs/chatchat-server/chatchat/server/knowledge_base/kb_service/pg_kb_service.py +++ b/libs/chatchat-server/chatchat/server/knowledge_base/kb_service/pg_kb_service.py @@ -29,7 +29,7 @@ def _load_pg_vector(self): self.pg_vector = PGVector( embedding_function=get_Embeddings(self.embed_model), collection_name=self.kb_name, - distance_strategy=DistanceStrategy.EUCLIDEAN, + distance_strategy=DistanceStrategy.COSINE, connection=PGKBService.engine, connection_string=Settings.kb_settings.kbs_config.get("pg").get("connection_uri"), )