Files
task-6a02e23da6fe2e4ac16acf65/init_loader.py
T
2026-05-12 20:01:07 +00:00

11 lines
320 B
Python

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")