fix: pg_similarity_search error: Relevance scores must be between 0 a…

…nd 1 chatchat-space#5023
onewesong · Nov 28, 2024 · 0a0054f · 0a0054f
1 parent 223915c
commit 0a0054f
Showing 1 changed file with 1 addition and 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
@@ -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"),
         )