ts/virgil-ai
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add price_id in result

Browse source
This commit is contained in:
ipu 2025-08-19 21:28:22 +03:00
parent 8ee261b1b3
commit 2713cc75a8
2 changed files with 8 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/models.py
View file

@ -68,6 +68,7 @@ class EstimationRequest(BaseModel):
class EstimationDetails(BaseModel):
dtq: bool
reason: str
priceId: int
tier: float
total_price: float

17
src/services/estimation_service_v2.py
View file

@ -226,7 +226,7 @@ class EstimationService:
prices = plan_data["prices"]
for price in prices:
if price["coverage"] == coverage and price["tier"] == f"Tier {tier.value}":
return price["price"]
return price["id"]
return 0
def calculate_bmi(self, weight, ft, inch):
@ -295,16 +295,15 @@ class EstimationService:
)
)
plan_price = self.get_plan_price(plans[0], base_tier, plan_coverage)
plan_price_id = self.get_plan_price(plans[0], base_tier, plan_coverage)
if is_dtq:
return EstimationResponse(
status="rejected",
status="rejected",
details=EstimationDetails(
dtq=is_dtq,
reason=reason,
tier=base_tier.value,
total_price=plan_price
price_id=plan_price_id,
),
results=estimation_results
)
@ -317,8 +316,7 @@ class EstimationService:
details=EstimationDetails(
dtq=True,
reason="Declined due to high Rx spend",
tier=base_tier.value,
total_price=plan_price
price_id=plan_price_id,
),
results=estimation_results
)
@ -326,7 +324,7 @@ class EstimationService:
if new_tier > base_tier:
base_tier = new_tier
plan_price = self.get_plan_price(plans[0], base_tier, plan_coverage)
plan_price_id = self.get_plan_price(plans[0], base_tier, plan_coverage)
status = "accepted" if base_tier is not None else "rejected"
@ -335,8 +333,7 @@ class EstimationService:
details=EstimationDetails(
dtq=is_dtq,
reason=reason,
tier=base_tier.value,
total_price=plan_price
price_id=plan_price_id,
),
results=estimation_results
)