Move rerank boolean to guru type (#310)

Author: aralyekta

src/gurubase-backend/backend/core/migrations/0083_remove_settings_rerank_gurutype_rerank.py

@@ -0,0 +1,22 @@
+# Generated by Django 4.2.18 on 2025-04-29 07:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0082_gurutype_private'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='settings',
+            name='rerank',
+        ),
+        migrations.AddField(
+            model_name='gurutype',
+            name='rerank',
+            field=models.BooleanField(default=True),
+        ),
+    ]

src/gurubase-backend/backend/core/models.py

@@ -375,6 +375,8 @@ def get_language_code(self):
     date_created = models.DateTimeField(auto_now_add=True)
     date_updated = models.DateTimeField(auto_now=True)
 
+    rerank = models.BooleanField(default=True)
+
     def __str__(self):
         return self.slug
 
@@ -1296,8 +1298,6 @@ class AIProvider(models.TextChoices):
     split_overlap = models.IntegerField(default=300)
     split_min_length = models.IntegerField(default=500)
 
-    rerank = models.BooleanField(default=True)
-
     @classmethod
     def get_default_embedding_model(cls):
         """

src/gurubase-backend/backend/core/utils.py

@@ -565,7 +565,7 @@ def fetch_and_merge_answers(question_id):
 
     def rerank_batch(batch, question, user_question, enhanced_question, llm_eval):
         default_settings = get_default_settings()
-        if settings.ENV == 'selfhosted' or not default_settings.rerank:
+        if settings.ENV == 'selfhosted' or not guru_type.rerank:
             # Do not rerank in selfhosted
             return [i for i in range(len(batch))], [1 for _ in range(len(batch))]
         batch_texts = [result['entity']['text'] for result in batch]