clean up

swarms/agents/utils/callback.py

@@ -2,7 +2,7 @@ from typing import Any, Dict, List, Optional, Union
 
 from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema import AgentAction, AgentFinish, LLMResult
-from celery import Task
+# from celery import Task
 
 # from ansi import ANSI, Color, Style, dim_multiline
 from swarms.utils.utils import ANSI, Color, Style, dim_multiline

swarms/agents/utils/manager.py

@@ -1,5 +1,5 @@
 from typing import Dict, Optional
-from celery import Task
+# from celery import Task
 
 from langchain.agents.agent import AgentExecutor
 from langchain.callbacks.base import CallbackManager

swarms/agents/workers/agents.py

@@ -222,7 +222,7 @@ from typing import Any, Dict, List, Optional, Union
 
 from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema import AgentAction, AgentFinish, LLMResult
-from celery import Task
+# from celery import Task
 
 from swarms.utils.utils import ANSI, Color, Style, dim_multiline, logger