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

pr 2775 maybe need？

Browse Source
This commit is contained in:
zR 2024-01-20 22:20:02 +08:00
parent 4a77880ee1
commit 8c2019f8b9
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server/knowledge_base/kb_service/milvus_kb_service.py
View File

@ -18,13 +18,10 @@ class MilvusKBService(KBService):
from pymilvus import Collection from pymilvus import Collection
return Collection(milvus_name) return Collection(milvus_name)
# def save_vector_store(self):
# if self.milvus.col:
# self.milvus.col.flush()
def get_doc_by_ids(self, ids: List[str]) -> List[Document]: def get_doc_by_ids(self, ids: List[str]) -> List[Document]:
result = [] result = []
if self.milvus.col: if self.milvus.col:
# ids = [int(id) for id in ids] # for milvus if needed #pr 2725
data_list = self.milvus.col.query(expr=f'pk in {ids}', output_fields=["*"]) data_list = self.milvus.col.query(expr=f'pk in {ids}', output_fields=["*"])
for data in data_list: for data in data_list:
text = data.pop("text") text = data.pop("text")

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

@ -19,7 +19,7 @@ class ZillizKBService(KBService):
def get_doc_by_ids(self, ids: List[str]) -> List[Document]: def get_doc_by_ids(self, ids: List[str]) -> List[Document]:
result = [] result = []
if self.zilliz.col: if self.zilliz.col:
ids = [int(id) for id in ids] # for milvus if needed # ids = [int(id) for id in ids] # for zilliz if needed #pr 2725
data_list = self.zilliz.col.query(expr=f'pk in {ids}', output_fields=["*"]) data_list = self.zilliz.col.query(expr=f'pk in {ids}', output_fields=["*"])
for data in data_list: for data in data_list:
text = data.pop("text") text = data.pop("text")