From 88c4460da9fea7e547d5bf6d46fef604ad3e9d44 Mon Sep 17 00:00:00 2001
From: Kita Trofimov <144846094+NiTro005@users.noreply.github.com>
Date: Sun, 26 Oct 2025 15:52:48 +0300
Subject: [PATCH] feat(model/ai_prediction.py): Added new model AIPrediction

---
 model/__init__.py      |  3 ++-
 model/ai_prediction.py | 12 ++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 model/ai_prediction.py

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