diff --git a/chain_service/database/models/chain.py b/chain_service/database/models/chain.py
index 91c967c..ce8abc2 100644
--- a/chain_service/database/models/chain.py
+++ b/chain_service/database/models/chain.py
@@ -19,7 +19,8 @@ class WaitAction(BaseAction):
 
 class CommentAction(BaseAction):
     action_type: Literal["comment"]
-    text: str
+    text: Annotated[Optional[str], Field(default=None)]
+    files: Annotated[Optional[List[str]], Field(default=[])]
 
     class Config(BaseConfig):
         pass
diff --git a/chain_service/database/models/progress_chain.py b/chain_service/database/models/progress_chain.py
index 14f9925..3a99939 100644
--- a/chain_service/database/models/progress_chain.py
+++ b/chain_service/database/models/progress_chain.py
@@ -35,7 +35,8 @@ class WaitProgressAction(BaseProgressAction):
 
 class CommentProgressAction(BaseProgressAction):
     action_type: Literal["comment"]
-    text: str
+    text: Annotated[Optional[str], Field(default=None)]
+    files: Annotated[Optional[List[str]], Field(default=[])]
 
     class Config(BaseConfig):
         pass