19 lines
911 B
Python
19 lines
911 B
Python
from typing import TYPE_CHECKING
|
|
from fusionagi.agents.base_agent import BaseAgent
|
|
from fusionagi.schemas.messages import AgentMessageEnvelope
|
|
from fusionagi._logger import logger
|
|
if TYPE_CHECKING:
|
|
from fusionagi.core.orchestrator import Orchestrator
|
|
from fusionagi.core.goal_manager import GoalManager
|
|
|
|
class CoordinatorAgent(BaseAgent):
|
|
def __init__(self, identity="coordinator", orchestrator=None, goal_manager=None, planner_id="planner"):
|
|
super().__init__(identity=identity, role="Coordinator", objective="Own goals and assign tasks", memory_access=True, tool_permissions=[])
|
|
self._orch = orchestrator
|
|
self._goal_manager = goal_manager
|
|
self._planner_id = planner_id
|
|
def handle_message(self, envelope):
|
|
if envelope.message.intent == "goal_created" and self._orch and self._planner_id:
|
|
self._orch.route_message(envelope)
|
|
return None
|