Update init_loader.py
This commit is contained in:
+1
-37
@@ -1,37 +1 @@
|
||||
"""
|
||||
Load documents from a directory into the Qdrant knowledge base.
|
||||
"""
|
||||
|
||||
import argparse
|
||||
from pathlib import Path
|
||||
|
||||
from vector_store import add_document
|
||||
|
||||
def load_documents_from_dir(directory: str, title_prefix: str = "") -> None:
|
||||
"""Recursively load all text files from *directory* and add them to the store.
|
||||
|
||||
Parameters
|
||||
----------
|
||||
directory: str
|
||||
Path to the directory containing documents.
|
||||
title_prefix: str, optional
|
||||
Prefix to add to each document title (useful when loading many files).
|
||||
"""
|
||||
dir_path = Path(directory)
|
||||
if not dir_path.is_dir():
|
||||
raise ValueError(f"{directory} is not a valid directory")
|
||||
|
||||
for file_path in dir_path.rglob("*.txt"):
|
||||
with file_path.open("r", encoding="utf-8") as f:
|
||||
content = f.read()
|
||||
title = f"{title_prefix}{file_path.stem}"
|
||||
add_document(content, title)
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="Load documents into the knowledge base")
|
||||
parser.add_argument("directory", help="Directory with .txt files")
|
||||
parser.add_argument("--prefix", default="", help="Optional title prefix")
|
||||
args = parser.parse_args()
|
||||
|
||||
load_documents_from_dir(args.directory, args.prefix)
|
||||
print("Loading completed.")
|
||||
# Legacy file removed
|
||||
Reference in New Issue
Block a user