clean up

swarms/hivemind.py

@@ -4,10 +4,9 @@
 import concurrent.futures
 import logging
 
-from swarms.swarms import HierarchicalSwarm
-
 #this needs to change, we need to specify exactly what needs to be imported
 from swarms.agents.tools.agent_tools import *
+from swarms.swarms import HierarchicalSwarm
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 

swarms/swarms.py

@@ -1,11 +1,11 @@
-import logging
 import asyncio
+import logging
+from typing import Optional
 
+from swarms.agents.models.hf import HuggingFaceLLM
 from swarms.agents.tools.agent_tools import *
-from swarms.workers.worker_node import WorkerNodeInitializer
 from swarms.boss.boss_node import BossNodeInitializer as BossNode
-
+from swarms.workers.worker_node import WorkerNodeInitializer
-from swarms.agents.models.hf import HuggingFaceLLM
 
 # from langchain import LLMMathChain
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

tests/agents/workers/omni_worker.py

@@ -1,5 +1,7 @@
 import pytest
-from swarms.worker.omni_worker import OmniWorkerAgent  # replace 'yourmodule' with the actual module name
+
+from swarms.worker.omni_worker import OmniWorkerAgent
+
 
 @pytest.fixture
 def omni_worker():