add ml_rerank metadata to snapshot response

Author: SentienceDEV

src/types.ts

@@ -143,6 +143,30 @@ export interface GridInfo {
   viewport_coverage?: number;
 }
 
+export interface MlRerankTags {
+  repeated: boolean;
+  sponsored_ish: boolean;
+  non_sponsored: boolean;
+  pos: boolean;
+  occ: boolean;
+  vocc: boolean;
+  short: boolean;
+  action_ish: boolean;
+  nav_ish: boolean;
+}
+
+export interface MlRerankInfo {
+  enabled: boolean;
+  applied: boolean;
+  reason?: string | null;
+  candidate_count?: number | null;
+  top_probability?: number | null;
+  min_confidence?: number | null;
+  is_high_confidence?: boolean | null;
+  tags?: MlRerankTags | null;
+  error?: string | null;
+}
+
 export interface Snapshot {
   status: 'success' | 'error';
   timestamp?: string;
@@ -162,6 +186,8 @@ export interface Snapshot {
   modal_detected?: boolean;
   /** Array of GridInfo for detected modal grids */
   modal_grids?: GridInfo[];
+  /** ML rerank metadata (optional) */
+  ml_rerank?: MlRerankInfo;
 }
 
 export interface StepHookContext {

tests/snapshot.test.ts

@@ -145,3 +145,36 @@ describe('Element ML Fields', () => {
     expect(element).not.toHaveProperty('ml_score');
   });
 });
+
+describe('Snapshot ML Rerank Metadata', () => {
+  it('should accept snapshot with ml_rerank metadata', () => {
+    const snap = {
+      status: 'success' as const,
+      url: 'https://example.com',
+      elements: [],
+      ml_rerank: {
+        enabled: true,
+        applied: false,
+        reason: 'low_confidence',
+        candidate_count: 25,
+        top_probability: 0.42,
+        min_confidence: 0.6,
+        is_high_confidence: false,
+        tags: {
+          repeated: true,
+          sponsored_ish: true,
+          non_sponsored: false,
+          pos: true,
+          occ: true,
+          vocc: false,
+          short: true,
+          action_ish: false,
+          nav_ish: false,
+        },
+      },
+    };
+
+    expect(snap.ml_rerank.enabled).toBe(true);
+    expect(snap.ml_rerank.is_high_confidence).toBe(false);
+  });
+});