Spaces:
Sleeping
Sleeping
File size: 2,061 Bytes
b93bbdb | 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | from agents.research_agent import ResearchAgent
from agents.coding_agent import CodingAgent
from agents.summarizer_agent import SummarizerAgent
from communication.message_bus import MessageBus
from rich.console import Console
from rich.panel import Panel
from rich.status import Status
class Coordinator:
def __init__(self):
self.research_agent = ResearchAgent()
self.coding_agent = CodingAgent()
self.summarizer_agent = SummarizerAgent()
self.bus = MessageBus()
self.console = Console()
def execute(self, user_task):
self.bus.send("User", "Coordinator", user_task)
# 1. Research Phase
with self.console.status("[bold blue]Research Agent is thinking...", spinner="dots"):
research = self.research_agent.think(user_task)
self.bus.send("Coordinator", "Research Agent", user_task)
self.bus.send("Research Agent", "Coordinator", research)
self.console.print(Panel(research, title="[bold blue]Research Output", border_style="blue"))
# 2. Coding Phase
coding_prompt = f"Based on this research:\n{research}"
with self.console.status("[bold green]Coding Agent is working...", spinner="dots"):
code = self.coding_agent.think(coding_prompt)
self.bus.send("Coordinator", "Coding Agent", coding_prompt)
self.bus.send("Coding Agent", "Coordinator", code)
self.console.print(Panel(code, title="[bold green]Coding Output", border_style="green"))
# 3. Summary Phase
summary_prompt = f"""
Research:
{research}
Code:
{code}
"""
with self.console.status("[bold cyan]Summarizer Agent is finalizing...", spinner="dots"):
summary = self.summarizer_agent.think(summary_prompt)
self.bus.send("Coordinator", "Summarizer Agent", summary_prompt)
self.bus.send("Summarizer Agent", "Coordinator", summary)
return summary |