diff --git a/pyproject.toml b/pyproject.toml index c826f9ce..d90ef974 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,7 @@ classifiers = [ ] [tool.poetry.dependencies] -python = "^3.6" +python = "^3.8.1" transformers = "*" openai = "*" langchain = "0.0.240" diff --git a/swarms/workers/worker.py b/swarms/workers/worker.py index 5cd01f94..ba658f9d 100644 --- a/swarms/workers/worker.py +++ b/swarms/workers/worker.py @@ -180,7 +180,7 @@ class Worker: """ try: embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key) - embedding_size = 1892 + embedding_size = 1536 index = faiss.IndexFlatL2(embedding_size) self.vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})