4949ENV_AZURE_OPENAI_EMBEDDINGS_NAME = "text-embedding-ada-002"
5050ENV_AZURE_OPENAI_AUDIO_NAME = "whisper"
5151ENV_AZURE_OPENAI_DALLE_NAME = "dall-e-3"
52- ENV_AZURE_OPENAI_CHAT_COMPLETIONS_GPT4_NAME = "gpt-4o"
52+ ENV_AZURE_OPENAI_CHAT_COMPLETIONS_GPT4_NAME = "gpt-4o-mini"
53+ ENV_AZURE_OPENAI_REALTIME_NAME = "gpt-4o-mini-realtime-preview-1217"
5354ENV_AZURE_OPENAI_TTS_NAME = "tts"
5455
5556ENV_OPENAI_KEY = "OPENAI_KEY"
5859ENV_OPENAI_EMBEDDINGS_MODEL = "text-embedding-ada-002"
5960ENV_OPENAI_AUDIO_MODEL = "whisper-1"
6061ENV_OPENAI_DALLE_MODEL = "dall-e-3"
61- ENV_OPENAI_CHAT_COMPLETIONS_GPT4_MODEL = "gpt-4o-2024-08-06"
62+ ENV_OPENAI_CHAT_COMPLETIONS_GPT4_MODEL = "gpt-4o-mini-2024-07-18"
63+ ENV_OPENAI_REALTIME_NAME = "gpt-4o-mini-realtime-preview-2024-10-01"
6264ENV_OPENAI_TTS_MODEL = "tts-1"
6365
6466
@@ -149,8 +151,10 @@ def build_kwargs(args, api_type):
149151 return {"model" : ENV_OPENAI_TTS_MODEL }
150152 elif api_type in ["tts_azure" ]:
151153 return {"model" : ENV_AZURE_OPENAI_TTS_NAME }
152- if test_feature .startswith ("test_chat_completions" ) \
153- or test_feature .startswith (("test_client" , "test_models" )):
154+ if test_feature .startswith (
155+ ("test_client" , "test_models" , "test_chat_completions" ,
156+ "test_assistants" , "test_responses" , "test_vector_stores" )
157+ ):
154158 if api_type in ["azure" , "azure_key" , "asst_azure" ]:
155159 return {"model" : ENV_AZURE_OPENAI_CHAT_COMPLETIONS_NAME }
156160 elif api_type == "openai" :
@@ -174,16 +178,11 @@ def build_kwargs(args, api_type):
174178 return {"model" : ENV_AZURE_OPENAI_DALLE_NAME }
175179 elif api_type == "openai" :
176180 return {"model" : ENV_OPENAI_DALLE_MODEL }
177- if test_feature .startswith ("test_assistants" ) or test_feature .startswith ("test_vector_stores" ):
178- if api_type in ["asst_azure" ]:
179- return {"model" : ENV_AZURE_OPENAI_CHAT_COMPLETIONS_GPT4_NAME }
180- elif api_type == "gpt_4_openai" :
181- return {"model" : ENV_OPENAI_CHAT_COMPLETIONS_GPT4_MODEL }
182181 if test_feature .startswith ("test_realtime" ):
183182 if api_type in ["gpt_4_azure" ]:
184- return {"model" : "gpt-4o-realtime-preview-1001" }
183+ return {"model" : ENV_AZURE_OPENAI_REALTIME_NAME }
185184 elif api_type == "gpt_4_openai" :
186- return {"model" : "gpt-4o-realtime-preview-2024-10-01" }
185+ return {"model" : ENV_OPENAI_REALTIME_NAME }
187186 if test_feature .startswith (("test_module_client" , "test_cli" )):
188187 return {}
189188 raise ValueError (f"Test feature: { test_feature } needs to have its kwargs configured." )
0 commit comments