remove trailing whitespace, sort imports, remove unused arguments

This commit is contained in:
asoderlind 2023-10-16 11:22:42 +02:00
parent e2a8ca143a
commit e73636bef3

View File

@ -1,5 +1,5 @@
from application.vectorstore.base import BaseVectorStore
from langchain.vectorstores import FAISS
from application.vectorstore.base import BaseVectorStore
from application.core.settings import settings
class FaissStore(BaseVectorStore):
@ -27,7 +27,7 @@ class FaissStore(BaseVectorStore):
def save_local(self, *args, **kwargs):
return self.docsearch.save_local(*args, **kwargs)
def assert_embedding_dimensions(self, embeddings, *args, **kwargs):
def assert_embedding_dimensions(self, embeddings):
"""
Check that the word embedding dimension of the docsearch index matches
the dimension of the word embeddings used