From f40145d03a0083ef9fd875e32515a9e3a8ee3938 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 9 Aug 2023 18:30:17 -0400 Subject: [PATCH] embedding size not defined Former-commit-id: 53ea599a68bf00676c280acd74a5d69c3f3a18fb --- swarms/workers/worker_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py index 3f6b8460..c03ce54c 100644 --- a/swarms/workers/worker_node.py +++ b/swarms/workers/worker_node.py @@ -31,7 +31,7 @@ class WorkerNodeInitializer: openai_api_key: str, llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, tools: Optional[List[Tool]] = None, - embedding_size: Optional[int] = 1926, + embedding_size: Optional[int] = 8192, worker_name: Optional[str] = "Swarm Worker AI Assistant", worker_role: Optional[str] = "Assistant", human_in_the_loop: Optional[bool] = False, @@ -45,7 +45,7 @@ class WorkerNodeInitializer: self.worker_name = worker_name self.worker_role = worker_role - self.embedding_size = embedding_size + self.embedding_size = 8192 self.human_in_the_loop = human_in_the_loop self.search_kwargs = search_kwargs self.verbose = verbose