From eecece615c28881f389f316ade2a6fee55bca2ca Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 29 Jul 2023 17:39:25 -0400 Subject: [PATCH] clean up Former-commit-id: e20903522eb11c52f6ec078ee1258147a78b9ff9 --- swarms/workers/worker_node.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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,