From 0e383502c94c2823b9746d6297077595c3b233bb Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 29 Jul 2023 17:48:49 -0400 Subject: [PATCH] clean up --- swarms/boss/boss_node.py | 2 +- swarms/swarms.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/swarms/boss/boss_node.py b/swarms/boss/boss_node.py index c9dc98c1..1bf26831 100644 --- a/swarms/boss/boss_node.py +++ b/swarms/boss/boss_node.py @@ -161,7 +161,7 @@ class BossNode: 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) return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) except Exception as e: diff --git a/swarms/swarms.py b/swarms/swarms.py index 5c56306d..558b825d 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -28,7 +28,7 @@ class HierarchicalSwarm: self, openai_api_key: Optional[str] = "", use_vectorstore: Optional[bool] = True, - # embedding_size: Optional[int] = None, + use_async: Optional[bool] = True, worker_name: Optional[str] = "Swarm Worker AI Assistant", verbose: Optional[bool] = False, @@ -44,7 +44,7 @@ class HierarchicalSwarm: self.use_async = use_async self.worker_name = worker_name self.human_in_the_loop = human_in_the_loop - # self.embedding_size = embedding_size + self.boss_prompt = boss_prompt self.temperature = temperature self.max_iterations = max_iterations