Fix env variable name

Odrec · Sep 9, 2024 · 2154993 · 2154993
1 parent 8c851de
commit 2154993
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'])