add chat

submission/models.py

@@ -8,6 +8,7 @@ from django.dispatch import receiver  # 导入receiver
 
 from account.models import Profile, RoleChoices, User
 from task.models import Task
+from prompt.models import Conversation
 
 
 class Submission(TimeStampedModel):
@@ -26,6 +27,10 @@ class Submission(TimeStampedModel):
     html = models.TextField(null=True, blank=True, verbose_name="HTML代码")
     css = models.TextField(null=True, blank=True, verbose_name="CSS代码")
     js = models.TextField(null=True, blank=True, verbose_name="JS代码")
+    conversation = models.ForeignKey(
+        Conversation, on_delete=models.SET_NULL, null=True, blank=True,
+        related_name="submissions", verbose_name="对话"
+    )
 
     class Meta:
         ordering = ("-created",)