Update models.py

models.py

@@ -2,6 +2,16 @@ from pydantic import BaseModel
 from typing import List, Dict, Optional
 from enum import Enum
 
+
+class NotificationType(str, Enum):
+    RISK_ALERT = "risk_alert"
+    SYSTEM = "system"
+    MESSAGE = "message"
+
+class NotificationStatus(str, Enum):
+    UNREAD = "unread"
+    READ = "read"
+    ARCHIVED = "archived"
 class ChatRequest(BaseModel):
     message: str
     temperature: float = 0.7