diff --git a/main.py b/main.py new file mode 100644 index 0000000..dea4247 --- /dev/null +++ b/main.py @@ -0,0 +1,24 @@ +"""Entry point for the RAG agent. + +The script loads documents from a directory, initializes the knowledge base, and +provides a simple CLI for interacting with the agent. +""" + +from __future__ import annotations + +import argparse +from pathlib import Path + +from src.cli import run_cli + +if __name__ == "__main__": + parser = argparse.ArgumentParser(description="RAG agent CLI") + parser.add_argument( + "--docs", + type=str, + default="docs", + help="Directory with documents to load into the knowledge base.", + ) + args = parser.parse_args() + + run_cli(args.docs) \ No newline at end of file