本教程旨在解决langchain中`textloader`和`charactertextsplitter`在处理多个文本文件及大型文本块时遇到的常见问题，如仅处理首个文档、分割失效及chunk大小异常。我们将详细介绍如何利用`recursivecharactertextsplitter`实现智能文本分割，并构建一个支持批量加载多类型文档的解决方案，最终将处理后的文本高效、可靠地持久化至chromadb向量数据库，确保llm能准确检索所需信息。
在构建基于大型语言模型（LLM）的检索增强生成（RAG）系统时，准确高效地加载、分割和存储文档是至关重要的一步。然而，开发者在使用LangChain的TextLoader和CharacterTextSplitter时，常会遇到一些挑战，例如系统仅处理目录中的第一个文档、文本块（chunk）大小远超预期、以及后续文档未能被正确分割和存储，导致LLM无法检索到这些信息。
在使用LangChain处理本地文档时，如果代码逻辑未能正确迭代处理所有文件，TextLoader默认可能只加载指定路径的单个文件。例如，以下代码片段在处理多个文件时，通常只会加载./folder/file.txt这一个文件，而忽略同目录下的其他文件。
loader = TextLoader("./folder/file.txt") # 明确指向单个文件
documents = loader.load()
text_splitter = CharacterTextSplitter(chunk_size=300, chunk_overlap=0)
# texts = text_splitter.split_documents(documents) # 假设这里有split操作
chromaDirectory = "./folder/chroma_db"
# Chroma.from_documents(texts, embeddings, persist_directory=chromaDirectory)此外，CharacterTextSplitter在面对非常大的文本块时，可能会出现分割异常，例如即便设置了chunk_size=300，也可能生成远超此限制的文本块，甚至在处理后续文本时完全失效，不再进行分割。这通常是由于其基于简单字符分割的机制，对于结构复杂的文档或超长无分隔符的文本段落表现不佳。当这些未正确分割的文本被存储到向量数据库（如ChromaDB）中时，LLM在检索时自然无法找到相关信息，因为其上下文窗口和检索机制依赖于合理大小的文本块。
为了克服上述挑战，我们需要一套更健壮的文档加载和文本分割策略。核心解决方案包括：
我们将通过以下步骤，构建一个能够高效处理多文档、智能分割文本并持久化到ChromaDB的完整流程。
首先，定义一个映射表，用于支持不同文件类型的加载器。这使得我们的系统更具扩展性，可以轻松添加对.pdf、.docx等其他文件类型的支持。
import os
import glob
from typing import List
from langchain.docstore.document import Document
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.document_loaders import TextLoader
from langchain_community.vectorstores import Chroma # 使用 langchain_community 替代旧的 Chroma 导入
from langchain_openai import OpenAIEmbeddings # 假设使用OpenAI的嵌入模型
from chromadb.config import Settings
# 定义支持的文档加载器映射
DOC_LOADERS_MAPPING = {
".txt": (TextLoader, {"encoding": "utf8"}),
# 可以根据需要添加更多文档加载器，例如：
# ".pdf": (PyPDFLoader, {}),
# ".docx": (Docx2txtLoader, {}),
}
def load_document(path: str) -> Document:
"""
加载单个文档。
"""
try:
ext = "." + path.rsplit(".", 1)[-1]
if ext in DOC_LOADERS_MAPPING:
loader_class, loader_args = DOC_LOADERS_MAPPING[ext]
loader = loader_class(path, **loader_args)
# load() 方法返回一个 Document 列表，我们通常只取第一个
return loader.load()[0]
raise ValueError(f"不支持的文件扩展名: {ext}")
except Exception as exception:
raise ValueError(f"加载文档时发生错误 '{path}': {exception}")接着，实现一个函数来遍历指定目录及其子目录，查找所有支持的文件类型，并使用load_document函数批量加载它们。
def load_documents_from_dir(path: str) -> List[Document]:
"""
从指定目录加载所有支持的文档。
"""
try:
all_files = []
for ext in DOC_LOADERS_MAPPING:
# 递归查找目录中所有匹配扩展名的文件
all_files.extend(
glob.glob(os.path.join(path, f"**/*{ext}"), recursive=True)
)
# 批量加载文件
return [load_document(file_path) for file_path in all_files]
except Exception as exception:
raise RuntimeError(f"加载文件时发生错误: {exception}")现在，我们使用RecursiveCharacterTextSplitter来对加载的文档进行智能分割。它会尝试不同的分隔符（如\n\n, \n, `,.等），直到文本块大小符合预期，这比CharacterTextSplitter`更灵活和鲁棒。
# 加载所有文档
documents = load_documents_from_dir("./folder/")
# 初始化RecursiveCharacterTextSplitter
# chunk_size: 每个文本块的最大长度
# chunk_overlap: 相邻文本块之间的重叠字符数，有助于保持上下文连贯性
text_splitter = RecursiveCharacterTextSplitter(
chunk_size=300,
chunk_overlap=50
)
# 分割文档
texts = text_splitter.split_documents(documents)最后一步是将分割后的文本块及其对应的嵌入（embeddings）存储到ChromaDB中。确保ChromaDB的persist_directory设置正确，并且通过client_settings明确指定持久化选项，以保证数据在程序关闭后不会丢失。
# 初始化嵌入模型，例如OpenAIEmbeddings
# 请确保已设置OPENAI_API_KEY环境变量
embeddings = OpenAIEmbeddings()
chroma_db_path = "./folder/chroma_db"
# 初始化ChromaDB并持久化
chroma_db = Chroma.from_documents(
texts,
embeddings,
persist_directory=chroma_db_path,
client_settings= Settings(
persist_directory=chroma_db_path,
chroma_db_impl="duckdb+parquet", # 指定ChromaDB的实现方式，确保持久化
anonymized_telemetry=False, # 关闭匿名遥测
),
)
# 显式调用persist()方法确保数据写入磁盘
chroma_db.persist()
# 清除内存中的ChromaDB实例（可选，但有助于释放资源）
chroma_db = None将上述所有组件整合，形成一个完整的文档处理和ChromaDB持久化流程。
import os
import glob
from typing import List
from langchain.docstore.document import Document
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.document_loaders import TextLoader
from langchain_community.vectorstores import Chroma
from langchain_openai import OpenAIEmbeddings # 假设使用OpenAI的嵌入模型
from chromadb.config import Settings
# --- 1. 定义支持的文档加载器映射 ---
DOC_LOADERS_MAPPING = {
".txt": (TextLoader, {"encoding": "utf8"}),
# 可以根据需要添加更多文档加载器
# ".pdf": (PyPDFLoader, {}),
# ".docx": (Docx2txtLoader, {}),
}
# --- 2. 加载单个文档函数 ---
def load_document(path: str) -> Document:
"""
加载单个文档。
"""
try:
ext = "." + path.rsplit(".", 1)[-1]
if ext in DOC_LOADERS_MAPPING:
loader_class, loader_args = DOC_LOADERS_MAPPING[ext]
loader = loader_class(path, **loader_args)
return loader.load()[0]
raise ValueError(f"不支持的文件扩展名: {ext}")
except Exception as exception:
raise ValueError(f"加载文档时发生错误 '{path}': {exception}")
# --- 3. 批量加载目录文档函数 ---
def load_documents_from_dir(path: str) -> List[Document]:
"""
从指定目录加载所有支持的文档。
"""
try:
all_files = []
for ext in DOC_LOADERS_MAPPING:
all_files.extend(
glob.glob(os.path.join(path, f"**/*{ext}"), recursive=True)
)
return [load_document(file_path) for file_path in all_files]
except Exception as exception:
raise RuntimeError(f"加载文件时发生错误: {exception}")
# --- 主执行流程 ---
if __name__ == "__main__":
# 确保存在一个名为 'folder' 的目录，并在其中放置一些 .txt 文件进行测试
# 例如：
# ./folder/doc1.txt
# ./folder/doc2.txt
# ...
# 1. 设置文档目录和ChromaDB持久化目录
source_directory = "./folder/"
chroma_db_path = "./folder/chroma_db"
# 确保ChromaDB目录存在
os.makedirs(chroma_db_path, exist_ok=True)
# 2. 批量加载文档
print(f"正在从目录 '{source_directory}' 加载文档...")
documents = load_documents_from_dir(source_directory)
print(f"共加载了 {len(documents)} 个文档。")
if not documents:
print("未找到任何文档，请检查目录和文件。")
else:
# 3. 初始化文本分割器
print("正在初始化文本分割器...")
text_splitter = RecursiveCharacterTextSplitter(
chunk_size=300,
chunk_overlap=50
)
# 4. 分割文档
print("正在分割文档...")
texts = text_splitter.split_documents(documents)
print(f"文档被分割成 {len(texts)} 个文本块。")
# 5. 初始化嵌入模型
# 请确保已设置OPENAI_API_KEY环境变量
# 或者使用其他本地嵌入模型，例如 SentenceTransformers
print("正在初始化嵌入模型...")
try:
embeddings = OpenAIEmbeddings()
except Exception as e:
print(f"初始化OpenAIEmbeddings失败，请检查OPENAI_API_KEY：{e}")
print("尝试使用其他嵌入模型或退出。")
exit() # 或者选择使用其他嵌入模型
# 6. 持久化到ChromaDB
print(f"正在将文本块及嵌入持久化到ChromaDB，路径：'{chroma_db_path}'...")
chroma_db = Chroma.from_documents(
texts,
embeddings,
persist_directory=chroma_db_path,
client_settings= Settings(
persist_directory=chroma_db_path,
chroma_db_impl="duckdb+parquet",
anonymized_telemetry=False,
),
)
chroma_db.persist()
print("ChromaDB数据已成功持久化。")
# 7. 验证（可选）：加载并查询ChromaDB
print("正在加载ChromaDB并进行简单查询验证...")
loaded_db = Chroma(
persist_directory=chroma_db_path,
embedding_function=embeddings,
client_settings= Settings(
persist_directory=chroma_db_path,
chroma_db_impl="duckdb+parquet",
anonymized_telemetry=False,
),
)
# 尝试查询一个与文档内容相关的短语
query = "关于文档内容的关键信息" # 根据你的文档内容修改查询
results = loaded_db.similarity_search(query, k=2)
print(f"查询 '{query}' 的结果:")
for i, doc in enumerate(results):
print(f"--- 结果 {i+1} ---")
print(f"内容: {doc.page_content[:100]}...") # 打印前100字符
print(f"元数据: {doc.metadata}")
print("-" * 20)
print("文档处理和ChromaDB持久化流程完成。")通过本教程，我们解决了LangChain在处理多文档和文本分割时遇到的常见问题。通过采用RecursiveCharacterTextSplitter进行智能文本分割，并构建一个支持批量加载多类型文档的健壮流程，我们能够确保所有文档都被正确处理，并高效、可靠地持久化到ChromaDB。这一优化方案将显著提升基于LLM的RAG系统的检索准确性和整体性能，使得LLM能够从你提供的所有信息中有效地学习和回答问题。
以上就是LangChain多文档处理与ChromaDB持久化：解决文本加载与分割挑战的详细内容，更多请关注php中文网其它相关文章！
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移，垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是，许多工具可以让 Windows 保持平稳运行。
广告Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
970
收藏
