dnd-helpers/main.py

import argparse

from src.rag.manager import RAGManager


def main():
    parser = argparse.ArgumentParser(description="D&D Helpers CLI")
    parser.add_argument(
        "--ingest-pdf",
        type=str,
        help="Path to a PDF file to ingest into the RAG system",
    )

    args = parser.parse_args()

    if args.ingest_pdf:
        print(f"Ingesting PDF: {args.ingest_pdf}...")
        rag_manager = RAGManager()
        rag_manager.ingest_pdf(args.ingest_pdf)
        print("PDF ingestion complete.")

    print("Hello from dnd-helpers!")


if __name__ == "__main__":
    main()