Or just regular gpt-4.1, it's a quite capable model.