diff --git a/create_parser.py b/create_parser.py index 0e2cd5c..2370850 100644 --- a/create_parser.py +++ b/create_parser.py @@ -39,7 +39,7 @@ def create_parser(): subparsers.add_parser( "add", - help="Add the documents to the databaser", + help="Add the documents to the database", parents=[path_arg_parent] ) diff --git a/main.py b/main.py index da37c64..428ab4a 100644 --- a/main.py +++ b/main.py @@ -32,7 +32,7 @@ def main(): source_path = getattr(args, "path", DEFAULT_SOURCE_PATH) or DEFAULT_SOURCE_PATH documents_path = get_files_in_directory(source_path=source_path) - eval_path = args.eval_file if args.eval_file else DEFAULT_EVAL_PATH + eval_path = getattr(args, "eval_file", DEFAULT_EVAL_PATH) or DEFAULT_EVAL_PATH sample_questions = json.load(open(eval_path, "r")) commands = { diff --git a/src/RAG_pipeline.py b/src/RAG_pipeline.py index d9a16e6..c111899 100644 --- a/src/RAG_pipeline.py +++ b/src/RAG_pipeline.py @@ -23,7 +23,7 @@ class RAGpipeline: def add_documents(self, documents_path : List[str]) -> None: items = self.indexer.index(documents_path) - #print(f"🔍 Adding documents: {', '.join(docu)}") + print(f"🔍 Adding documents: {documents_path}") self.datastore.add_items(items= items) return