Remove the parent_message field in Message model

chat/migrations/0005_remove_message_parent_message.py

@@ -0,0 +1,17 @@
+# Generated by Django 4.1.7 on 2023-03-22 08:11
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('chat', '0004_prompt'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='message',
+            name='parent_message',
+        ),
+    ]

chat/models.py

@@ -10,7 +10,6 @@ class Conversation(models.Model):
 
 class Message(models.Model):
     conversation = models.ForeignKey(Conversation, on_delete=models.CASCADE)
-    parent_message = models.ForeignKey('self', on_delete=models.CASCADE, null=True, blank=True, related_name='replies')
     message = models.TextField()
     is_bot = models.BooleanField(default=False)
     created_at = models.DateTimeField(auto_now_add=True)

chat/serializers.py

@@ -9,7 +9,7 @@ class Meta:
 class MessageSerializer(serializers.ModelSerializer):
     class Meta:
         model = Message
-        fields = ['id', 'parent_message', 'message', 'is_bot', 'created_at']
+        fields = ['id', 'message', 'is_bot', 'created_at']
 
 
 class PromptSerializer(serializers.ModelSerializer):

chat/views.py

@@ -129,7 +129,6 @@ def conversation(request):
     model = get_current_model()
     message = request.data.get('message')
     conversation_id = request.data.get('conversationId')
-    # parent_message_id = request.data.get('parentMessageId')
     max_tokens = request.data.get('max_tokens', model['max_response_tokens'])
     temperature = request.data.get('temperature', 0.7)
     top_p = request.data.get('top_p', 1)
@@ -146,7 +145,6 @@ def conversation(request):
     # insert a new message
     message_obj = Message(
         conversation_id=conversation_obj.id,
-        # parent_message_id=parent_message_id,
         message=message
     )
     message_obj.save()
@@ -200,7 +198,6 @@ def stream_content():
 
         ai_message_obj = Message(
             conversation_id=conversation_obj.id,
-            # parent_message_id=message_obj.id,
             message=completion_text,
             is_bot=True
         )