Former-commit-id: a89cb9ef95
pull/47/head
Kye 2 years ago
parent cada7651b0
commit 2b3e77c3df

@ -91,8 +91,9 @@ class WorkerNodeInitializer:
def initialize_vectorstore(self): def initialize_vectorstore(self):
try: try:
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embedding_size = self.embedding_size embedding_size = self.embedding_size
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embedding_size = embedding_size
index = faiss.IndexFlatL2(embedding_size=embedding_size) index = faiss.IndexFlatL2(embedding_size=embedding_size)
return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})

Loading…
Cancel
Save