From 9dfe78a961d58031bf8b09386a79b1af538824cd Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 16:50:17 -0400 Subject: [PATCH] clean up Former-commit-id: 437f7595024092e09562969a7c11a235e059fa05 --- swarms/orchestrate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/swarms/orchestrate.py b/swarms/orchestrate.py index 912e0f39..294ac57b 100644 --- a/swarms/orchestrate.py +++ b/swarms/orchestrate.py @@ -51,7 +51,7 @@ class Orchestrator(ABC): def __init__(self, agent, agent_list: List[Any], - task_queue: celery.Celery, + task_queue: List[Any], vector_db: OceanDB ): self.agent = agent @@ -97,7 +97,7 @@ class Orchestrator(ABC): raise @abstractmethod - def update_vector_db(self, data: np.ndarray) -> None: + def update_vector_db(self, data) -> None: """Update the vector database""" try: self.vector_db.add_documents([data['vector']], [str(data['task_id'])]) @@ -107,7 +107,7 @@ class Orchestrator(ABC): @abstractmethod - def get_vector_db(self) -> np.ndarray: + def get_vector_db(self): """Retrieve the vector database""" return self.vector_db