okstra 0.130.3 → 0.131.0

package/runtime/python/okstra_vendor/graphify/analyze.py

@@ -1,540 +0,0 @@
-"""Graph analysis: god nodes (most connected), surprising connections (cross-community), suggested questions."""
-from __future__ import annotations
-import networkx as nx
-
-
-def _node_community_map(communities: dict[int, list[str]]) -> dict[str, int]:
-    """Invert communities dict: node_id -> community_id."""
-    return {n: cid for cid, nodes in communities.items() for n in nodes}
-
-
-def _is_file_node(G: nx.Graph, node_id: str) -> bool:
-    """
-    Return True if this node is a file-level hub node (e.g. 'client', 'models')
-    or an AST method stub (e.g. '.auth_flow()', '.__init__()').
-
-    These are synthetic nodes created by the AST extractor and should be excluded
-    from god nodes, surprising connections, and knowledge gap reporting.
-    """
-    attrs = G.nodes[node_id]
-    label = attrs.get("label", "")
-    if not label:
-        return False
-    # File-level hub: label matches the actual source filename (not just any label ending in .py)
-    source_file = attrs.get("source_file", "")
-    if source_file:
-        from pathlib import Path as _Path
-        if label == _Path(source_file).name:
-            return True
-    # Method stub: AST extractor labels methods as '.method_name()'
-    if label.startswith(".") and label.endswith("()"):
-        return True
-    # Module-level function stub: labeled 'function_name()' - only has a contains edge
-    # These are real functions but structurally isolated by definition; not a gap worth flagging
-    if label.endswith("()") and G.degree(node_id) <= 1:
-        return True
-    return False
-
-
-def god_nodes(G: nx.Graph, top_n: int = 10) -> list[dict]:
-    """Return the top_n most-connected real entities - the core abstractions.
-
-    File-level hub nodes are excluded: they accumulate import/contains edges
-    mechanically and don't represent meaningful architectural abstractions.
-    """
-    degree = dict(G.degree())
-    sorted_nodes = sorted(degree.items(), key=lambda x: x[1], reverse=True)
-    result = []
-    for node_id, deg in sorted_nodes:
-        if _is_file_node(G, node_id) or _is_concept_node(G, node_id):
-            continue
-        result.append({
-            "id": node_id,
-            "label": G.nodes[node_id].get("label", node_id),
-            "degree": deg,
-        })
-        if len(result) >= top_n:
-            break
-    return result
-
-
-def surprising_connections(
-    G: nx.Graph,
-    communities: dict[int, list[str]] | None = None,
-    top_n: int = 5,
-) -> list[dict]:
-    """
-    Find connections that are genuinely surprising - not obvious from file structure.
-
-    Strategy:
-    - Multi-file corpora: cross-file edges between real entities (not concept nodes).
-      Sorted AMBIGUOUS → INFERRED → EXTRACTED.
-    - Single-file / single-source corpora: cross-community edges that bridge
-      distant parts of the graph (betweenness centrality on edges).
-      These reveal non-obvious structural couplings.
-
-    Concept nodes (empty source_file, or injected semantic annotations) are excluded
-    from surprising connections because they are intentional, not discovered.
-    """
-    # Identify unique source files (ignore empty/null source_file)
-    source_files = {
-        data.get("source_file", "")
-        for _, data in G.nodes(data=True)
-        if data.get("source_file", "")
-    }
-    is_multi_source = len(source_files) > 1
-
-    if is_multi_source:
-        return _cross_file_surprises(G, communities or {}, top_n)
-    else:
-        return _cross_community_surprises(G, communities or {}, top_n)
-
-
-def _is_concept_node(G: nx.Graph, node_id: str) -> bool:
-    """
-    Return True if this node is a manually-injected semantic concept node
-    rather than a real entity found in source code.
-
-    Signals:
-    - Empty source_file
-    - source_file doesn't look like a real file path (no extension)
-    """
-    data = G.nodes[node_id]
-    source = data.get("source_file", "")
-    if not source:
-        return True
-    # Has no file extension → probably a concept label, not a real file
-    if "." not in source.split("/")[-1]:
-        return True
-    return False
-
-
-from graphify.detect import CODE_EXTENSIONS, DOC_EXTENSIONS, PAPER_EXTENSIONS, IMAGE_EXTENSIONS
-
-
-def _file_category(path: str) -> str:
-    ext = ("." + path.rsplit(".", 1)[-1].lower()) if "." in path else ""
-    if ext in CODE_EXTENSIONS:
-        return "code"
-    if ext in PAPER_EXTENSIONS:
-        return "paper"
-    if ext in IMAGE_EXTENSIONS:
-        return "image"
-    return "doc"
-
-
-def _top_level_dir(path: str) -> str:
-    """Return the first path component - used to detect cross-repo edges."""
-    return path.split("/")[0] if "/" in path else path
-
-
-def _surprise_score(
-    G: nx.Graph,
-    u: str,
-    v: str,
-    data: dict,
-    node_community: dict[str, int],
-    u_source: str,
-    v_source: str,
-) -> tuple[int, list[str]]:
-    """Score how surprising a cross-file edge is. Returns (score, reasons)."""
-    score = 0
-    reasons: list[str] = []
-
-    # 1. Confidence weight - uncertain connections are more noteworthy
-    conf = data.get("confidence", "EXTRACTED")
-    conf_bonus = {"AMBIGUOUS": 3, "INFERRED": 2, "EXTRACTED": 1}.get(conf, 1)
-    score += conf_bonus
-    if conf in ("AMBIGUOUS", "INFERRED"):
-        reasons.append(f"{conf.lower()} connection - not explicitly stated in source")
-
-    # 2. Cross file-type bonus - code↔paper or code↔image is non-obvious
-    cat_u = _file_category(u_source)
-    cat_v = _file_category(v_source)
-    if cat_u != cat_v:
-        score += 2
-        reasons.append(f"crosses file types ({cat_u} ↔ {cat_v})")
-
-    # 3. Cross-repo bonus - different top-level directory
-    if _top_level_dir(u_source) != _top_level_dir(v_source):
-        score += 2
-        reasons.append("connects across different repos/directories")
-
-    # 4. Cross-community bonus - Leiden says these are structurally distant
-    cid_u = node_community.get(u)
-    cid_v = node_community.get(v)
-    if cid_u is not None and cid_v is not None and cid_u != cid_v:
-        score += 1
-        reasons.append("bridges separate communities")
-
-    # 4b. Semantic similarity bonus - non-obvious conceptual links score higher
-    if data.get("relation") == "semantically_similar_to":
-        score = int(score * 1.5)
-        reasons.append("semantically similar concepts with no structural link")
-
-    # 5. Peripheral→hub: a low-degree node connecting to a high-degree one
-    deg_u = G.degree(u)
-    deg_v = G.degree(v)
-    if min(deg_u, deg_v) <= 2 and max(deg_u, deg_v) >= 5:
-        score += 1
-        peripheral = G.nodes[u].get("label", u) if deg_u <= 2 else G.nodes[v].get("label", v)
-        hub = G.nodes[v].get("label", v) if deg_u <= 2 else G.nodes[u].get("label", u)
-        reasons.append(f"peripheral node `{peripheral}` unexpectedly reaches hub `{hub}`")
-
-    return score, reasons
-
-
-def _cross_file_surprises(G: nx.Graph, communities: dict[int, list[str]], top_n: int) -> list[dict]:
-    """
-    Cross-file edges between real code/doc entities, ranked by a composite
-    surprise score rather than confidence alone.
-
-    Surprise score accounts for:
-    - Confidence (AMBIGUOUS > INFERRED > EXTRACTED)
-    - Cross file-type (code↔paper is more surprising than code↔code)
-    - Cross-repo (different top-level directory)
-    - Cross-community (Leiden says structurally distant)
-    - Peripheral→hub (low-degree node reaching a god node)
-
-    Each result includes a 'why' field explaining what makes it non-obvious.
-    """
-    node_community = _node_community_map(communities)
-    candidates = []
-
-    for u, v, data in G.edges(data=True):
-        relation = data.get("relation", "")
-        if relation in ("imports", "imports_from", "contains", "method"):
-            continue
-        if _is_concept_node(G, u) or _is_concept_node(G, v):
-            continue
-        if _is_file_node(G, u) or _is_file_node(G, v):
-            continue
-
-        u_source = G.nodes[u].get("source_file", "")
-        v_source = G.nodes[v].get("source_file", "")
-
-        if not u_source or not v_source or u_source == v_source:
-            continue
-
-        score, reasons = _surprise_score(G, u, v, data, node_community, u_source, v_source)
-        src_id = data.get("_src", u)
-        if src_id not in G.nodes:
-            src_id = u
-        tgt_id = data.get("_tgt", v)
-        if tgt_id not in G.nodes:
-            tgt_id = v
-        candidates.append({
-            "_score": score,
-            "source": G.nodes[src_id].get("label", src_id),
-            "target": G.nodes[tgt_id].get("label", tgt_id),
-            "source_files": [
-                G.nodes[src_id].get("source_file", ""),
-                G.nodes[tgt_id].get("source_file", ""),
-            ],
-            "confidence": data.get("confidence", "EXTRACTED"),
-            "relation": relation,
-            "why": "; ".join(reasons) if reasons else "cross-file semantic connection",
-        })
-
-    candidates.sort(key=lambda x: x["_score"], reverse=True)
-    for c in candidates:
-        c.pop("_score")
-
-    if candidates:
-        return candidates[:top_n]
-
-    return _cross_community_surprises(G, communities, top_n)
-
-
-def _cross_community_surprises(
-    G: nx.Graph,
-    communities: dict[int, list[str]],
-    top_n: int,
-) -> list[dict]:
-    """
-    For single-source corpora: find edges that bridge different communities.
-    These are surprising because Leiden grouped everything else tightly -
-    these edges cut across the natural structure.
-
-    Falls back to high-betweenness edges if no community info is provided.
-    """
-    if not communities:
-        # No community info - use edge betweenness centrality
-        if G.number_of_edges() == 0:
-            return []
-        if G.number_of_nodes() > 5000:
-            return []
-        betweenness = nx.edge_betweenness_centrality(G)
-        top_edges = sorted(betweenness.items(), key=lambda x: x[1], reverse=True)[:top_n]
-        result = []
-        for (u, v), score in top_edges:
-            data = G.edges[u, v]
-            result.append({
-                "source": G.nodes[u].get("label", u),
-                "target": G.nodes[v].get("label", v),
-                "source_files": [
-                    G.nodes[u].get("source_file", ""),
-                    G.nodes[v].get("source_file", ""),
-                ],
-                "confidence": data.get("confidence", "EXTRACTED"),
-                "relation": data.get("relation", ""),
-                "note": f"Bridges graph structure (betweenness={score:.3f})",
-            })
-        return result
-
-    # Build node → community map
-    node_community = _node_community_map(communities)
-
-    surprises = []
-    for u, v, data in G.edges(data=True):
-        cid_u = node_community.get(u)
-        cid_v = node_community.get(v)
-        if cid_u is None or cid_v is None or cid_u == cid_v:
-            continue
-        # Skip file hub nodes and plain structural edges
-        if _is_file_node(G, u) or _is_file_node(G, v):
-            continue
-        relation = data.get("relation", "")
-        if relation in ("imports", "imports_from", "contains", "method"):
-            continue
-        # This edge crosses community boundaries - interesting
-        confidence = data.get("confidence", "EXTRACTED")
-        src_id = data.get("_src", u)
-        if src_id not in G.nodes:
-            src_id = u
-        tgt_id = data.get("_tgt", v)
-        if tgt_id not in G.nodes:
-            tgt_id = v
-        surprises.append({
-            "source": G.nodes[src_id].get("label", src_id),
-            "target": G.nodes[tgt_id].get("label", tgt_id),
-            "source_files": [
-                G.nodes[src_id].get("source_file", ""),
-                G.nodes[tgt_id].get("source_file", ""),
-            ],
-            "confidence": confidence,
-            "relation": relation,
-            "note": f"Bridges community {cid_u} → community {cid_v}",
-            "_pair": tuple(sorted([cid_u, cid_v])),
-        })
-
-    # Sort: AMBIGUOUS first, then INFERRED, then EXTRACTED
-    order = {"AMBIGUOUS": 0, "INFERRED": 1, "EXTRACTED": 2}
-    surprises.sort(key=lambda x: order.get(x["confidence"], 3))
-
-    # Deduplicate by community pair - one representative edge per (A→B) boundary.
-    # Without this, a single high-betweenness god node dominates all results.
-    seen_pairs: set[tuple] = set()
-    deduped = []
-    for s in surprises:
-        pair = s.pop("_pair")
-        if pair not in seen_pairs:
-            seen_pairs.add(pair)
-            deduped.append(s)
-    return deduped[:top_n]
-
-
-def suggest_questions(
-    G: nx.Graph,
-    communities: dict[int, list[str]],
-    community_labels: dict[int, str],
-    top_n: int = 7,
-) -> list[dict]:
-    """
-    Generate questions the graph is uniquely positioned to answer.
-    Based on: AMBIGUOUS edges, bridge nodes, underexplored god nodes, isolated nodes.
-    Each question has a 'type', 'question', and 'why' field.
-    """
-    questions = []
-    node_community = _node_community_map(communities)
-
-    # 1. AMBIGUOUS edges → unresolved relationship questions
-    for u, v, data in G.edges(data=True):
-        if data.get("confidence") == "AMBIGUOUS":
-            ul = G.nodes[u].get("label", u)
-            vl = G.nodes[v].get("label", v)
-            relation = data.get("relation", "related to")
-            questions.append({
-                "type": "ambiguous_edge",
-                "question": f"What is the exact relationship between `{ul}` and `{vl}`?",
-                "why": f"Edge tagged AMBIGUOUS (relation: {relation}) - confidence is low.",
-            })
-
-    # 2. Bridge nodes (high betweenness) → cross-cutting concern questions
-    if G.number_of_edges() > 0:
-        k = min(100, G.number_of_nodes()) if G.number_of_nodes() > 1000 else None
-        betweenness = nx.betweenness_centrality(G, k=k)
-        # Top bridge nodes that are NOT file-level hubs
-        bridges = sorted(
-            [(n, s) for n, s in betweenness.items()
-             if not _is_file_node(G, n) and not _is_concept_node(G, n) and s > 0],
-            key=lambda x: x[1],
-            reverse=True,
-        )[:3]
-        for node_id, score in bridges:
-            label = G.nodes[node_id].get("label", node_id)
-            cid = node_community.get(node_id)
-            comm_label = community_labels.get(cid, f"Community {cid}") if cid is not None else "unknown"
-            neighbors = list(G.neighbors(node_id))
-            neighbor_comms = {node_community.get(n) for n in neighbors if node_community.get(n) != cid}
-            if neighbor_comms:
-                other_labels = [community_labels.get(c, f"Community {c}") for c in neighbor_comms]
-                questions.append({
-                    "type": "bridge_node",
-                    "question": f"Why does `{label}` connect `{comm_label}` to {', '.join(f'`{l}`' for l in other_labels)}?",
-                    "why": f"High betweenness centrality ({score:.3f}) - this node is a cross-community bridge.",
-                })
-
-    # 3. God nodes with many INFERRED edges → verification questions
-    degree = dict(G.degree())
-    top_nodes = sorted(
-        [(n, d) for n, d in degree.items() if not _is_file_node(G, n)],
-        key=lambda x: x[1],
-        reverse=True,
-    )[:5]
-    for node_id, _ in top_nodes:
-        inferred = [
-            (u, v, d) for u, v, d in G.edges(node_id, data=True)
-            if d.get("confidence") == "INFERRED"
-        ]
-        if len(inferred) >= 2:
-            label = G.nodes[node_id].get("label", node_id)
-            # Use _src/_tgt to get the correct direction; fall back to v (the other node)
-            others = []
-            for u, v, d in inferred[:2]:
-                src_id = d.get("_src", u)
-                if src_id not in G.nodes:
-                    src_id = u
-                tgt_id = d.get("_tgt", v)
-                if tgt_id not in G.nodes:
-                    tgt_id = v
-                other_id = tgt_id if src_id == node_id else src_id
-                others.append(G.nodes[other_id].get("label", other_id))
-            questions.append({
-                "type": "verify_inferred",
-                "question": f"Are the {len(inferred)} inferred relationships involving `{label}` (e.g. with `{others[0]}` and `{others[1]}`) actually correct?",
-                "why": f"`{label}` has {len(inferred)} INFERRED edges - model-reasoned connections that need verification.",
-            })
-
-    # 4. Isolated or weakly-connected nodes → exploration questions
-    isolated = [
-        n for n in G.nodes()
-        if G.degree(n) <= 1 and not _is_file_node(G, n) and not _is_concept_node(G, n)
-    ]
-    if isolated:
-        labels = [G.nodes[n].get("label", n) for n in isolated[:3]]
-        questions.append({
-            "type": "isolated_nodes",
-            "question": f"What connects {', '.join(f'`{l}`' for l in labels)} to the rest of the system?",
-            "why": f"{len(isolated)} weakly-connected nodes found - possible documentation gaps or missing edges.",
-        })
-
-    # 5. Low-cohesion communities → structural questions
-    from .cluster import cohesion_score
-    for cid, nodes in communities.items():
-        score = cohesion_score(G, nodes)
-        if score < 0.15 and len(nodes) >= 5:
-            label = community_labels.get(cid, f"Community {cid}")
-            questions.append({
-                "type": "low_cohesion",
-                "question": f"Should `{label}` be split into smaller, more focused modules?",
-                "why": f"Cohesion score {score} - nodes in this community are weakly interconnected.",
-            })
-
-    if not questions:
-        return [{
-            "type": "no_signal",
-            "question": None,
-            "why": (
-                "Not enough signal to generate questions. "
-                "This usually means the corpus has no AMBIGUOUS edges, no bridge nodes, "
-                "no INFERRED relationships, and all communities are tightly cohesive. "
-                "Add more files or run with --mode deep to extract richer edges."
-            ),
-        }]
-
-    return questions[:top_n]
-
-
-def graph_diff(G_old: nx.Graph, G_new: nx.Graph) -> dict:
-    """Compare two graph snapshots and return what changed.
-
-    Returns:
-        {
-          "new_nodes": [{"id": ..., "label": ...}],
-          "removed_nodes": [{"id": ..., "label": ...}],
-          "new_edges": [{"source": ..., "target": ..., "relation": ..., "confidence": ...}],
-          "removed_edges": [...],
-          "summary": "3 new nodes, 5 new edges, 1 node removed"
-        }
-    """
-    old_nodes = set(G_old.nodes())
-    new_nodes = set(G_new.nodes())
-
-    added_node_ids = new_nodes - old_nodes
-    removed_node_ids = old_nodes - new_nodes
-
-    new_nodes_list = [
-        {"id": n, "label": G_new.nodes[n].get("label", n)}
-        for n in added_node_ids
-    ]
-    removed_nodes_list = [
-        {"id": n, "label": G_old.nodes[n].get("label", n)}
-        for n in removed_node_ids
-    ]
-
-    def edge_key(G: nx.Graph, u: str, v: str, data: dict) -> tuple:
-        if G.is_directed():
-            return (u, v, data.get("relation", ""))
-        return (min(u, v), max(u, v), data.get("relation", ""))
-
-    old_edge_keys = {
-        edge_key(G_old, u, v, d)
-        for u, v, d in G_old.edges(data=True)
-    }
-    new_edge_keys = {
-        edge_key(G_new, u, v, d)
-        for u, v, d in G_new.edges(data=True)
-    }
-
-    added_edge_keys = new_edge_keys - old_edge_keys
-    removed_edge_keys = old_edge_keys - new_edge_keys
-
-    new_edges_list = []
-    for u, v, d in G_new.edges(data=True):
-        if edge_key(G_new, u, v, d) in added_edge_keys:
-            new_edges_list.append({
-                "source": u,
-                "target": v,
-                "relation": d.get("relation", ""),
-                "confidence": d.get("confidence", ""),
-            })
-
-    removed_edges_list = []
-    for u, v, d in G_old.edges(data=True):
-        if edge_key(G_old, u, v, d) in removed_edge_keys:
-            removed_edges_list.append({
-                "source": u,
-                "target": v,
-                "relation": d.get("relation", ""),
-                "confidence": d.get("confidence", ""),
-            })
-
-    parts = []
-    if new_nodes_list:
-        parts.append(f"{len(new_nodes_list)} new node{'s' if len(new_nodes_list) != 1 else ''}")
-    if new_edges_list:
-        parts.append(f"{len(new_edges_list)} new edge{'s' if len(new_edges_list) != 1 else ''}")
-    if removed_nodes_list:
-        parts.append(f"{len(removed_nodes_list)} node{'s' if len(removed_nodes_list) != 1 else ''} removed")
-    if removed_edges_list:
-        parts.append(f"{len(removed_edges_list)} edge{'s' if len(removed_edges_list) != 1 else ''} removed")
-    summary = ", ".join(parts) if parts else "no changes"
-
-    return {
-        "new_nodes": new_nodes_list,
-        "removed_nodes": removed_nodes_list,
-        "new_edges": new_edges_list,
-        "removed_edges": removed_edges_list,
-        "summary": summary,
-    }

package/runtime/python/okstra_vendor/graphify/benchmark.py

@@ -1,129 +0,0 @@
-"""Token-reduction benchmark - measures how much context graphify saves vs naive full-corpus approach."""
-from __future__ import annotations
-import json
-from pathlib import Path
-import networkx as nx
-from networkx.readwrite import json_graph
-
-
-_CHARS_PER_TOKEN = 4  # standard approximation
-
-
-def _estimate_tokens(text: str) -> int:
-    return max(1, len(text) // _CHARS_PER_TOKEN)
-
-
-def _query_subgraph_tokens(G: nx.Graph, question: str, depth: int = 3) -> int:
-    """Run BFS from best-matching nodes and return estimated tokens in the subgraph context."""
-    terms = [t.lower() for t in question.split() if len(t) > 2]
-    scored = []
-    for nid, data in G.nodes(data=True):
-        label = data.get("label", "").lower()
-        score = sum(1 for t in terms if t in label)
-        if score > 0:
-            scored.append((score, nid))
-    scored.sort(reverse=True)
-    start_nodes = [nid for _, nid in scored[:3]]
-    if not start_nodes:
-        return 0
-
-    visited: set[str] = set(start_nodes)
-    frontier = set(start_nodes)
-    edges_seen: list[tuple] = []
-    for _ in range(depth):
-        next_frontier: set[str] = set()
-        for n in frontier:
-            for neighbor in G.neighbors(n):
-                if neighbor not in visited:
-                    next_frontier.add(neighbor)
-                    edges_seen.append((n, neighbor))
-        visited.update(next_frontier)
-        frontier = next_frontier
-
-    lines = []
-    for nid in visited:
-        d = G.nodes[nid]
-        lines.append(f"NODE {d.get('label', nid)} src={d.get('source_file', '')} loc={d.get('source_location', '')}")
-    for u, v in edges_seen:
-        if u in visited and v in visited:
-            d = G.edges[u, v]
-            lines.append(f"EDGE {G.nodes[u].get('label', u)} --{d.get('relation', '')}--> {G.nodes[v].get('label', v)}")
-
-    return _estimate_tokens("\n".join(lines))
-
-
-_SAMPLE_QUESTIONS = [
-    "how does authentication work",
-    "what is the main entry point",
-    "how are errors handled",
-    "what connects the data layer to the api",
-    "what are the core abstractions",
-]
-
-
-def run_benchmark(
-    graph_path: str = "graphify-out/graph.json",
-    corpus_words: int | None = None,
-    questions: list[str] | None = None,
-) -> dict:
-    """Measure token reduction: corpus tokens vs graphify query tokens.
-
-    Args:
-        graph_path: path to the built graph
-        corpus_words: total word count from detect() output; if None, estimated from graph
-        questions: list of questions to benchmark; defaults to _SAMPLE_QUESTIONS
-
-    Returns dict with: corpus_tokens, avg_query_tokens, reduction_ratio, per_question
-    """
-    data = json.loads(Path(graph_path).read_text(encoding="utf-8"))
-    try:
-        G = json_graph.node_link_graph(data, edges="links")
-    except TypeError:
-        G = json_graph.node_link_graph(data)
-
-    if corpus_words is None:
-        # Rough estimate: each node label is ~3 words, plus source context
-        corpus_words = G.number_of_nodes() * 50
-
-    corpus_tokens = corpus_words * 100 // 75  # words → tokens (100 words ≈ 133 tokens)
-
-    qs = questions or _SAMPLE_QUESTIONS
-    per_question = []
-    for q in qs:
-        qt = _query_subgraph_tokens(G, q)
-        if qt > 0:
-            per_question.append({"question": q, "query_tokens": qt, "reduction": round(corpus_tokens / qt, 1)})
-
-    if not per_question:
-        return {"error": "No matching nodes found for sample questions. Build the graph first."}
-
-    avg_query_tokens = sum(p["query_tokens"] for p in per_question) // len(per_question)
-    reduction_ratio = round(corpus_tokens / avg_query_tokens, 1) if avg_query_tokens > 0 else 0
-
-    return {
-        "corpus_tokens": corpus_tokens,
-        "corpus_words": corpus_words,
-        "nodes": G.number_of_nodes(),
-        "edges": G.number_of_edges(),
-        "avg_query_tokens": avg_query_tokens,
-        "reduction_ratio": reduction_ratio,
-        "per_question": per_question,
-    }
-
-
-def print_benchmark(result: dict) -> None:
-    """Print a human-readable benchmark report."""
-    if "error" in result:
-        print(f"Benchmark error: {result['error']}")
-        return
-
-    print(f"\ngraphify token reduction benchmark")
-    print(f"{'─' * 50}")
-    print(f"  Corpus:          {result['corpus_words']:,} words → ~{result['corpus_tokens']:,} tokens (naive)")
-    print(f"  Graph:           {result['nodes']:,} nodes, {result['edges']:,} edges")
-    print(f"  Avg query cost:  ~{result['avg_query_tokens']:,} tokens")
-    print(f"  Reduction:       {result['reduction_ratio']}x fewer tokens per query")
-    print(f"\n  Per question:")
-    for p in result["per_question"]:
-        print(f"    [{p['reduction']}x] {p['question'][:55]}")
-    print()