From 201107429bc289fb0937830ba6d333dde5614ba2 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 12 Jul 2023 18:37:20 -0400
Subject: [PATCH] clean up codebase

Former-commit-id: 19c40279c009afea24f6dd2ece034e9175fed227
---
 swarms/agents/boss/{boss_agent.py => BossNode.py}            | 0
 swarms/agents/workers/{omni_worker.py => OmniWorkerAgent.py} | 0
 swarms/swarms.py                                             | 2 +-
 tests/LLM.py                                                 | 2 +-
 tests/swarms.py                                              | 2 +-
 5 files changed, 3 insertions(+), 3 deletions(-)
 rename swarms/agents/boss/{boss_agent.py => BossNode.py} (100%)
 rename swarms/agents/workers/{omni_worker.py => OmniWorkerAgent.py} (100%)

diff --git a/swarms/agents/boss/boss_agent.py b/swarms/agents/boss/BossNode.py
similarity index 100%
rename from swarms/agents/boss/boss_agent.py
rename to swarms/agents/boss/BossNode.py
diff --git a/swarms/agents/workers/omni_worker.py b/swarms/agents/workers/OmniWorkerAgent.py
similarity index 100%
rename from swarms/agents/workers/omni_worker.py
rename to swarms/agents/workers/OmniWorkerAgent.py
diff --git a/swarms/swarms.py b/swarms/swarms.py
index fce876af..6b51a99c 100644
--- a/swarms/swarms.py
+++ b/swarms/swarms.py
@@ -1,6 +1,6 @@
 from swarms.tools.agent_tools import *
 from swarms.agents.workers.WorkerNode import WorkerNode
-from swarms.agents.boss.boss_agent import BossNode
+from swarms.agents.boss.BossNode import BossNode
 # from swarms.agents.workers.omni_worker import OmniWorkerAgent
 # from swarms.tools.main import RequestsGet, ExitConversation
 # visual agent
diff --git a/tests/LLM.py b/tests/LLM.py
index 57360294..9f384ade 100644
--- a/tests/LLM.py
+++ b/tests/LLM.py
@@ -3,7 +3,7 @@ import os
 from unittest.mock import patch, MagicMock
 from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain
 
-from swarms.utils.llm import LLM
+from swarms.utils.LLM import LLM
 
 class TestLLM(unittest.TestCase):
     @patch.object(HuggingFaceHub, '__init__', return_value=None)
diff --git a/tests/swarms.py b/tests/swarms.py
index 524e8d0c..291aad08 100644
--- a/tests/swarms.py
+++ b/tests/swarms.py
@@ -1,7 +1,7 @@
 import unittest
 import swarms
 from swarms.agents.workers.WorkerNode import WorkerNode
-from swarms.agents.boss.boss_agent import BossNode
+from swarms.agents.boss.BossNode import BossNode
 
 class TestSwarms(unittest.TestCase):
     def setUp(self):