From 95d9fb0ee968f969b2c44f997105a019425db050 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E9=B9=8F?=
 <37921589+shutter-cp@users.noreply.github.com>
Date: Fri, 18 Aug 2023 14:41:11 +0800
Subject: [PATCH] fix bug 1159
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修复使用使用pg存储向量时，向量检索错误 {TypeError}TypeError("'Document' object is not subscriptable")
---
 server/knowledge_base/kb_service/pg_kb_service.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/knowledge_base/kb_service/pg_kb_service.py b/server/knowledge_base/kb_service/pg_kb_service.py
index a3126ec..a3f9318 100644
--- 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):
         """