RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-07 07:23:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

调整KBService 接口函数do_search 并适配pg

Browse Source
This commit is contained in:
zqt 2023-08-24 22:35:30 +08:00
parent 24e7351d32
commit 2604341e9e
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/knowledge_base/kb_service/base.py
View File

@ -157,6 +157,7 @@ class KBService(ABC):
def do_search(self, def do_search(self,
query: str, query: str,
top_k: int, top_k: int,
score_threshold: float,
embeddings: Embeddings, embeddings: Embeddings,
) -> List[Document]: ) -> List[Document]:
""" """

2
server/knowledge_base/kb_service/pg_kb_service.py
View File

@ -43,7 +43,7 @@ class PGKBService(KBService):
''')) '''))
connect.commit() 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 # todo: support score threshold
self._load_pg_vector(embeddings=embeddings) self._load_pg_vector(embeddings=embeddings)
return self.pg_vector.similarity_search_with_score(query, top_k) return self.pg_vector.similarity_search_with_score(query, top_k)