clean up codebase

Former-commit-id: 19c40279c009afea24f6dd2ece034e9175fed227
pull/160/head
Kye 2 years ago
parent 3f639104db
commit 201107429b

@ -1,6 +1,6 @@
from swarms.tools.agent_tools import * from swarms.tools.agent_tools import *
from swarms.agents.workers.WorkerNode import WorkerNode 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.agents.workers.omni_worker import OmniWorkerAgent
# from swarms.tools.main import RequestsGet, ExitConversation # from swarms.tools.main import RequestsGet, ExitConversation
# visual agent # visual agent

@ -3,7 +3,7 @@ import os
from unittest.mock import patch, MagicMock from unittest.mock import patch, MagicMock
from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain
from swarms.utils.llm import LLM from swarms.utils.LLM import LLM
class TestLLM(unittest.TestCase): class TestLLM(unittest.TestCase):
@patch.object(HuggingFaceHub, '__init__', return_value=None) @patch.object(HuggingFaceHub, '__init__', return_value=None)

@ -1,7 +1,7 @@
import unittest import unittest
import swarms import swarms
from swarms.agents.workers.WorkerNode import WorkerNode 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): class TestSwarms(unittest.TestCase):
def setUp(self): def setUp(self):

Loading…
Cancel
Save