just testing communicate method for the orchestrator

Former-commit-id: 5ace98e927
group-chat
Kye 1 year ago
parent 73d91a51f8
commit 9416d06495

@ -221,3 +221,24 @@ class Orchestrator(ABC):
logging.error(f"An error occured in swarm: {e}")
return None
def communicate(
self,
sender_id: int,
receiver_id: int,
message: str
):
"""Allows the agents to communicate with eachother thrught the vectordatabase"""
message_vector = self.embed(
message,
self.api_key,
self.model_name
)
#store the mesage in the vector database
self.collection.add(
embeddings=[message_vector],
documents=[message],
ids=[f"{sender_id}_to_{receiver_id}"]
)

Loading…
Cancel
Save