Function cmd_evidence

scripts/afml_docs_loop.py:392–525

(args: argparse.Namespace)

def cmd_evidence(args: argparse.Namespace) -> int:
    state = read_state()
    evidence_chapters: list[dict[str, Any]] = []
    processed = 0

    for chapter in state["chapters"]:
        chapter_name = chapter["chapter"]
        chapter_query = chapter.get("query") or chapter.get("theme") or chapter_name
        chapter_search = afml_search(
            query=chapter_query,
            chapter=chapter_name,
            top_k=args.top_k,
            include_text=False,
        )
        chapter_hits = filter_hits_to_chapter(chapter_search.get("results", []), chapter_name)
        chapter_anchor = summarize_hit(chapter_hits[0]) if chapter_hits else chapter.get("afml_anchor", {})
        chapter["afml_anchor"] = chapter_anchor

        chapter_evidence: dict[str, Any] = {
            "chapter": chapter_name,
            "query": chapter_query,
            "anchor": chapter_anchor,
            "sections": [],
        }

        for section in chapter["sections"]:
            if args.only_pending and section["status"] != "pending":
                continue
            processed += 1

            module = section["module"]
            module_query = MODULE_QUERIES.get(module, module.replace("_", " "))
            search_out = afml_search(
                query=module_query,
                chapter=chapter_name,
                top_k=args.top_k,
                include_text=False,
            )
            search_hits = filter_hits_to_chapter(search_out.get("results", []), chapter_name)

            ctx_out = afml_get_context(
                query=module_query,
                chapter=chapter_name,
                top_k=args.top_k_context,
                window=args.window,
            )
            contexts = [
                c for c in ctx_out.get("contexts", []) if chapter_eq(str(c.get("hit", {}).get("chapter", "")), chapter_name)
            ]

            primary_hit: dict[str, Any] = {}
            if contexts:
                primary_hit = contexts[0].get("hit", {})
            elif search_hits:
                primary_hit = search_hits[0]

            if primary_hit:
                section["afml_anchor"] = summarize_hit(primary_hit)