abinazebinoy commited on
Commit
84e6f45
·
1 Parent(s): 719142b

fix(api): BUG-12 -- SearchResult model missing risk_level field

Browse files

The frontend ResultCard component renders:
class='risk-badge risk-badge--${entity.risk_level || "LOW"}'

risk_level was never in the SearchResult Pydantic model so it
was never included in API responses. Every search result risk
badge rendered as LOW regardless of actual risk. Added
risk_level: Optional[str] = None to SearchResult.

Files changed (1) hide show
  1. api/models.py +4 -0
api/models.py CHANGED
@@ -21,6 +21,10 @@ class SearchResult(BaseModel):
21
  state: Optional[str] = None
22
  party: Optional[str] = None
23
  risk_score: Optional[int] = None
 
 
 
 
24
  sources: List[SourceDocument] = []
25
 
26
 
 
21
  state: Optional[str] = None
22
  party: Optional[str] = None
23
  risk_score: Optional[int] = None
24
+ # BUG-12 FIX: risk_level was missing -- frontend Components.ResultCard renders
25
+ # class="risk-badge risk-badge--${entity.risk_level}" so without this field
26
+ # every badge was always risk-badge--undefined which rendered as LOW visually.
27
+ risk_level: Optional[str] = None
28
  sources: List[SourceDocument] = []
29
 
30