typing

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__()