49
49
ENV_AZURE_OPENAI_EMBEDDINGS_NAME = "text-embedding-ada-002"
50
50
ENV_AZURE_OPENAI_AUDIO_NAME = "whisper"
51
51
ENV_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"
53
54
ENV_AZURE_OPENAI_TTS_NAME = "tts"
54
55
55
56
ENV_OPENAI_KEY = "OPENAI_KEY"
58
59
ENV_OPENAI_EMBEDDINGS_MODEL = "text-embedding-ada-002"
59
60
ENV_OPENAI_AUDIO_MODEL = "whisper-1"
60
61
ENV_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"
62
64
ENV_OPENAI_TTS_MODEL = "tts-1"
63
65
64
66
@@ -149,8 +151,10 @@ def build_kwargs(args, api_type):
149
151
return {"model" : ENV_OPENAI_TTS_MODEL }
150
152
elif api_type in ["tts_azure" ]:
151
153
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
+ ):
154
158
if api_type in ["azure" , "azure_key" , "asst_azure" ]:
155
159
return {"model" : ENV_AZURE_OPENAI_CHAT_COMPLETIONS_NAME }
156
160
elif api_type == "openai" :
@@ -174,16 +178,11 @@ def build_kwargs(args, api_type):
174
178
return {"model" : ENV_AZURE_OPENAI_DALLE_NAME }
175
179
elif api_type == "openai" :
176
180
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 }
182
181
if test_feature .startswith ("test_realtime" ):
183
182
if api_type in ["gpt_4_azure" ]:
184
- return {"model" : "gpt-4o-realtime-preview-1001" }
183
+ return {"model" : ENV_AZURE_OPENAI_REALTIME_NAME }
185
184
elif api_type == "gpt_4_openai" :
186
- return {"model" : "gpt-4o-realtime-preview-2024-10-01" }
185
+ return {"model" : ENV_OPENAI_REALTIME_NAME }
187
186
if test_feature .startswith (("test_module_client" , "test_cli" )):
188
187
return {}
189
188
raise ValueError (f"Test feature: { test_feature } needs to have its kwargs configured." )
0 commit comments