model/__init__.py

@@ -2,5 +2,6 @@ from .robot import Robot
 from .product import Product
 from .inventory import InventoryRecord
 from .user import User
+from .ai_prediction import AIPrediction
 
-__all__ = ['Robot', 'Product', 'InventoryRecord', 'User']
+__all__ = ['Robot', 'Product', 'InventoryRecord', 'User', 'AIPrediction']

model/ai_prediction.py

@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+from datetime import datetime, date
+
+@dataclass
+class AIPrediction:
+    id: int
+    product_id: str
+    prediction_date: date
+    days_until_stockout: int
+    recommended_order: int
+    confidence_score: float
+    created_at: datetime