Implement new AI

Implement new AI models as the API becomes available