文件读写#

RAG、Prompt 工具箱、项目日志整理都离不开文件读写。你需要能稳定读取 Markdown、保存模型输出、遍历文档目录，并处理编码问题。

使用 pathlib#

推荐使用 pathlib：

1
from pathlib import Path
2

3
path = Path("docs/note.md")
4
text = path.read_text(encoding="utf-8")
5
print(text)

相比字符串路径，Path 更适合跨平台处理。

写入文件#

1
from pathlib import Path
2

3
output = Path("outputs/answer.md")
4
output.parent.mkdir(parents=True, exist_ok=True)
5
output.write_text("# 模型回答\n\n这里是回答内容。", encoding="utf-8")

注意先创建目录，否则写入会失败。

遍历目录#

1
from pathlib import Path
2

3
docs_dir = Path("docs")
4

5
for path in docs_dir.glob("*.md"):
6
    text = path.read_text(encoding="utf-8")
7
    print(path.name, len(text))

如果要递归读取子目录：

1
for path in docs_dir.rglob("*.md"):
2
    print(path)

RAG 中的文件记录#

读取文档时要保留来源：

1
def load_documents(folder: str):
2
    records = []
3
    for path in Path(folder).rglob("*.md"):
4
        records.append({
5
            "source": str(path),
6
            "content": path.read_text(encoding="utf-8"),
7
        })
8
    return records