clean up imports

2 years ago · 9ef6d56c63
parent 3272b7b5c3
commit 9ef6d56c63
1 changed files with 1 additions and 0 deletions
--- a/unittesting/agents/swarms.py
+++ b/unittesting/agents/swarms.py
@ -1,5 +1,6 @@
 import unittest
 from unittest.mock import patch
+from swarms.agents.swarms import WorkerNode, BossNode, llm, tools, vectorstore, task_execution_chain

 class TestSwarms(unittest.TestCase):