new fields with ml_rerank metadata

Author: SentienceDEV

sentience/models.py

@@ -49,7 +49,7 @@ class Element(BaseModel):
     z_index: int = 0
 
     # ML reranking metadata (optional - can be absent or null)
-    rerank_index: int | None = None  # 0-based, The rank after ML reranking
+    fused_rank_index: int | None = None  # 0-based, The rank index after ML reranking
     heuristic_index: int | None = None  # 0-based, Where it would have been without ML
     ml_probability: float | None = None  # Confidence score from ONNX model (0.0 - 1.0)
     ml_score: float | None = None  # Raw logit score (optional, for debugging)

sentience/schemas/trace_v1.json

@@ -149,7 +149,7 @@
                   "in_viewport": {"type": "boolean"},
                   "is_occluded": {"type": "boolean"},
                   "z_index": {"type": "integer"},
-                  "rerank_index": {"type": ["integer", "null"]},
+                  "fused_rank_index": {"type": ["integer", "null"]},
                   "heuristic_index": {"type": ["integer", "null"]},
                   "ml_probability": {"type": ["number", "null"]},
                   "ml_score": {"type": ["number", "null"]},

tests/test_snapshot.py

@@ -152,7 +152,7 @@ def test_element_ml_fields_optional():
         is_occluded=False,
         z_index=0,
     )
-    assert element_without_ml.rerank_index is None
+    assert element_without_ml.fused_rank_index is None
     assert element_without_ml.heuristic_index is None
     assert element_without_ml.ml_probability is None
     assert element_without_ml.ml_score is None
@@ -168,12 +168,12 @@ def test_element_ml_fields_optional():
         in_viewport=True,
         is_occluded=False,
         z_index=1,
-        rerank_index=0,
+        fused_rank_index=0,
         heuristic_index=5,
         ml_probability=0.95,
         ml_score=2.34,
     )
-    assert element_with_ml.rerank_index == 0
+    assert element_with_ml.fused_rank_index == 0
     assert element_with_ml.heuristic_index == 5
     assert element_with_ml.ml_probability == 0.95
     assert element_with_ml.ml_score == 2.34
@@ -189,10 +189,10 @@ def test_element_ml_fields_optional():
         in_viewport=True,
         is_occluded=False,
         z_index=0,
-        rerank_index=1,
+        fused_rank_index=1,
         ml_probability=0.87,
     )
-    assert element_partial.rerank_index == 1
+    assert element_partial.fused_rank_index == 1
     assert element_partial.heuristic_index is None
     assert element_partial.ml_probability == 0.87
     assert element_partial.ml_score is None