diff --git a/tests/unit/config/fixtures/minimal_config/settings.yaml b/tests/unit/config/fixtures/minimal_config/settings.yaml index 2fec50a3..82183a45 100644 --- a/tests/unit/config/fixtures/minimal_config/settings.yaml +++ b/tests/unit/config/fixtures/minimal_config/settings.yaml @@ -1,9 +1,11 @@ models: default_chat_model: api_key: ${CUSTOM_API_KEY} - type: openai_chat + type: chat + model_provider: openai model: gpt-4-turbo-preview default_embedding_model: api_key: ${CUSTOM_API_KEY} - type: openai_embedding + type: embedding + model_provider: openai model: text-embedding-3-small \ No newline at end of file diff --git a/tests/unit/config/fixtures/minimal_config_missing_env_var/settings.yaml b/tests/unit/config/fixtures/minimal_config_missing_env_var/settings.yaml index 9e1c4571..651b997a 100644 --- a/tests/unit/config/fixtures/minimal_config_missing_env_var/settings.yaml +++ b/tests/unit/config/fixtures/minimal_config_missing_env_var/settings.yaml @@ -1,9 +1,11 @@ models: default_chat_model: api_key: ${SOME_NON_EXISTENT_ENV_VAR} - type: openai_chat + type: chat + model_provider: openai model: gpt-4-turbo-preview default_embedding_model: api_key: ${SOME_NON_EXISTENT_ENV_VAR} - type: openai_embedding + type: embedding + model_provider: openai model: text-embedding-3-small \ No newline at end of file