Update lineage_tracker.py

lineage_tracker.py

@@ -1,15 +1,48 @@
+# lineage_tracker.py
 # Author: Liam Grinstead
-# Tracks symbolic agent lineage and mutation history
+# Tracks symbolic agent lineage, mutation history, and parent mapping
 
-LINEAGE = {}
+from typing import Dict, List
 
-def register_lineage(parent_id, child_id, mutation_profile):
+LINEAGE: Dict[str, List[dict]] = {}
+PARENT_OF: Dict[str, str] = {}  # child_id -> parent_id
+
+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({
         "child_id": child_id,
         "mutation": mutation_profile
     })
+    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
+    while cur in PARENT_OF:
+        p = PARENT_OF[cur]
+        chain.append(p)
+        cur = p
+    return chain[::-1]  # oldest → newest
 
-def get_lineage(agent_id):
-    return LINEAGE.get(agent_id, [])
+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]] = {}
+    frontier = [agent_id]
+    for d in range(1, depth + 1):
+        next_frontier = []
+        for node in frontier:
+            children = [c["child_id"] for c in LINEAGE.get(node, [])]
+            if children:
+                levels.setdefault(d, []).extend(children)
+                next_frontier.extend(children)
+        frontier = next_frontier
+        if not frontier:
+            break
+    return {f"level_{k}": v for k, v in levels.items()}