From a89cb9ef95d65f33a3fec1072f2aa1b8300cd9a8 Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 29 Jul 2023 12:28:53 -0400 Subject: [PATCH] clena up --- swarms/workers/worker_node.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py index 4854611d..e7584c73 100644 --- a/swarms/workers/worker_node.py +++ b/swarms/workers/worker_node.py @@ -91,8 +91,9 @@ class WorkerNodeInitializer: def initialize_vectorstore(self): try: - embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key) 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) return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})