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 23:03:06 +03:00
parent 13d9a76bed
commit 25f091d02c
2 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/models.py
View file

@ -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

20
src/services/estimation_service_v2.py
View file

@ -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
)