LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-30 09:03:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

working es

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

5
application/vectorstore/faiss.py
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)