From 20524a5503f51b58839292605d85c66d2cdec0bd Mon Sep 17 00:00:00 2001
From: Yansong Zhang <916125788@qq.com>
Date: Mon, 20 Apr 2026 14:36:31 +0800
Subject: [PATCH] fix mode

---
 api/models/comment.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/api/models/comment.py b/api/models/comment.py
index a30a52ec86..4104861d10 100644
--- a/api/models/comment.py
+++ b/api/models/comment.py
@@ -154,6 +154,10 @@ class WorkflowCommentReply(Base):
 
     id: Mapped[str] = mapped_column(
         StringUUID, server_default=sa.text("uuidv7()"))
+    comment_id: Mapped[str] = mapped_column(
+        StringUUID, sa.ForeignKey("workflow_comments.id", ondelete="CASCADE"), nullable=False
+    )
+    content: Mapped[str] = mapped_column(sa.Text, nullable=False)
     created_by: Mapped[str] = mapped_column(StringUUID, nullable=False)
     created_at: Mapped[datetime] = mapped_column(
         sa.DateTime, nullable=False, server_default=func.current_timestamp())
@@ -201,6 +205,9 @@ class WorkflowCommentMention(Base):
     comment_id: Mapped[str] = mapped_column(
         StringUUID, sa.ForeignKey("workflow_comments.id", ondelete="CASCADE"), nullable=False
     )
+    reply_id: Mapped[str | None] = mapped_column(
+        StringUUID, sa.ForeignKey("workflow_comment_replies.id", ondelete="CASCADE"), nullable=True
+    )
     mentioned_user_id: Mapped[str] = mapped_column(StringUUID, nullable=False)
     # Relationships
     comment: Mapped["WorkflowComment"] = relationship(