clean up for boss node

NewTools
Kye 2 years ago
parent 54cfdb2f05
commit 6b4845ab74

@ -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}")
print(f"Unexpected Error while initializing BabyAGI: {e}")

Loading…
Cancel
Save