|
|
@ -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({}), {})
|
|
|
|
|
|
|
|
|
|
|
|