pull/53/head
Kye 1 year ago
parent 79bf66349f
commit 99f60ac9be

@ -7,7 +7,12 @@ from pegasus import Pegasus
class PegasusEmbedding:
def __init__(self, modality: str, multi_process: bool = False, n_processes: int = 4):
def __init__(
self,
modality: str,
multi_process: bool = False,
n_processes: int = 4
):
self.modality = modality
self.multi_process = multi_process
self.n_processes = n_processes

@ -13,7 +13,12 @@ from swarms.swarms.swarms import HierarchicalSwarm
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
class HiveMind:
def __init__(self, openai_api_key="", num_swarms=1, max_workers=None):
def __init__(
self,
openai_api_key="",
num_swarms=1,
max_workers=None
):
self.openai_api_key = openai_api_key
self.num_swarms = num_swarms
self.swarms = [HierarchicalSwarm(openai_api_key) for _ in range(num_swarms)]

Loading…
Cancel
Save