Merge pull request #1160 from shutter-cp/master

fix bug 1159
2023-08-22 16:38:06 +08:00 · 2023-08-22 16:38:06 +08:00 · 11856cc542
parent c571585ffd 95d9fb0ee9
commit 11856cc542
1 changed files with 1 additions and 1 deletions
--- a/server/knowledge_base/kb_service/pg_kb_service.py
+++ b/server/knowledge_base/kb_service/pg_kb_service.py
@ -46,7 +46,7 @@ class PGKBService(KBService):
    def do_search(self, query: str, top_k: int, score_threshold: float, embeddings: Embeddings) -> List[Document]:
        # todo: support score threshold
        self._load_pg_vector(embeddings=embeddings)
-        return self.pg_vector.similarity_search(query, top_k)
+        return self.pg_vector.similarity_search_with_score(query, top_k)

    def add_doc(self, kb_file: KnowledgeFile):
        """