From b144604f156ed48fba76562f732805d37025ed4b 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: b717b3c6dcf7f247bb6d793182974cc637c2c22d --- 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