乐于分享
好东西不私藏

第6篇:文档加载与分割——Go处理你的资料

第6篇:文档加载与分割——Go处理你的资料

📝作者说:上篇讲了Embedding,但还有一个关键问题没解决——100万字的文章怎么切成小块喂给RAG? 切不好,检索质量直接崩。这篇把分割策略讲透,顺便把成本问题说清楚。

你是否也这样处理过文档?

上篇之后,有读者问:

"博主,我把整篇文章向量化了,为什么检索质量很差?"

"我按固定字数切,但搜出来的东西牛头不对马嘴。"

然后我看了他的代码:

// 他的切分方式funcSplitText(text string) []string {    var chunks []string    for i := 0; i < len(text); i += 500 {        chunks = append(chunks, text[i:i+500])    }    return chunks}

问题在哪?

❌ 按固定字数切:500个字,正好切在句子中间❌ "Go语言的并发编程" → 切成了 "Go语言的并发" + "编程"→ 检索"并发" → 能匹配→ 检索"Go并发编程" → 匹配不到,因为向量被截断了

一、Go生态文档处理:PDF / HTML / Markdown / TXT

1.1 四种格式的处理方案

格式

推荐库

难度

特点

TXT

os.ReadFile

直接读,最简单

Markdown

blackfriday

⭐⭐

保留结构,去标记

HTML

goquery

⭐⭐⭐

去掉标签,取正文

PDF

uni-docviewer

⭐⭐⭐⭐

中文支持差,容易乱码

1.2 TXT处理(最简单)

import "os"funcReadTxt(path string) (stringerror) {    data, err := os.ReadFile(path)    if err != nil {        return "", err    }    return string(data), nil}

1.3 Markdown处理(保留结构)

import (    "github.com/russross/blackfriday/v2")funcReadMarkdown(path string) (stringerror) {    data, err := os.ReadFile(path)    if err != nil {        return "", err    }    // 用blackfriday解析Markdown    html := blackfriday.Run(data)    // 用goquery提取纯文本(去掉HTML标签)    doc, _ := goquery.NewDocumentFromReader(bytes.NewReader(html))    text := doc.Text()    return text, nil}

1.4 HTML处理(去掉标签取正文)

import "github.com/PuerkitoBio/goquery"funcReadHTML(path string) (stringerror) {    data, err := os.ReadFile(path)    if err != nil {        return "", err    }    doc, err := goquery.NewDocumentFromReader(bytes.NewReader(data))    if err != nil {        return "", err    }    // 去掉script和style(这些不是正文)    doc.Find("script, style").Remove()    // 提取纯文本    text := doc.Text()    return text, nil}

1.5 PDF处理(最麻烦)

import "github.com/ledongthuc/pdf"funcReadPDF(path string) (stringerror) {    f, r, err := pdf.Open(path)    if err != nil {        return "", err    }    defer f.Close()    var text string    totalPages := r.NumPage()    for i := 1; i <= totalPages; i++ {        p := r.Page(i)        if p.V.IsNull() {            continue        }        content, _ := p.GetPlainText()        text += content + "\n"    }    return text, nil}

⚠️ PDF中文问题:Go的PDF库对中文支持普遍不好。实战建议先把PDF转成TXT/Markdown再用Go处理,或者用Python的PyPDF2处理完再传给Go。

二、文本分割策略:三种切法对比

2.1 按字符数分割(最简单,效果最差)

funcSplitByChars(text string, chunkSize int) []string {    var chunks []string    runes := []rune(text)    for i := 0; i < len(runes); i += chunkSize {        end := i + chunkSize        if end > len(runes) {            end = len(runes)        }        chunks = append(chunks, string(runes[i:end]))    }    return chunks}

问题:

❌ 可能在句子中间切断❌ 上下文被割裂,检索"完整语义"时丢失❌ "Go语言的并发编程" → "Go语言的并发" + "编程"

2.2 按句子分割(推荐,最常用)

// 第6篇:文本分割(使用langchaingo textsplitter)//// 运行前提:无外部服务依赖//   - go run ./article06/splitter/package mainimport (	"context"	"fmt"	"strings"	"unicode/utf8"	"github.com/tmc/langchaingo/documentloaders"	"github.com/tmc/langchaingo/schema"	"github.com/tmc/langchaingo/textsplitter")funcmain() {	ctx := context.Background()	text := `Go语言是一门静态类型的编译型语言,由Google于2009年发布。它具有高并发、高性能的特点,特别适合构建后端服务和分布式系统。Goroutine是Go语言的核心特性之一,它是一种轻量级的线程,由Go运行时管理。创建一个Goroutine只需要在函数调用前加上go关键字。Channel是Go语言中用于Goroutine之间通信的机制。通过Channel,可以实现安全的跨Goroutine数据传递,避免共享内存带来的并发问题。Go语言的错误处理采用显式返回错误的方式,而非异常机制。这虽然让代码稍微冗长,但让错误处理逻辑更加清晰。Go的标准库非常强大,包含了HTTP服务器、JSON处理、加密、测试等常用功能,减少了对外部依赖的需求。`	fmt.Println("=== 第6篇:文本分割(langchaingo textsplitter) ===")	fmt.Printf("原文长度: %d 字符\n\n", utf8.RuneCountInString(text))	// ===== 1. RecursiveCharacter 分割器(推荐) =====	fmt.Println("--- 1. RecursiveCharacter 分割器(推荐) ---")	// langchaingo 的 RecursiveCharacter 是最常用的分割器	// 默认分隔符: ["\n\n", "\n", " ", ""]	splitter := textsplitter.NewRecursiveCharacter(		textsplitter.WithChunkSize(100),		textsplitter.WithChunkOverlap(20),		textsplitter.WithSeparators([]string{"\n""。"","" """}),	)	chunks, err := splitter.SplitText(text)	if err != nil {		fmt.Printf("分割失败: %v\n", err)		return	}	for i, chunk := range chunks {		fmt.Printf("  Chunk %d (%d字符): %s\n", i+1, utf8.RuneCountInString(chunk), truncate(chunk, 50))	}	// ===== 2. 对比:不同ChunkSize =====	fmt.Println()	fmt.Println("--- 2. ChunkSize 对比 ---")	for _, size := range []int{50100200} {		s := textsplitter.NewRecursiveCharacter(			textsplitter.WithChunkSize(size),			textsplitter.WithChunkOverlap(10),		)		parts, _ := s.SplitText(text)		fmt.Printf("  ChunkSize=%d → %d 个分片\n", size, len(parts))	}	// ===== 3. 与文档加载器结合使用(LoadAndSplit) =====	fmt.Println()	fmt.Println("--- 3. LoadAndSplit: 加载+分割一体化 ---")	// 创建测试文件	testContent := strings.Repeat("Go语言是一门编译型语言。"20)	docs, err := documentloaders.NewText(		strings.NewReader(testContent),	).LoadAndSplit(ctx,		textsplitter.NewRecursiveCharacter(			textsplitter.WithChunkSize(50),			textsplitter.WithChunkOverlap(10),		),	)	if err != nil {		fmt.Printf("LoadAndSplit失败: %v\n", err)else {		fmt.Printf("  加载+分割得到 %d 个文档\n"len(docs))		for i, doc := range docs {			if i >= 3 {				fmt.Printf("  ... 共%d个\n"len(docs))				break			}			fmt.Printf("  Doc %d (%d字符): %s\n", i+1, utf8.RuneCountInString(doc.PageContent), truncate(doc.PageContent, 40))		}	}	// ===== 4. 自定义分割(中文按句子) =====	fmt.Println()	fmt.Println("--- 4. 自定义分割器(实现TextSplitter接口) ---")	sentenceSplitter := &SentenceSplitter{MaxChars: 150}	sentenceChunks, err := sentenceSplitter.SplitText(text)	if err != nil {		fmt.Printf("分割失败: %v\n", err)	}	for i, chunk := range sentenceChunks {		fmt.Printf("  Chunk %d (%d字符): %s\n", i+1, utf8.RuneCountInString(chunk), truncate(chunk, 50))	}	// ===== 5. SplitDocuments 批量分割 =====	fmt.Println()	fmt.Println("--- 5. SplitDocuments 批量分割 ---")	inputDocs := []schema.Document{		{PageContent: strings.Repeat("Go语言并发编程。"15), Metadata: map[string]any{"source""article1"}},		{PageContent: strings.Repeat("RAG知识库搭建。"15), Metadata: map[string]any{"source""article2"}},	}	splitDocs, err := textsplitter.SplitDocuments(		textsplitter.NewRecursiveCharacter(			textsplitter.WithChunkSize(80),			textsplitter.WithChunkOverlap(10),		),		inputDocs,	)	if err != nil {		fmt.Printf("批量分割失败: %v\n", err)else {		fmt.Printf("  %d个文档 → 分割为 %d 个片段\n"len(inputDocs), len(splitDocs))		for _, doc := range splitDocs {			fmt.Printf("  [source=%v] %s\n", doc.Metadata["source"], truncate(doc.PageContent, 40))		}	}}// SentenceSplitter 自定义按句子分割器(实现 textsplitter.TextSplitter 接口)type SentenceSplitter struct {	MaxChars int}func(s *SentenceSplitter) SplitText(text string) ([]stringerror) {	var chunks []string	var current string	sentences := splitChineseSentences(text)	for _, sentence := range sentences {		sentence = strings.TrimSpace(sentence)		if sentence == "" {			continue		}		if utf8.RuneCountInString(current)+utf8.RuneCountInString(sentence) <= s.MaxChars {			current += sentenceelse {			if current != "" {				chunks = append(chunks, current)			}			current = sentence		}	}	if current != "" {		chunks = append(chunks, current)	}	return chunks, nil}funcsplitChineseSentences(text string) []string {	var sentences []string	var current string	for _, r := range text {		current += string(r)		if r == '。' || r == '!' || r == '?' || r == '.' || r == '!' || r == '?' {			sentences = append(sentences, current)			current = ""		}	}	if current != "" {		sentences = append(sentences, current)	}	return sentences}functruncate(s string, n int) string {	runes := []rune(s)	if len(runes) <= n {		return s	}	return string(runes[:n]) + "..."}

优点:

✅ 按语义单元切,不会切断句子✅ 上下文完整✅ 实现简单,效果好

2.3 按语义分割(最精准,效果最好,但最复杂)

funcSplitBySemantic(text string) ([]stringerror) {    // 1. 先按句子切分    sentences := SplitBySentence(text, 200)    // 2. 相邻句子做Embedding,计算相似度    var chunks []string    var currentChunk []string    for i := 0; i < len(sentences); i++ {        emb1, _ := EmbedText(strings.Join(currentChunk, ""))        if len(currentChunk) > 0 {            // 计算当前chunk和下一句的相似度            nextEmb, _ := EmbedText(sentences[i])            sim := Cosine(emb1, nextEmb)            // 3. 如果相似度高(>0.9),合并;否则断开            if sim > 0.9 && len(strings.Join(currentChunk, "")) < 1000 {                currentChunk = append(currentChunk, sentences[i])            } else {                chunks = append(chunks, strings.Join(currentChunk, ""))                currentChunk = []string{sentences[i]}            }        } else {            currentChunk = []string{sentences[i]}        }    }    if len(currentChunk) > 0 {        chunks = append(chunks, strings.Join(currentChunk, ""))    }    return chunks, nil

原理

相邻两段话 → 计算相似度→ 相似度高 → 合并(因为说的是同一件事)→ 相似度低 → 断开(因为话题切换了)

2.4 三种策略对比

策略

效果

速度

实现难度

推荐度

按字符

⭐⭐⭐⭐⭐

❌ 不推荐

按句子

⭐⭐⭐⭐

⭐⭐⭐

⭐⭐

✅ 推荐

按语义

⭐⭐⭐⭐⭐

⭐⭐

⭐⭐⭐⭐

✅ 生产级推荐

三、分割粒度对检索效果的影响(核心问题)

3.1 太短的问题

Chunk = 50字→ "Go语言的并发编程"→ 切成:["Go语言的并发""编程"]检索"Go并发"  → 可能匹配到"Go语言的并发" ✅  → 但丢失了"编程"这个上下文问题:  → 上下文碎片化  → LLM看到的是残缺的句子  → 生成的回答质量差

3.2 太长的问题

Chunk = 5000字→ 整篇文章作为一个Chunk→ 文章里混杂了:  → 开头:引言(100字)  → 中间:正文(4600字)  → 结尾:广告(300字)检索"Go并发怎么学"  → 检索结果包含了整篇文章  → LLM的Context里全是噪声  → 生成的回答被广告污染

3.3 黄金粒度选择

根据场景选择:

场景

推荐Chunk

原因

短问答/FAQ

128-256字

问题短,相关段落也短

技术文章/博客

512-1024字

一个知识点一段

长文档/书籍

1024-2048字

需要更多上下文

博客CMS推荐:

大多数文章 → 512字/Chunk  → 能覆盖一个完整段落  → 不会太短丢失上下文  → 不会太长引入噪声

四、【实战】博客文章入库RAG完整流程

4.1 整体流程

编辑发布文章     ↓提取内容(标题/正文/标签/发布时间)     ↓文本清洗(去HTML标签/特殊字符)     ↓语义分割(按句子切,512字/Chunk)     ↓逐段Embedding(调用Embedding API)     ↓存入向量数据库(Chroma/Milvus)     ↓完成

4.2 完整代码

// 第6篇:文档加载(使用langchaingo documentloaders)//// 运行前提:无外部服务依赖//   - go run ./article06/loader/package mainimport (	"bytes"	"context"	"fmt"	"log"	"os"	"strings"	"github.com/tmc/langchaingo/documentloaders"	"github.com/tmc/langchaingo/schema")funcmain() {	ctx := context.Background()	fmt.Println("=== 第6篇:文档加载演示(langchaingo documentloaders) ===")	// 创建测试文件	os.MkdirAll("testdata"0755)	os.WriteFile("testdata/test.txt", []byte("Go语言是一门静态类型的编译型语言,由Google开发。\n它具有高并发、高性能的特点。"), 0644)	os.WriteFile("testdata/test.html", []byte("<html><head><title>测试</title></head><body><h1>Go语言</h1><p>Go是一门高性能的编程语言。</p><p>它支持并发编程。</p></body></html>"), 0644)	os.WriteFile("testdata/test.csv", []byte("title,tag,views\nGo并发编程,Go,12580\nRAG知识库,AI,8960\nK8s部署,DevOps,15230"), 0644)	defer os.RemoveAll("testdata")	// ===== 1. TXT 文本加载 =====	fmt.Println("\n--- 1. Text Loader(纯文本) ---")	txtFile, err := os.Open("testdata/test.txt")	if err != nil {		log.Fatal(err)	}	defer txtFile.Close()	txtLoader := documentloaders.NewText(txtFile)	txtDocs, err := txtLoader.Load(ctx)	if err != nil {		log.Fatal(err)	}	for _, doc := range txtDocs {		fmt.Printf("  内容: %s\n", truncate(doc.PageContent, 80))		fmt.Printf("  元数据: %v\n", doc.Metadata)	}	// ===== 2. HTML 文档加载 =====	fmt.Println("\n--- 2. HTML Loader(HTML解析+清洗) ---")	htmlContent, err := os.ReadFile("testdata/test.html")	if err != nil {		log.Fatal(err)	}	htmlLoader := documentloaders.NewHTML(bytes.NewReader(htmlContent))	htmlDocs, err := htmlLoader.Load(ctx)	if err != nil {		log.Fatal(err)	}	for _, doc := range htmlDocs {		fmt.Printf("  内容: %s\n", truncate(doc.PageContent, 80))	}	// ===== 3. CSV 文档加载 =====	fmt.Println("\n--- 3. CSV Loader(表格数据) ---")	csvFile, err := os.Open("testdata/test.csv")	if err != nil {		log.Fatal(err)	}	defer csvFile.Close()	csvLoader := documentloaders.NewCSV(csvFile)	csvDocs, err := csvLoader.Load(ctx)	if err != nil {		log.Fatal(err)	}	for _, doc := range csvDocs {		fmt.Printf("  行内容: %s\n", truncate(doc.PageContent, 60))		fmt.Printf("  元数据: %v\n", doc.Metadata)	}	// ===== 4. 自定义Loader(模拟Markdown) =====	fmt.Println("\n--- 4. 自定义 Loader(Markdown演示) ---")	mdContent := "# Go并发编程\n\n## Goroutine\n\nGoroutine是Go语言的**轻量级线程**。\n\n## Channel\n\nChannel用于Goroutine间通信。"	mdDocs := loadMarkdown(ctx, mdContent)	for _, doc := range mdDocs {		fmt.Printf("  内容: %s\n", truncate(doc.PageContent, 80))	}	// ===== 5. 批量加载目录 =====	fmt.Println("\n--- 5. 批量加载演示 ---")	fmt.Println("  documentloaders.NewDirectory(path, suffix...) 可批量加载目录文件")	fmt.Println("  例如:NewDirectory(\"./docs\", \".txt\", \".md\")")}// loadMarkdown 自定义Markdown加载(langchaingo暂无内置Markdown loader,// 可使用 RecursiveCharacter splitter 或自定义实现)funcloadMarkdown(ctx context.Context, content string) []schema.Document {	text := content	// 简化的Markdown清洗	for _, pair := range []struct{ from, to string }{		{"# """}, {"## """}, {"**"""}, {"`"""},	} {		text = strings.ReplaceAll(text, pair.from, pair.to)	}	return []schema.Document{		{			PageContent: strings.TrimSpace(text),			Metadata:    map[string]any{"format""markdown"},		},	}}functruncate(s string, n int) string {	runes := []rune(s)	if len(runes) <= n {		return s	}	return string(runes[:n]) + "..."

4.3 读者搜索流程

读者输入问题     ↓问题Embedding     ↓向量数据库检索(Top5)     ↓组装Context     ↓LLM生成回答     ↓返回给读者     ↓附带原文链接(可跳转

五、【核心】Context长度 = Token成本 = 钱

5.1 成本计算

这是最容易忽视的成本。

Chunk大小的选择直接影响LLM调用成本:Chunk = 512字:  → 5个Chunk = 2560字 = 约2560 Token(按1:1估算)  → 每次搜索的LLM调用成本 = 2560 Token × 单价Chunk = 1024字:  → 5个Chunk = 5120字 = 约5120 Token(按1:1估算)  → 每次搜索的LLM调用成本 = 5120 Token × 单价差距:2倍!❗️ Chunk越大 → Context越长 → Token越多 → 费用越高

5.2 成本控制策略

策略一:控制Chunk数量(TopK)  Top5 → Top3:省40% Context  → 但可能漏掉相关结果策略二:控制Chunk大小  1024字 → 512字:省50% Context  → 但上下文变少策略三:Context压缩(进阶)  → 先用简单模型总结Chunk  → 再用主模型生成

5.3 最佳实践

推荐配置(博客CMS):  Chunk大小:512  TopK:5  Context上限:2560字(≈主模型单次Context,按1:1估算)超出上限时:  → 按相关性排序,只取最相关的  → 或用Context压缩技术

六、【踩坑】文档处理的5个高频坑

6.1 PDF中文乱码

// ❌ 错误:用Go原生PDF库处理中文text, _ := pdf.GetPlainText(page)// → 中文全是方块 □□□□□// ✅ 正确:用Python转成TXT,再给Go处理// Python: PyPDF2 + 字体映射表// 或:用云服务(阿里云OCR)转写

6.2 Markdown格式丢失

// ❌ 错误:直接读Markdown当纯文本data, _ := os.ReadFile("article.md")text := string(data)// → "## 标题" "**加粗**" 全部丢失语义// ✅ 正确:用blackfriday解析,保留结构html := blackfriday.Run(data)text := extractText(html)  // 去标签保留内容

6.3 特殊字符处理

// ❌ 错误:不处理特殊字符text := doc.Text()// → "Go\u00a0并发" 换行符、制表符混在文本里// ✅ 正确:清理特殊字符text = regexp.MustCompile(`[\u00a0\t\n\r]+`).ReplaceAllString(text, " ")text = strings.TrimSpace(text)

6.4 切分后Chunk太小

// ❌ 错误:最短Chunk不加限制for _, s := range sentences {    if current.Len() > 0 && current.Len()+len(s) <= chunkSize {        // 可能最后只剩几个字    }}// ✅ 正确:设置最小Chunk大小const minChunkSize = 100  // 至少100字才成一个Chunk// 否则合并到上一个Chunk

6.5 标题和正文混在一起

// ❌ 错误:标题+正文一起向量化combined := article.Title + article.Content// → 标题重复出现在每个Chunk里// ✅ 正确:标题作为元数据,不进Contentmetadatas: []map[string]interface{}{    {"title": article.Title, ...},  // 标题放元数据}// Content只放正文

总结

这一篇搞定了文档处理的核心:

 四种格式处理(TXT/Markdown/HTML/PDF) 三种分割策略(字符/句子/语义) 分割粒度选择(128-1024字) 博客CMS文章入库RAG完整流程 Context长度=Token成本=钱(核心) 踩坑表(5个高频坑)

一个最重要的认知升级:

Chunk大小不是越小越好,也不是越大越好。              

最优Chunk = 能表达一个完整语义的最小单元。

选512字,是因为大多数段落正好是300-500字;选1024字,是因为需要更多上下文。

下一篇预告:

RAG完整链路——Go实战博客知识库问答 + 效果评估。

等等,光有向量数据库和Embedding还不够,RAG还有几个关键环节没讲:

  • ·Query改写(用户说"学Go并发" → 改写成检索词)

  • ·重排序(Top20 → Top5)

  • ·效果评估(怎么知道RAG做得好不好?)

下一篇文章,把RAG全链路讲透。

🤔 留一个思考题(答案在评论区里找哟):

把1000字的文章切成5个200字的Chunk,和切成2个500字的Chunk,RAG检索质量一样吗?

📱 小貔貅技术日记

专注Java AI + Go AI应用开发

📦 源码地址:https://github.com/x-PiXiu/goStudySample

关注我,带你用Go玩转AI!