def route_query(query: str):
    category = classify_query(query)

    if category == "direct":
        return direct_agent.query(query)
    elif category == "tool":
        return tool_agent.query(query)
    else category == "react":
        return react_agent.query(query)