add init_loader

This commit is contained in:
2026-05-12 20:01:07 +00:00
parent 70b8d4aed4
commit b08f1d0e26
+10
View File
@@ -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")