From 1774fe2592ddcb2fe82cbece41f2ab6ab7daabda Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 5 Jul 2023 18:00:43 -0400 Subject: [PATCH] workernode --- swarms/agents/workers/worker_agent.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/swarms/agents/workers/worker_agent.py b/swarms/agents/workers/worker_agent.py index 62166134..64f38886 100644 --- a/swarms/agents/workers/worker_agent.py +++ b/swarms/agents/workers/worker_agent.py @@ -7,10 +7,15 @@ from langchain.callbacks.manager import ( ) # ---------- Worker Node ---------- +from pydantic import Field + 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__()