from .flex2 import Flex2 AI_TOOLKIT_MODELS = [ # put a list of models here Flex2 ]