From 444d50f751fca1fc93f3a8d8c7a73cf0e15c0ab9 Mon Sep 17 00:00:00 2001 From: Alex Date: Mon, 9 Sep 2024 16:43:20 +0100 Subject: [PATCH] fix: faiss source name --- application/vectorstore/faiss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/application/vectorstore/faiss.py b/application/vectorstore/faiss.py index 957e61ef..b504ebf8 100644 --- a/application/vectorstore/faiss.py +++ b/application/vectorstore/faiss.py @@ -14,9 +14,9 @@ def get_vectorstore(path): class FaissStore(BaseVectorStore): - def __init__(self, path, embeddings_key, docs_init=None): + def __init__(self, source_id, embeddings_key, docs_init=None): super().__init__() - self.path = get_vectorstore(path) + self.path = get_vectorstore(source_id) embeddings = self._get_embeddings(settings.EMBEDDINGS_NAME, embeddings_key) if docs_init: self.docsearch = FAISS.from_documents(