From 2604341e9e45d0e6b6621f9ca9f6495105d2c4e3 Mon Sep 17 00:00:00 2001 From: zqt <1178747941@qq.com> Date: Thu, 24 Aug 2023 22:35:30 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4KBService=20=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3=E5=87=BD=E6=95=B0do=5Fsearch=20=E5=B9=B6=E9=80=82?= =?UTF-8?q?=E9=85=8Dpg?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/knowledge_base/kb_service/base.py | 1 + server/knowledge_base/kb_service/pg_kb_service.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/server/knowledge_base/kb_service/base.py b/server/knowledge_base/kb_service/base.py index 9af5b0e..8d1de48 100644 --- a/server/knowledge_base/kb_service/base.py +++ b/server/knowledge_base/kb_service/base.py @@ -157,6 +157,7 @@ class KBService(ABC): def do_search(self, query: str, top_k: int, + score_threshold: float, embeddings: Embeddings, ) -> List[Document]: """ diff --git a/server/knowledge_base/kb_service/pg_kb_service.py b/server/knowledge_base/kb_service/pg_kb_service.py index 38d8065..6876bd8 100644 --- a/server/knowledge_base/kb_service/pg_kb_service.py +++ b/server/knowledge_base/kb_service/pg_kb_service.py @@ -43,7 +43,7 @@ class PGKBService(KBService): ''')) connect.commit() - def do_search(self, query: str, top_k: int, embeddings: Embeddings): + def do_search(self, query: str, top_k: int, score_threshold: float, embeddings: Embeddings): # todo: support score threshold self._load_pg_vector(embeddings=embeddings) return self.pg_vector.similarity_search_with_score(query, top_k)