with open('autonomous_swarm.py', 'r') as f:
    content = f.read()

# Fix 1: CoordinatorAgent is called with wrong parameter
content = content.replace(
    'self.coordinator = CoordinatorAgent(self.agents)',
    'self.coordinator = CoordinatorAgent(self)'
)

# Fix 2: Update CoordinatorAgent __init__ signature
# It currently expects agents parameter, change to swarm
old_coord_init = '''class CoordinatorAgent(BaseAgent):
    def __init__(self, swarm=None):
        super().__init__("Coordinator", "coordinating the agent swarm")
        self.swarm = swarm
        self.agents = swarm'''

new_coord_init = '''class CoordinatorAgent(BaseAgent):
    def __init__(self, swarm=None):
        super().__init__("Coordinator", "coordinating the agent swarm")
        self.swarm = swarm
        self.agents = swarm.agents if swarm else []'''

content = content.replace(old_coord_init, new_coord_init)

with open('autonomous_swarm.py', 'w') as f:
    f.write(content)

print("✅ Fixed CoordinatorAgent")