Update parenttext_models.py

IDEMSInternational · Apr 5, 2024 · 68d766b · 68d766b
1 parent d3077bb
commit 68d766b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/parenttext_pipeline/models/parenttext_models.py b/src/parenttext_pipeline/models/parenttext_models.py
@@ -5,10 +5,14 @@
 
 ###########################################
 #general
-class VariableModel(DataRowModel):
+class VariableModel(ParserModel):
 	name: str = ''
 	value: str = '' 
 
+class AttachmentModel(ParserModel):
+	file_type: str = ''
+	url: str = ''
+
 ###################################################################
 class IntroductionBlockModel(ParserModel):
 	msg_list: List[str] = []
@@ -426,6 +430,7 @@ class MetadataModel(ParserModel):
 class SingleMessageModel(DataRowModel):
 	msg: str = ''
 	next_button_option: str = ''
+	attachment: AttachmentModel = AttachmentModel()
 	metadata: MetadataModel = MetadataModel()
 
 ####################################