diff --git a/swarms/agents/workers/worker_agent.py b/swarms/agents/workers/worker_agent.py index 3eb867dc..823af18d 100644 --- a/swarms/agents/workers/worker_agent.py +++ b/swarms/agents/workers/worker_agent.py @@ -8,12 +8,14 @@ from langchain.callbacks.manager import ( 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, llm, tools, vectorstore): - super().__init__() - self.llm = llm - self.tools = tools - self.vectorstore = vectorstore + def __init__(self, **data): + super().__init__(**data) def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs): self.agent = AutoGPT.from_llm_and_tools(