Add main.py
This commit is contained in:
@@ -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)
|
||||||
Reference in New Issue
Block a user