bug fix

i-dot-ai · Jan 27, 2025 · f1ad023 · f1ad023
1 parent 234c148
commit f1ad023
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/django_app/redbox_app/redbox_core/migrations/0081_remove_user_ai_settings_and_more.py b/django_app/redbox_app/redbox_core/migrations/0081_remove_user_ai_settings_and_more.py
@@ -16,12 +16,9 @@
 def back_populate_context_window_size(apps, schema_editor):
     ChatLLMBackend = apps.get_model("redbox_core", "ChatLLMBackend")
     for llm in ChatLLMBackend.objects.all():
-        llm.context_window_size = context_window_sizes.get(llm.name)
+        llm.context_window_size = context_window_sizes.get(llm.name, 0)
         llm.save()
 
-    ChatLLMBackend.objects.filter(context_window_size__isnull=True).delete()
-
-
 class Migration(migrations.Migration):
 
     dependencies = [