From b14aee130babf96104c5a592a7aff2fca9934e6b Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 6 Jul 2023 14:23:06 -0400 Subject: [PATCH] clean up for boss node --- swarms/agents/boss/boss_agent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/swarms/agents/boss/boss_agent.py b/swarms/agents/boss/boss_agent.py index 1d7d7444..fd237d02 100644 --- a/swarms/agents/boss/boss_agent.py +++ b/swarms/agents/boss/boss_agent.py @@ -7,16 +7,16 @@ class BossNode: self.llm = llm self.vectorstore = vectorstore self.agent_executor = agent_executor - # self.max_iterations = max_iterations + self.max_iterations = max_iterations try: self.baby_agi = BabyAGI.from_llm( llm=self.llm, vectorstore=self.vectorstore, task_execution_chain=self.agent_executor, - # max_iterations=self.max_iterations, + max_iterations=self.max_iterations, ) except ValidationError as e: print(f"Validation Error while initializing BabyAGI: {e}") except Exception as e: - print(f"Unexpected Error while initializing BabyAGI: {e}") \ No newline at end of file + print(f"Unexpected Error while initializing BabyAGI: {e}")