Function afml_get_context

scripts/afml_mcp_server.py:249–281

(
    query: str,
    top_k: int = 4,
    window: int = 1,
    chapter: str = "",
    section: str = "",
    semantic_weight: float = 0.86,
)

    ),
)
def afml_get_context(
    query: str,
    top_k: int = 4,
    window: int = 1,
    chapter: str = "",
    section: str = "",
    semantic_weight: float = 0.86,
) -> dict[str, Any]:
    if window < 0:
        window = 0
    if window > 5:
        window = 5

    hits = _score_results(
        query=query,
        top_k=top_k,
        chapter=chapter,
        section=section,
        semantic_weight=semantic_weight,
    )

    contexts: list[dict[str, Any]] = []
    for hit in hits:
        bundle = _chunk_with_neighbors(hit["chunk_id"], window=window)
        contexts.append({"hit": hit, "context": bundle})

    return {
        "query": query,
        "top_k": top_k,
        "window": window,
        "filters": {"chapter": chapter, "section": section},
        "contexts": contexts,
    }


if __name__ == "__main__":