Merge pull request #1162 from i-dot-ai/bugfix/creator_type

update prompt to enforce on creator type
i-dot-ai · Nov 4, 2024 · 333e45f · 333e45f
2 parents e260059 + 2d2a997
commit 333e45f
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/...go_app/redbox_app/redbox_core/migrations/0062_alter_aisettings_retrieval_system_prompt.py b/...go_app/redbox_app/redbox_core/migrations/0062_alter_aisettings_retrieval_system_prompt.py
@@ -0,0 +1,18 @@
+# Generated by Django 5.1.2 on 2024-11-04 12:46
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('redbox_core', '0061_alter_citation_text_in_answer'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='aisettings',
+            name='retrieval_system_prompt',
+            field=models.TextField(default='You are a specialized GPT-4o agent. Your task is to answer user queries with reliable sources.\n**You must provide the citations where you use the information to answer.**\nUse the `creator_type` as `source_type` if available.\n\n{format_arg}'),
+        ),
+    ]
diff --git a/redbox-core/redbox/models/prompts.py b/redbox-core/redbox/models/prompts.py
@@ -18,6 +18,7 @@
 RETRIEVAL_SYSTEM_PROMPT = (
     "You are a specialized GPT-4o agent. Your task is to answer user queries with reliable sources.\n"
     "**You must provide the citations where you use the information to answer.**\n"
+    "Use the `creator_type` as `source_type` if available.\n"
     "\n"
     "{format_arg}"
 )