37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""
|
|
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.") |