from context_graph import ContextMemoryGraph

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

    def route(self, prompt, context):
        if "summarize" in prompt.lower():
            result = "This agent summarizes and organizes context."
        elif "memory" in prompt.lower():
            result = f"Memory size: {len(context)}"
        else:
            result = f"Processed: {prompt}"

        self.graph.link_context(self.agent_core.agent_id, prompt, result)
        return result