You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.2 KiB
53 lines
1.2 KiB
from typing import List, Callable
|
|
|
|
|
|
# Define a selection function
|
|
def select_speaker(step: int, agents) -> int:
|
|
# This function selects the speaker in a round-robin fashion
|
|
return step % len(agents)
|
|
|
|
|
|
class MultiAgentDebate:
|
|
"""
|
|
MultiAgentDebate
|
|
|
|
Args:
|
|
|
|
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
agents,
|
|
selection_func,
|
|
):
|
|
self.agents = agents
|
|
self.selection_func = selection_func
|
|
|
|
# def reset_agents(self):
|
|
# for agent in self.agents:
|
|
# agent.reset()
|
|
|
|
def inject_agent(self, agent):
|
|
self.agents.append(agent)
|
|
|
|
def run(self, task: str, max_iters: int = None):
|
|
# self.reset_agents()
|
|
results = []
|
|
for i in range(max_iters or len(self.agents)):
|
|
speaker_idx = self.selection_func(i, self.agents)
|
|
speaker = self.agents[speaker_idx]
|
|
response = speaker(task)
|
|
results.append({"response": response})
|
|
return results
|
|
|
|
def update_task(self, task: str):
|
|
self.task = task
|
|
|
|
def format_results(self, results):
|
|
formatted_results = "\n".join(
|
|
[f"Agent responded: {result['response']}" for result in results]
|
|
)
|
|
|
|
return formatted_results
|