chat, add_agents

Former-commit-id: 01f59d073da6a1b61b3ca06371068fd1e512dbf6
pull/160/head
Kye 2 years ago
parent 44df59bfc0
commit 601de9d741

@ -245,3 +245,23 @@ class Orchestrator:
documents=[message],
ids=[f"{sender_id}_to_{receiver_id}"]
)
def add_agents(
self,
num_agents: int
):
for _ in range(num_agents):
self.agents.put(self.agent())
self.executor = ThreadPoolExecutor(
max_workers=self.agents.qsize()
)
def remove_agents(self, num_agents):
for _ in range(num_agents):
if not self.agents.empty():
self.agents.get()
self.executor = ThreadPoolExecutor(
max_workers=self.agents.qsize()
)
Loading…
Cancel
Save