apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-11 15:15:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

pass kwargs in faiss search

Browse source
This commit is contained in:
seehi 2023-09-04 11:44:08 +08:00
parent 206bc252de
commit 1f0cf924c1
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
metagpt/document_store/faiss_store.py
View file

@ -51,7 +51,7 @@ class FaissStore(LocalStore):
store.index = index
def search(self, query, expand_cols=False, sep='\n', *args, k=5, **kwargs):
rsp = self.store.similarity_search(query, k=k)
rsp = self.store.similarity_search(query, k=k, **kwargs)
logger.debug(rsp)
if expand_cols:
return str(sep.join([f"{x.page_content}: {x.metadata}" for x in rsp]))