From 0966bce54f54c6d500f0266116701fcc7634ca0d Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 5 Jul 2023 19:15:54 -0400 Subject: [PATCH] clean up --- swarms/agents/workers/worker_agent.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/swarms/agents/workers/worker_agent.py b/swarms/agents/workers/worker_agent.py index 206dca42..01f524e4 100644 --- a/swarms/agents/workers/worker_agent.py +++ b/swarms/agents/workers/worker_agent.py @@ -10,12 +10,12 @@ class WorkerNode(BaseTool): """Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on """ name = "WorkerNode" description = "A worker node that can perform complex tasks" - llm: Any = Field(description="Language model") - tools: List[Tool] = Field(description="List of tools") - vectorstore: VectorStore = Field(description="Vector store") - - def __init__(self, **data): - super().__init__(**data) + + def __init__(self, llm, tools, vectorstore): + super().__init__() + self.llm = llm + self.tools = tools + self.vectorstore = vectorstore def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs): self.agent = AutoGPT.from_llm_and_tools(