如何将Flask的FileStorage对象传入LangChain的TextLoader().load()？

TextLoader的.load()方法仅支持文件路径字符串或符合Blob规范的对象，直接传入FileStorage.read()的字节流会触发错误，以下是两种可行的解决办法：

方法一：保存为临时文件后通过路径加载 #

将Flask的FileStorage对象先写入临时文件，再用文件路径传给TextLoader，这是最稳妥的兼容方式：

from langchain.document_loaders import TextLoader
from flask import request
import tempfile
import os

# 获取前端上传的文件
uploaded_file = request.files['file']

# 创建临时文件并写入内容
with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.filename)[1]) as temp_file:
    temp_file.write(uploaded_file.read())
    temp_path = temp_file.name

# 用TextLoader加载临时文件
loader = TextLoader(temp_path)
docs = loader.load()

# 可选：使用后删除临时文件清理资源
os.unlink(temp_path)

方法二：直接处理类文件对象或手动构造Document #

如果不想创建临时文件，可选择以下两种方式：

• 用UnstructuredFileLoader替代TextLoader：它支持直接传入类文件对象，同时会自动识别文件类型：

from langchain.document_loaders import UnstructuredFileLoader
from flask import request

uploaded_file = request.files['file']
uploaded_file.seek(0)  # 重置文件指针到开头
loader = UnstructuredFileLoader(uploaded_file, mode="rb")
docs = loader.load()

• 手动构造Document对象：TextLoader的核心逻辑是读取文件内容并生成Document，可直接模拟这个过程：

from langchain.schema import Document
from flask import request

uploaded_file = request.files['file']
# 读取内容时注意匹配文件实际编码，比如gbk、utf-8等
content = uploaded_file.read().decode('utf-8')
# 生成和TextLoader.load()格式一致的结果
docs = [Document(page_content=content, metadata={"source": uploaded_file.filename})]

内容的提问来源于stack exchange，提问作者wafflehouse67