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

get coverage from priceId

Browse source
This commit is contained in:
ipu 2025-08-19 13:22:19 +03:00
parent 0b54765c9b
commit c31f8e82e4
2 changed files with 16 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/models.py
View file

@ -17,8 +17,7 @@ class Applicant(BaseModel):
class Plan(BaseModel): class Plan(BaseModel):
id: int id: int
coverage: int priceId: int
tier: Optional[str] = Field(None, description="Tier assignment")
class Medication(BaseModel): class Medication(BaseModel):
applicant: int applicant: int

26
src/services/estimation_service_v2.py
View file

@ -213,12 +213,19 @@ class EstimationService:
return None return None
def get_plan_coverage(self, plan: Plan):
plan_data = get_plan_by_id(plan.id)
prices = plan_data["prices"]
for price in prices:
if price["id"] == plan.priceId:
return price["coverage"]
return 0
def get_plan_price(self, plan: Plan, tier: Tier, coverage: int): def get_plan_price(self, plan: Plan, tier: Tier, coverage: int):
plan_data = get_plan_by_id(plan.id) plan_data = get_plan_by_id(plan.id)
prices = plan_data["prices"] prices = plan_data["prices"]
tier_str = f"Tier {tier.value}"
for price in prices: for price in prices:
if price["tier"] == tier_str and price["coverage"] == coverage: if price["coverage"] == coverage and price["tier"] == f"Tier {tier.value}":
return price["price"] return price["price"]
return 0 return 0
@ -242,6 +249,7 @@ class EstimationService:
base_tier = tier base_tier = tier
break break
plan_coverage = self.get_plan_coverage(plans[0])
rx_spend = 0 rx_spend = 0
for applicant_id, applicant in enumerate(applicants): for applicant_id, applicant in enumerate(applicants):
applicant_age = self.calculate_age(applicant.dob) applicant_age = self.calculate_age(applicant.dob)
@ -266,7 +274,7 @@ class EstimationService:
rx_spend_applicant = await self.calculate_rx_spend(phq, applicant_id) rx_spend_applicant = await self.calculate_rx_spend(phq, applicant_id)
rx_spend += rx_spend_applicant rx_spend += rx_spend_applicant
applicant_new_tier = self.get_tier(plans[0].coverage, rx_spend_applicant) applicant_new_tier = self.get_tier(plan_coverage, rx_spend_applicant)
if applicant_new_tier is None: if applicant_new_tier is None:
is_dtq = True is_dtq = True
reason = "Declined due to high Rx spend" reason = "Declined due to high Rx spend"
@ -287,7 +295,7 @@ class EstimationService:
) )
) )
plan_price = self.get_plan_price(plans[0], base_tier, plans[0].coverage) plan_price = self.get_plan_price(plans[0], base_tier, plan_coverage)
if is_dtq: if is_dtq:
return EstimationResponse( return EstimationResponse(
@ -301,7 +309,7 @@ class EstimationService:
results=estimation_results results=estimation_results
) )
new_tier = self.get_tier(plans[0].coverage, rx_spend) new_tier = self.get_tier(plan_coverage, rx_spend)
if new_tier is None: if new_tier is None:
return EstimationResponse( return EstimationResponse(
@ -318,7 +326,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, plans[0].coverage) plan_price = 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"
@ -332,8 +340,4 @@ class EstimationService:
), ),
results=estimation_results results=estimation_results
) )