From b3c8ae76bd75d5a5b9256f5e0aaff7077971cc40 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 14:36:02 -0400 Subject: [PATCH] clean up --- swarms/orchestrate.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/swarms/orchestrate.py b/swarms/orchestrate.py index 9d483507..d5f53fcc 100644 --- a/swarms/orchestrate.py +++ b/swarms/orchestrate.py @@ -44,8 +44,11 @@ import celery from typing import List, Dict, Any import numpy as np +from swarms.agents. + class Orchestrator(ABC): - def __init__(self, agent_list: List[Any], task_queue: celery.Celery, vector_db: np.ndarray): + def __init__(self, agent, agent_list: List[Any], task_queue: celery.Celery, vector_db: np.ndarray): + self.agent = agent self.agents = agent_list self.task_queue = task_queue self.vector_db = vector_db