diff --git a/model/__init__.py b/model/__init__.py
index 5e9673e..988d60d 100644
--- a/model/__init__.py
+++ b/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']
\ No newline at end of file
+__all__ = ['Robot', 'Product', 'InventoryRecord', 'User', 'AIPrediction']
\ No newline at end of file
diff --git a/model/ai_prediction.py b/model/ai_prediction.py
new file mode 100644
index 0000000..1685105
--- /dev/null
+++ b/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
\ No newline at end of file