ingest.py

from __future__ import annotations

import argparse
from pathlib import Path

from rag_assistant.ingestion.indexer import ingest_path


def main() -> None:
    parser = argparse.ArgumentParser(description="Index a codebase or documentation folder.")
    parser.add_argument("path", type=Path, help="Path to the repo or docs folder to index.")
    parser.add_argument(
        "--reset",
        action="store_true",
        help="Delete and recreate the CodeChunk collection before indexing.",
    )
    args = parser.parse_args()

    file_count, chunk_count = ingest_path(args.path, reset=args.reset)
    print(f"Indexed {chunk_count} chunks from {file_count} files.")


if __name__ == "__main__":
    main()