Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-08 07:10:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle kwargs in FAISS.load_local() (#6987)

Browse Source 
- Description: This allows parameters such as `relevance_score_fn` to be
passed to the `FAISS` constructor via the `load_local()` class method.
-  Tag maintainer: @rlancemartin @eyurtsev
This commit is contained in:
Mike Salvatore 2023-07-05 15:56:40 -04:00 committed by GitHub
parent a2f191a322
commit 3ae11b7582
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
langchain/vectorstores/faiss.py
View File

@ -618,7 +618,11 @@ class FAISS(VectorStore):
@classmethod
def load_local(
cls, folder_path: str, embeddings: Embeddings, index_name: str = "index"
cls,
folder_path: str,
embeddings: Embeddings,
index_name: str = "index",
**kwargs: Any,
) -> FAISS:
"""Load FAISS index, docstore, and index_to_docstore_id from disk.
@ -638,7 +642,9 @@ class FAISS(VectorStore):
# load docstore and index_to_docstore_id
with open(path / "{index_name}.pkl".format(index_name=index_name), "rb") as f:
docstore, index_to_docstore_id = pickle.load(f)
return cls(embeddings.embed_query, index, docstore, index_to_docstore_id)
return cls(
embeddings.embed_query, index, docstore, index_to_docstore_id, **kwargs
)
def _similarity_search_with_relevance_scores(
self,