ai-toolkit restructured - for realz this time