clean up initialization

Former-commit-id: 80039367dc
pull/47/head
Kye 2 years ago
parent 284e21f7ed
commit 1ef4026630

@ -2,6 +2,9 @@ import asyncio
import logging import logging
from typing import Optional from typing import Optional
from langchain import LLMChain, OpenAI, PromptTemplate
from swarms.agents.tools.main import WebpageQATool, process_csv from swarms.agents.tools.main import WebpageQATool, process_csv
from swarms.boss.boss_node import BossNode from swarms.boss.boss_node import BossNode
from swarms.workers.worker_node import WorkerNode from swarms.workers.worker_node import WorkerNode
@ -56,10 +59,13 @@ class HierarchicalSwarm:
) )
self.boss_node = BossNode( self.boss_node = BossNode(
objective="",
api_key=self.openai_api_key,
vectorstore=None,
worker_node=self.worker_node, worker_node=self.worker_node,
llm_class=OpenAI,
max_iterations=self.max_iterations, max_iterations=self.max_iterations,
human_in_the_loop=self.human_in_the_loop, verbose=self.verbose
embedding_size=self.embedding_size
) )
self.logger = logging.getLogger() self.logger = logging.getLogger()

Loading…
Cancel
Save