From 25f091d02c68b4575d64f5a3178fb3d09da74f82 Mon Sep 17 00:00:00 2001
From: ipu <uwu@ipu.moe>
Date: Fri, 3 Oct 2025 23:03:06 +0300
Subject: [PATCH] new estimation data model

---
 src/models.py                         |  4 ++--
 src/services/estimation_service_v2.py | 20 ++++++++++----------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/models.py b/src/models.py
index 7cff934..71037d0 100644
--- a/src/models.py
+++ b/src/models.py
@@ -70,7 +70,7 @@ class EstimationRequest(BaseModel):
 class EstimationDetails(BaseModel):
     dtq: bool
     reason: str
-    price_id: int = -1
+    # price_id: int = -1
     tier: str
 
 class EstimationResult(BaseModel):
@@ -85,7 +85,7 @@ class EstimationResult(BaseModel):
 class EstimationResponse(BaseModel):
     status: str
     details: EstimationDetails
-    results: List[EstimationResult]
+    # results: List[EstimationResult]
 
 class UserNameContext(BaseModel):
     first_name: str
diff --git a/src/services/estimation_service_v2.py b/src/services/estimation_service_v2.py
index 2f7da6c..a5cecee 100644
--- a/src/services/estimation_service_v2.py
+++ b/src/services/estimation_service_v2.py
@@ -571,10 +571,10 @@ class EstimationService:
                 details=EstimationDetails(
                     dtq=is_dtq,
                     reason=reason,
-                    price_id=plan_price_id,
+                    # price_id=plan_price_id,
                     tier=f"Tier {base_tier.value}",
                 ),
-                results=estimation_results
+                # results=estimation_results
             )
 
         if is_review:
@@ -584,10 +584,10 @@ class EstimationService:
                 details=EstimationDetails(
                     dtq=is_dtq,
                     reason=reason,
-                    price_id=plan_price_id,
+                    # price_id=plan_price_id,
                     tier=f"Tier {base_tier.value}",
                 ),
-                results=estimation_results
+                # results=estimation_results
             )
         
         new_tier, tier_reason = self.get_tier(plan_coverage, rx_spend)
@@ -600,10 +600,10 @@ class EstimationService:
                 details=EstimationDetails(
                     dtq=True,
                     reason=reason,
-                    price_id=plan_price_id,
+                    # price_id=plan_price_id,
                     tier=f"Tier {base_tier.value}",
                 ),
-                results=estimation_results
+                # results=estimation_results
             )
 
         if new_tier > base_tier:
@@ -618,10 +618,10 @@ class EstimationService:
                 details=EstimationDetails(
                     dtq=is_dtq,
                     reason=reason,
-                    price_id=plan_price_id,
+                    # price_id=plan_price_id,
                     tier=f"Tier {base_tier.value}",
                 ),
-                results=estimation_results
+                # results=estimation_results
             )
         else:
             reason = "\n".join(dtq_reasons)
@@ -630,8 +630,8 @@ class EstimationService:
                 details=EstimationDetails(
                     dtq=is_dtq,
                     reason=reason,
-                    price_id=plan_price_id,
+                    # price_id=plan_price_id,
                     tier=f"Tier {base_tier.value}",
                 ),
-                results=estimation_results
+                # results=estimation_results
             )