From 8657697c25431e378c4a46439dd2b0f84a380994 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Wed, 15 Apr 2026 22:25:30 -0600
Subject: [PATCH] fix

---
 prompt/models.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/prompt/models.py b/prompt/models.py
index ccce241..c817e23 100644
--- a/prompt/models.py
+++ b/prompt/models.py
@@ -1,5 +1,7 @@
 import uuid
 from django.db import models
+from django.db.models.signals import pre_delete
+from django.dispatch import receiver
 from django_extensions.db.models import TimeStampedModel
 from account.models import User
 from task.models import Task
@@ -45,3 +47,10 @@ class Message(models.Model):
 
     def __str__(self):
         return f"[{self.role}] {self.content[:50]}"
+
+
+@receiver(pre_delete, sender=Message)
+def delete_linked_submission(sender, instance, **kwargs):
+    if instance.submission_id:
+        from submission.models import Submission
+        Submission.objects.filter(id=instance.submission_id).delete()