Update lineage_tracker.py

lineage_tracker.py

@@ -1,27 +1,22 @@
-# lineage_tracker.py
+# lineage_tracker_rft.py
 # Author: Liam Grinstead
-# Tracks symbolic agent lineage, mutation history, and parent mapping
+# Multi-gen lineage tracking with mutation provenance (operators & overlays)
 
 from typing import Dict, List
 
 LINEAGE: Dict[str, List[dict]] = {}
-PARENT_OF: Dict[str, str] = {}  # child_id -> parent_id
+PARENT_OF: Dict[str, str] = {}
 
 def register_lineage(parent_id: str, child_id: str, mutation_profile: dict):
-    if parent_id not in LINEAGE:
-        LINEAGE[parent_id] = []
-    LINEAGE[parent_id].append({
+    entry = {
         "child_id": child_id,
-        "mutation": mutation_profile
-    })
+        "tier": mutation_profile.get("tier_drift"),
+        "overlay": mutation_profile.get("collapse_torque"),
+        "operators": mutation_profile.get("symbolic_operators", []),
+    }
+    LINEAGE.setdefault(parent_id, []).append(entry)
     PARENT_OF[child_id] = parent_id
 
-def get_lineage(agent_id: str) -> List[dict]:
-    return LINEAGE.get(agent_id, [])
-
-def get_parent(child_id: str) -> str:
-    return PARENT_OF.get(child_id, "")
-
 def get_ancestors(agent_id: str) -> List[str]:
     chain = []
     cur = agent_id
@@ -29,20 +24,19 @@ def get_ancestors(agent_id: str) -> List[str]:
         p = PARENT_OF[cur]
         chain.append(p)
         cur = p
-    return chain[::-1]  # oldest → newest
+    return chain[::-1]
 
-def get_descendants(agent_id: str, depth: int = 5) -> Dict[str, List[str]]:
-    # returns a dict level -> list of child ids
-    levels: Dict[int, List[str]] = {}
+def get_descendants(agent_id: str, depth: int = 6) -> Dict[int, List[dict]]:
+    levels: Dict[int, List[dict]] = {}
     frontier = [agent_id]
     for d in range(1, depth + 1):
-        next_frontier = []
+        nxt = []
         for node in frontier:
-            children = [c["child_id"] for c in LINEAGE.get(node, [])]
+            children = LINEAGE.get(node, [])
             if children:
                 levels.setdefault(d, []).extend(children)
-                next_frontier.extend(children)
-        frontier = next_frontier
+                nxt.extend([c["child_id"] for c in children])
+        frontier = nxt
         if not frontier:
             break
-    return {f"level_{k}": v for k, v in levels.items()}
+    return levels