Add the "open_frugal_mode_control" setting option.

WongSaang · Apr 17, 2023 · 0db602b · 0db602b
1 parent 15d5583
commit 0db602b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/chat/signals.py b/chat/signals.py
@@ -6,9 +6,9 @@
 def load_default_settings(sender, **kwargs):
     if sender.name == 'chat':
         print('Setting up default settings...')
-        if not Setting.objects.filter(name='openai_api_key').exists():
-            Setting.objects.create(name='openai_api_key')
-            print('Created setting: openai_api_key')
+        if not Setting.objects.filter(name='open_frugal_mode_control').exists():
+            Setting.objects.create(name='open_frugal_mode_control', value='True')
+            print('Created setting: open_frugal_mode_control')
         if not Setting.objects.filter(name='open_registration').exists():
             Setting.objects.create(name='open_registration', value='True')
             print('Created setting: open_registration')

diff --git a/chat/views.py b/chat/views.py
@@ -29,7 +29,8 @@ def get_queryset(self):
         available_names = [
             'open_registration',
             'open_web_search',
-            'open_api_key_setting'
+            'open_api_key_setting',
+            'open_frugal_mode_control',
         ]
         return Setting.objects.filter(name__in=available_names)