diff --git a/generic_rag/app.py b/generic_rag/app.py index c09c5e8..d11f996 100644 --- a/generic_rag/app.py +++ b/generic_rag/app.py @@ -36,6 +36,7 @@ parser.add_argument( "If the path is relative it will be relative to the current working directory.", ) parser.add_argument( + "-u", "--unstructured-pdf", action="store_true", help="Use an unstructered PDF loader. " @@ -53,7 +54,7 @@ parser.add_argument( ) parser.add_argument("--web-chunk-size", type=int, default=200, help="The size of the chunks to split the text into.") parser.add_argument( - "-c", + "-d", "--chroma-db-location", type=Path, default=Path(".chroma_db"),