From 3cf1707875801c68ccee10af0f530a6a09a01ecc Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 5 Jul 2023 23:22:38 -0400
Subject: [PATCH] typing

---
 swarms/agents/workers/worker.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

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