clean up more modularization

Former-commit-id: 972c307e55
pull/47/head
Kye 1 year ago
parent 8dac197def
commit 0d835708c5

@ -9,7 +9,7 @@ from fastapi_cache.backends.redis import RedisBackend
from aioredis import Redis
from pydantic import BaseModel
from swarms.swarms import swarm
from swarms.swarms.swarms import swarm
from fastapi_limiter import FastAPILimiter
from fastapi_limiter.depends import RateLimiter

@ -1,5 +1,5 @@
import os
from swarms.swarms import HierarchicalSwarm
from swarms.swarms.swarms import HierarchicalSwarm
api_key = os.getenv("OPENAI_API_KEY")

@ -1,5 +1,5 @@
import os
from swarms.swarms import WorkerUltra
from swarms.swarms.swarms import WorkerUltra
api_key = os.getenv("OPENAI_API_KEY")

@ -1,4 +1,4 @@
from swarms.swarms import HierarchicalSwarm, swarm
from swarms.swarms.swarms import HierarchicalSwarm, swarm
from swarms.workers.worker_node import WorkerNodeInitializer, WorkerNode, worker_node
from swarms.boss.boss_node import BossNodeInitializer, BossNode

@ -6,7 +6,7 @@ import logging
#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
from swarms.swarms.swarms import HierarchicalSwarm
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')

@ -1,6 +1,6 @@
import pytest
from unittest.mock import Mock
from swarms.orchestrate import Orchestrator
from swarms.orchestrator.orchestrate import Orchestrator
@pytest.fixture

@ -1,7 +1,7 @@
import pytest
import logging
from unittest.mock import patch
from swarms.swarms import HierarchicalSwarm # replace with your actual module name
from swarms.swarms.swarms import HierarchicalSwarm # replace with your actual module name
@pytest.fixture
def swarm():

Loading…
Cancel
Save