From fbb76e41f1055a1e109e990e3c33141f6c55e4a3 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 10 Aug 2023 11:00:17 -0400 Subject: [PATCH] embedding plug in faiss Former-commit-id: 0e7760c3fa78397455e8523e0358a1b563622336 --- swarms/workers/vortex_worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swarms/workers/vortex_worker.py b/swarms/workers/vortex_worker.py index d15967b4..a225781f 100644 --- a/swarms/workers/vortex_worker.py +++ b/swarms/workers/vortex_worker.py @@ -88,7 +88,7 @@ class VortexWorkerAgent: openai_api_key = self.openai_api_key or os.getenv("OPENAI_API_KEY") embeddings_model = OpenAIEmbeddings(openai_api_key=openai_api_key) embedding_size = 8192 - index = faiss.IndexFlatL2(embedding_size=embedding_size) + index = faiss.IndexFlatL2(s) return FAISS(embeddings_model, index, InMemoryDocstore({}), {}) except Exception as error: logging.error(f"Failed to initialize vector store: {error}")