diff --git a/modules/datamodels/datamodelChat.py b/modules/datamodels/datamodelChat.py
index 31e65004..198360bf 100644
--- a/modules/datamodels/datamodelChat.py
+++ b/modules/datamodels/datamodelChat.py
@@ -1,6 +1,7 @@
 """Chat models: ChatWorkflow, ChatMessage, ChatLog, ChatStat, ChatDocument."""
 
 from typing import List, Dict, Any, Optional
+from enum import Enum
 from pydantic import BaseModel, Field
 from modules.shared.attributeUtils import register_model_labels, ModelMixin
 from modules.shared.timezoneUtils import get_utc_timestamp
@@ -574,7 +575,7 @@ register_model_labels(
 )
 
 
-class TaskStatus(str):
+class TaskStatus(str, Enum):
     PENDING = "pending"
     RUNNING = "running"
     COMPLETED = "completed"