Function _load_store

scripts/afml_mcp_server.py:51–74

()

def _load_store() -> IndexStore:
    global _STORE
    if _STORE is not None:
        return _STORE

    index_dir = _get_index_dir()
    manifest = json.loads((index_dir / "manifest.json").read_text(encoding="utf-8"))

    chunks: list[dict[str, Any]] = []
    with (index_dir / "chunks.jsonl").open("r", encoding="utf-8") as f:
        for line in f:
            chunks.append(json.loads(line))

    embeddings = np.load(index_dir / "embeddings.npy")
    chunk_id_to_idx = {c["chunk_id"]: i for i, c in enumerate(chunks)}

    _STORE = IndexStore(
        index_dir=index_dir,
        chunks=chunks,
        embeddings=embeddings,
        manifest=manifest,
        chunk_id_to_idx=chunk_id_to_idx,
    )
    return _STORE


def _load_model(model_name: str):