default main cli to openai add data dict as a args

This commit is contained in:
yichuan520030910320
2025-07-22 21:56:30 -07:00
parent 2ebb29de65
commit 0544f96b79

View File

@@ -20,7 +20,7 @@ async def main(args):
print("Loading documents...")
documents = SimpleDirectoryReader(
"examples/data",
args.data_dir,
recursive=True,
encoding="utf-8",
required_exts=[".pdf", ".txt", ".md"],
@@ -104,6 +104,12 @@ if __name__ == "__main__":
default="./test_doc_files",
help="Directory where the Leann index will be stored.",
)
parser.add_argument(
"--data-dir",
type=str,
default="examples/data",
help="Directory containing documents to index (PDF, TXT, MD files).",
)
args = parser.parse_args()
asyncio.run(main(args))