diff --git a/azureai/azure_ai.py b/azureai/azure_ai.py index 959a890..242ee69 100644 --- a/azureai/azure_ai.py +++ b/azureai/azure_ai.py @@ -25,7 +25,7 @@ def get_model_client( config = statsig.get_config(get_statsig_user(), config_name) endpoint = config.get_typed("endpoint", default_endpoint) api_key = config.get_typed("key", default_key) - completion_defaults = config.get_typed("api_version", {}) + completion_defaults = config.get_typed("completion_defaults", {}) return ModelClient( endpoint=endpoint,