Coverage for mindsdb / integrations / handlers / openai_handler / constants.py: 100%

1OPENAI_API_BASE = "https://api.openai.com/v1" 

2 

3CHAT_MODELS_PREFIXES = ("gpt-3.5", "gpt-3.5", "gpt-3.5", "gpt-4", "o3-mini", "o1-mini") 

4COMPLETION_MODELS = ("babbage-002", "davinci-002") 

5FINETUNING_MODELS = ("gpt-3.5-turbo", "babbage-002", "davinci-002", "gpt-4") 

6COMPLETION_LEGACY_BASE_MODELS = ("davinci", "curie", "babbage", "ada") 

7DEFAULT_CHAT_MODEL = "gpt-4o-mini" 

8 

9FINETUNING_LEGACY_MODELS = FINETUNING_MODELS 

10COMPLETION_LEGACY_MODELS = ( 

11 COMPLETION_LEGACY_BASE_MODELS 

12 + tuple(f"text-{model}-001" for model in COMPLETION_LEGACY_BASE_MODELS) 

13 + ("text-davinci-002", "text-davinci-003") 

14) 

15 

16DEFAULT_EMBEDDING_MODEL = "text-embedding-ada-002" 

17 

18IMAGE_MODELS = ("dall-e-2", "dall-e-3") 

19DEFAULT_IMAGE_MODEL = "dall-e-2"