diff --git a/init_loader.py b/init_loader.py new file mode 100644 index 0000000..52d8233 --- /dev/null +++ b/init_loader.py @@ -0,0 +1,10 @@ +from pathlib import Path +from rag_tools import add_to_knowledge_base + +def load_docs_from_dir(directory: str): + for file in Path(directory).glob("**/*.txt"): + text = file.read_text(encoding="utf-8") + add_to_knowledge_base(text, title=file.name) + +if __name__ == "__main__": + load_docs_from_dir("docs")