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): class EstimationDetails(BaseModel):
dtq: bool dtq: bool
reason: str reason: str
price_id: int = -1 # price_id: int = -1
tier: str tier: str
class EstimationResult(BaseModel): class EstimationResult(BaseModel):
@ -85,7 +85,7 @@ class EstimationResult(BaseModel):
class EstimationResponse(BaseModel): class EstimationResponse(BaseModel):
status: str status: str
details: EstimationDetails details: EstimationDetails
results: List[EstimationResult] # results: List[EstimationResult]
class UserNameContext(BaseModel): class UserNameContext(BaseModel):
first_name: str first_name: str

20
src/services/estimation_service_v2.py
View file

@ -571,10 +571,10 @@ class EstimationService:
details=EstimationDetails( details=EstimationDetails(
dtq=is_dtq, dtq=is_dtq,
reason=reason, reason=reason,
price_id=plan_price_id, # price_id=plan_price_id,
tier=f"Tier {base_tier.value}", tier=f"Tier {base_tier.value}",
), ),
results=estimation_results # results=estimation_results
) )
if is_review: if is_review:
@ -584,10 +584,10 @@ class EstimationService:
details=EstimationDetails( details=EstimationDetails(
dtq=is_dtq, dtq=is_dtq,
reason=reason, reason=reason,
price_id=plan_price_id, # price_id=plan_price_id,
tier=f"Tier {base_tier.value}", tier=f"Tier {base_tier.value}",
), ),
results=estimation_results # results=estimation_results
) )
new_tier, tier_reason = self.get_tier(plan_coverage, rx_spend) new_tier, tier_reason = self.get_tier(plan_coverage, rx_spend)
@ -600,10 +600,10 @@ class EstimationService:
details=EstimationDetails( details=EstimationDetails(
dtq=True, dtq=True,
reason=reason, reason=reason,
price_id=plan_price_id, # price_id=plan_price_id,
tier=f"Tier {base_tier.value}", tier=f"Tier {base_tier.value}",
), ),
results=estimation_results # results=estimation_results
) )
if new_tier > base_tier: if new_tier > base_tier:
@ -618,10 +618,10 @@ class EstimationService:
details=EstimationDetails( details=EstimationDetails(
dtq=is_dtq, dtq=is_dtq,
reason=reason, reason=reason,
price_id=plan_price_id, # price_id=plan_price_id,
tier=f"Tier {base_tier.value}", tier=f"Tier {base_tier.value}",
), ),
results=estimation_results # results=estimation_results
) )
else: else:
reason = "\n".join(dtq_reasons) reason = "\n".join(dtq_reasons)
@ -630,8 +630,8 @@ class EstimationService:
details=EstimationDetails( details=EstimationDetails(
dtq=is_dtq, dtq=is_dtq,
reason=reason, reason=reason,
price_id=plan_price_id, # price_id=plan_price_id,
tier=f"Tier {base_tier.value}", tier=f"Tier {base_tier.value}",
), ),
results=estimation_results # results=estimation_results
) )