clena up

2 years ago · a89cb9ef95
parent 1b70936868
commit a89cb9ef95
1 changed files with 2 additions and 1 deletions
--- 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({}), {})