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): class EstimationDetails(BaseModel):
dtq: bool dtq: bool
reason: str reason: str
priceId: int
tier: float tier: float
total_price: float total_price: float

17
src/services/estimation_service_v2.py
View file

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