From 7d503bebdde638c27e5f4b7290255aa4a9edb927 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 10 Aug 2023 11:01:08 -0400 Subject: [PATCH] embedding size Former-commit-id: be9841f316b8267273360f33ab7a0bc667606233 --- swarms/workers/vortex_worker.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/swarms/workers/vortex_worker.py b/swarms/workers/vortex_worker.py index a225781f..2c64243b 100644 --- a/swarms/workers/vortex_worker.py +++ b/swarms/workers/vortex_worker.py @@ -87,8 +87,7 @@ class VortexWorkerAgent: try: 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(s) + index = faiss.IndexFlatL2(self.embedding_size) return FAISS(embeddings_model, index, InMemoryDocstore({}), {}) except Exception as error: logging.error(f"Failed to initialize vector store: {error}")