diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py index edceade8..b87b0a49 100644 --- a/swarms/workers/worker_node.py +++ b/swarms/workers/worker_node.py @@ -34,7 +34,7 @@ class WorkerNodeInitializer: llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, tools: Optional[List[Tool]] = None, # vectorstore: Optional[FAISS] = None, - embedding_size: Optional[int] = 1926, + # embedding_size: Optional[int] = 1926, worker_name: Optional[str] = "Swarm Worker AI Assistant", worker_role: Optional[str] = "Assistant", human_in_the_loop: Optional[bool] = False, @@ -92,7 +92,7 @@ class WorkerNodeInitializer: def initialize_vectorstore(self): try: embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key) - embedding_size = self.embedding_size + embedding_size = 8192 index = faiss.IndexFlatL2(embedding_size=embedding_size) return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) @@ -123,7 +123,7 @@ class WorkerNode: llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, tools: Optional[List[Tool]] = None, # vectorstore: Optional[FAISS] = None, - embedding_size: Optional[int] = 4026, + # embedding_size: Optional[int] = 4026, worker_name: Optional[str] = "Swarm Worker AI Assistant", worker_role: Optional[str] = "Assistant", human_in_the_loop: Optional[bool] = False,