from telemetry import log_event
from memory_core import update_memory
from context_memory import ContextMemoryGraph

class Orchestrator:
    def __init__(self):
        self.graph = ContextMemoryGraph()

    def route(self, agent, prompt):
        log_event(agent.agent_id, "route_start", "in_progress", {"prompt": prompt})
        context = self.graph.get_context(agent.agent_id)
        if "last_response" in context:
            prompt = f"{context['last_response']} -> {prompt}"

        response = agent.run(prompt)
        self.graph.link_context(agent.agent_id, "last_response", response)
        log_event(agent.agent_id, "route_complete", "success", {"response": response})
        return response