From 8538821d0c08389ecdcdc153df41fa20809e1d90 Mon Sep 17 00:00:00 2001
From: Christopher Gondek <gondek.christopher@gmail.com>
Date: Thu, 9 Oct 2025 10:54:11 +0200
Subject: [PATCH] fix: pydantic schema issue

---
 modules/datamodels/datamodelChat.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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"