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

new estimation data model

Browse source
This commit is contained in:
ipu 2025-10-03 22:53:17 +03:00
parent 9b1a6340e8
commit 13d9a76bed
3 changed files with 15 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/api/v1/router.py
View file

@ -85,7 +85,7 @@ async def estimate(request: models.EstimationRequest):
)
estimation_service = EstimationService()
estimation_response = await estimation_service.estimate_insurance(request.applicants, request.phq, request.plans)
estimation_response = await estimation_service.estimate_insurance(request.applicants, request.phq, request.plans, request.coverage)
return estimation_response

6
src/models.py
View file

@ -17,7 +17,7 @@ class Applicant(BaseModel):
class Plan(BaseModel):
id: int
priceId: int
priceId: int | None = None
class Medication(BaseModel):
applicant: int
@ -60,6 +60,7 @@ class Address(BaseModel):
class EstimationRequest(BaseModel):
userId: str | int | None = Field(None, description="Unique identifier")
coverage: int = 1
applicants: List[Applicant]
plans: List[Plan]
phq: PHQ
@ -69,7 +70,8 @@ class EstimationRequest(BaseModel):
class EstimationDetails(BaseModel):
dtq: bool
reason: str
price_id: int
price_id: int = -1
tier: str
class EstimationResult(BaseModel):
name: str

14
src/services/estimation_service_v2.py
View file

@ -458,7 +458,7 @@ class EstimationService:
return None
async def estimate_insurance(self, applicants: list[Applicant], phq: PHQ, plans: list[Plan]):
async def estimate_insurance(self, applicants: list[Applicant], phq: PHQ, plans: list[Plan], plan_coverage: int):
estimation_results = []
is_review = False
review_reasons = []
@ -478,7 +478,9 @@ class EstimationService:
base_tier = tier
break
plan_coverage = self.get_plan_coverage(plans[0])
if not plan_coverage:
plan_coverage = self.get_plan_coverage(plans[0])
rx_spend = 0
for applicant_id, applicant in enumerate(applicants):
applicant_review_reasons = []
@ -559,7 +561,7 @@ class EstimationService:
message=final_reason,
)
)
plan_price_id = self.get_plan_price(plans[0], base_tier, plan_coverage)
if is_dtq:
@ -570,6 +572,7 @@ class EstimationService:
dtq=is_dtq,
reason=reason,
price_id=plan_price_id,
tier=f"Tier {base_tier.value}",
),
results=estimation_results
)
@ -582,6 +585,7 @@ class EstimationService:
dtq=is_dtq,
reason=reason,
price_id=plan_price_id,
tier=f"Tier {base_tier.value}",
),
results=estimation_results
)
@ -597,6 +601,7 @@ class EstimationService:
dtq=True,
reason=reason,
price_id=plan_price_id,
tier=f"Tier {base_tier.value}",
),
results=estimation_results
)
@ -614,6 +619,7 @@ class EstimationService:
dtq=is_dtq,
reason=reason,
price_id=plan_price_id,
tier=f"Tier {base_tier.value}",
),
results=estimation_results
)
@ -625,7 +631,7 @@ class EstimationService:
dtq=is_dtq,
reason=reason,
price_id=plan_price_id,
tier=f"Tier {base_tier.value}",
),
results=estimation_results
)