diff --git a/swarms/agents/workers/worker.py b/swarms/agents/workers/worker.py index 83d0f26d..9748a96e 100644 --- a/swarms/agents/workers/worker.py +++ b/swarms/agents/workers/worker.py @@ -5,6 +5,7 @@ from langchain.callbacks.manager import ( AsyncCallbackManagerForToolRun, CallbackManagerForToolRun, ) +from typing import List, Any # from langchain.memory.chat_message_histories import FileChatMessageHistory @@ -13,9 +14,14 @@ from langchain.memory.chat_message_histories import FileChatMessageHistory class WorkerNode(BaseTool): name = "WorkerNode" description = "A worker node that can perform complex tasks" - llm: ChatOpenAI # add this line - tools: List[Tool] - vectorstore: VectorStore + # llm: ChatOpenAI # add this line + # tools: List[Tool] + # vectorstore: VectorStore + llm: Any + tools: List[Any] + vectorstore: Any + agent: Any = None + def __init__(self, llm, tools, vectorstore): super().__init__()