File size: 627 Bytes
3d92de3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from dataclasses import dataclass
from autogen_core import AgentId
import glob
import os
import random
@dataclass
class Message:
content: str
def find_recipient() -> AgentId:
try:
agent_files = glob.glob("agent*.py")
agent_names = [os.path.splitext(file)[0] for file in agent_files]
agent_names.remove("agent")
agent_name = random.choice(agent_names)
print(f"Selecting agent for refinement: {agent_name}")
return AgentId(agent_name, "default")
except Exception as e:
print(f"Exception finding recipient: {e}")
return AgentId("agent1", "default")
|