diff --git a/src/ai_client.py b/src/ai_client.py index a523755..699f546 100644 --- a/src/ai_client.py +++ b/src/ai_client.py @@ -57,7 +57,7 @@ def set_accessible_models(): Get accessible models for the current user """ # Load accessible user models from environment variables - default_models = json.loads(os.environ['OPENAI_DEFAULT_MODEL']) + default_models = json.loads(os.environ['OPENAI_MODELS']) default_extra_models = json.loads(os.environ['OPENAI_MODEL_EXTRA']) if 'accessible_models' not in session_state and {'USER_ROLES', 'MODELS_PER_ROLE'} <= os.environ.keys(): user_roles = json.loads(os.environ['USER_ROLES'])