catlog22/Claude-Code-Workflow
1
0
Fork 0
mirror of https://github.com/catlog22/Claude-Code-Workflow.git synced 2026-02-12 02:37:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(ranking): add explicit NaN check in normalize_weights

Browse Source 
Add math.isnan() check before math.isfinite() to properly catch
NaN values in weight totals. Prevents division by NaN which could
produce unexpected results in RRF fusion calculations.

Solution-ID: SOL-20251228113631
Issue-ID: ISS-1766921318981-0
Task-ID: T1
This commit is contained in:
catlog22
2025-12-28 20:55:03 +08:00
parent d3e7ecca21
commit 598eed92cb
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
codex-lens/src/codexlens/search/ranking.py
View File

@@ -22,12 +22,23 @@ class QueryIntent(str, Enum):
MIXED = "mixed" MIXED = "mixed"
def normalize_weights(weights: Dict[str, float]) -> Dict[str, float]: def normalize_weights(weights: Dict[str, float | None]) -> Dict[str, float | None]:
"""Normalize weights to sum to 1.0 (best-effort).""" """Normalize weights to sum to 1.0 (best-effort)."""
total = sum(float(v) for v in weights.values() if v is not None) total = sum(float(v) for v in weights.values() if v is not None)
if not math.isfinite(total) or total <= 0:
return {k: float(v) for k, v in weights.items()} # NaN total: do not attempt to normalize (division would propagate NaNs).
return {k: float(v) / total for k, v in weights.items()} if math.isnan(total):
return dict(weights)
# Infinite total: do not attempt to normalize (division yields 0 or NaN).
if not math.isfinite(total):
return dict(weights)
# Zero/negative total: do not attempt to normalize (invalid denominator).
if total <= 0:
return dict(weights)
return {k: (float(v) / total if v is not None else None) for k, v in weights.items()}
def detect_query_intent(query: str) -> QueryIntent: def detect_query_intent(query: str) -> QueryIntent: