embedding size

1 year ago · 92cf48c76d
parent 3111cf0821
commit 92cf48c76d
1 changed files with 1 additions and 2 deletions
--- 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}")