mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-25 17:42:24 +08:00
fix(parser): derive eng from content for DOCX/HTML TOC removal (#17585)
## Summary - The `eng` flag passed to `removeTOCWord`/`removeContentsTable` was hardcoded `false`, so English documents used the CJK 3-char TOC prefix instead of Python's 2-word English prefix. This caused English tables of contents to be under-deleted (left in indexed text) or over-deleted (body paragraphs sharing the 3-char prefix dropped). - `eng` is now derived from the parsed item content via `isEnglishItems` (a port of Python `is_english`/`_is_english`), mirroring Python's content heuristic. ## Changes - `internal/parser/parser/text_toc.go`: add `isEnglishItems`/`isEnglishTexts` (ASCII-ratio >80% ⇒ English, fullmatch-anchored regex). - `internal/parser/parser/docx_parser.go`: pass `isEnglishItems(sections)` / `isEnglishItems(lineItems)` to `removeTOCWord` (json + markdown paths). - `internal/parser/parser/html_parser.go`: pass `isEnglishItems(items)` to `removeContentsTable`. - `docs/migration_python_go_diff.md`: close the 1.9/1.10 `eng` residual and resolve the contradictory Parser 2.11 "Partially fixed" note. ## Notes - `remove_toc` is a Parser-stage, DSL-configured feature (`remove_toc: true/false` in the parser family setup). This change only fixes the internal English/CJK prefix decision; it does not alter the pipeline or the DSL contract. `remove_header_footer` (precise match) is unaffected. - `eng` is auto-derived from content rather than exposed as a new config knob, to stay faithful to Python behavior. ## Test plan - `CGO_ENABLED=0 go test ./internal/parser/parser/ -run 'TestRemoveContentsTable|TestIsEnglishTexts|TestRemoveTOCWordEnglishDetection'` - Added `TestIsEnglishTexts` (ASCII-ratio cases) and `TestRemoveTOCWordEnglishDetection` (English TOC no longer over-deletes "Chapter 2 Method"). --------- Co-authored-by: CodeBuddy <noreply@codebuddy.ai>
This commit is contained in:
@@ -106,3 +106,60 @@ func TestRemoveContentsTable(t *testing.T) {
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// TestIsEnglishTexts classifies text as English when >80% of sampled segments
|
||||
// are ASCII.
|
||||
func TestIsEnglishTexts(t *testing.T) {
|
||||
cases := []struct {
|
||||
name string
|
||||
texts []string
|
||||
want bool
|
||||
}{
|
||||
{name: "empty", texts: nil, want: false},
|
||||
{name: "all english", texts: []string{"Chapter One", "Section Two", "Body text here"}, want: true},
|
||||
{name: "all chinese", texts: []string{"第一章 概述", "第二章 方法", "正文内容"}, want: false},
|
||||
{name: "mostly english (>80%)", texts: []string{
|
||||
"Introduction to the system", "Background and related work",
|
||||
"Methodology details", "Evaluation results", "Conclusion",
|
||||
"第一章", // single CJK line among 6 (>80% english)
|
||||
}, want: true},
|
||||
{name: "exactly 80% english", texts: []string{
|
||||
"Introduction", "Background", "Method", "Conclusion", "第一章",
|
||||
}, want: false}, // 4/5 == 80% is NOT > 80%
|
||||
{name: "mostly chinese (<80% english)", texts: []string{
|
||||
"第一章 概述", "第二章 方法", "第三章 结果", "English abstract only",
|
||||
}, want: false},
|
||||
}
|
||||
for _, tc := range cases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
if got := isEnglishTexts(tc.texts); got != tc.want {
|
||||
t.Errorf("isEnglishTexts(%v) = %v, want %v", tc.texts, got, tc.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// TestRemoveTOCWordEnglishDetection verifies English content uses the 2-word
|
||||
// TOC prefix, so "Chapter 2 Method" is NOT over-deleted (a 3-character prefix
|
||||
// would have dropped it).
|
||||
func TestRemoveTOCWordEnglishDetection(t *testing.T) {
|
||||
items := []map[string]any{
|
||||
{"text": "Intro"},
|
||||
{"text": "Contents"},
|
||||
{"text": "Chapter 1 Overview"},
|
||||
{"text": "Chapter 2 Method"},
|
||||
{"text": "Body starts"},
|
||||
}
|
||||
if !isEnglishItems(items) {
|
||||
t.Fatalf("expected isEnglishItems=true for English content")
|
||||
}
|
||||
got := removeContentsTable(items, isEnglishItems(items))
|
||||
want := []map[string]any{
|
||||
{"text": "Intro"},
|
||||
{"text": "Chapter 2 Method"},
|
||||
{"text": "Body starts"},
|
||||
}
|
||||
if !reflect.DeepEqual(got, want) {
|
||||
t.Errorf("English TOC detection: got %v, want %v", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user