Selection of model for image generation #10
Description
Not everyone has the required licenses for openai to leverage the codebase as is, so putting in a parameter to select the model based on end user preference would be useful - for example i leverage claude and grok more than chatgpt, and would like to add that support in for others.