tiennm99/DocsGPT
1
0 Fork 0
mirror of https://github.com/tiennm99/DocsGPT.git synced 2026-10-14 12:16:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

working es

Browse Source
This commit is contained in:
Alex committed 2023-09-29 00:32:19 +01:00
1 parent c1c54f4848
commit 783e7f6939
6 files changed
+235 -31
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

application/vectorstore/faiss.py
+4 -1
View File
@@ -8,8 +8,11 @@ class FaissStore(BaseVectorStore):
super().__init__()
self.path = path
self.docsearch = FAISS.load_local(
self.path, self._get_docsearch(settings.EMBEDDINGS_NAME, settings.EMBEDDINGS_KEY)
self.path, self._get_embeddings(settings.EMBEDDINGS_NAME, settings.EMBEDDINGS_KEY)
)
def search(self, *args, **kwargs):
return self.docsearch.similarity_search(*args, **kwargs)
def add_texts(self, *args, **kwargs):
return self.docsearch.add_texts(*args, **kwargs)