clean up codebase

pull/11/head
Kye 2 years ago
parent 332c444483
commit eaf67c5259

@ -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