From 46b80f029037d2d662bfb228e4babaf4f3866d92 Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 25 Sep 2023 14:01:38 -0400 Subject: [PATCH] clean up worker --- example2.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/example2.py b/example2.py index 49a57336..15bc48a2 100644 --- a/example2.py +++ b/example2.py @@ -8,6 +8,7 @@ class MultiAgentDebate: def run(self, task: str): results = [] + for i in range(len(self.agents)): # Select the speaker based on the selection function speaker_idx = self.selection_func(i, self.agents) @@ -25,10 +26,14 @@ def select_speaker(step: int, agents: List[Worker]) -> int: return step % len(agents) # Initialize agents +worker1 = Worker(openai_api_key="", ai_name="Optimus Prime") +worker2 = Worker(openai_api_key="", ai_name="Bumblebee") +worker3 = Worker(openai_api_key="", ai_name="Megatron") + agents = [ - Worker(openai_api_key="", ai_name="Optimus Prime"), - Worker(openai_api_key="", ai_name="Bumblebee"), - Worker(openai_api_key="", ai_name="Megatron") + worker1, + worker2, + worker3 ] # Initialize multi-agent debate with the selection function