fix(data_source): scope table and link state to ancestors in format_document_soup (#17045)

### Summary

`format_document_soup` tracks "am I inside a table" and "am I inside a
link" with sticky flags that are meant to be reset by `elif e.name ==
"/table"` and `elif e.name == "/a"`. BeautifulSoup's `.descendants` only
yields opening tags — a `Tag` named `/table` or `/a` never exists — so
both branches are dead code and neither flag is ever cleared.

Everything after the first `<table>` on a page is therefore formatted as
if it were still table content: paragraphs lose their newline, list
items lose their `- ` marker, headings lose their break, and the text is
glued onto the last table cell. Under
`HTML_BASED_CONNECTOR_TRANSFORM_LINKS_STRATEGY=markdown` the same bug
leaks a link's `href` into everything that follows it, including whole
subsequent paragraphs. The Confluence connector
(`confluence_connector.py:948`) goes through this path.

Real output for a Confluence-shaped page (heading, intro, spec table,
then the body) via the public `parse_html_page_basic`:

**Before**

```
	prod	us-east-1 Rollback procedure If the canary fails, run the rollback script immediately. Drain the load balancer Revert the deployment Escalate to the on-call rota if the rollback stalls. Do not skip the post-mortem.
```

**After**

```
	prod	us-east-1
Rollback procedure
If the canary fails, run the rollback script immediately.
- Drain the load balancer
- Revert the deployment
Escalate to [the on-call rota](http://oncall.example.com) if the rollback stalls.
Do not skip the post-mortem.
```

Every heading, paragraph and list marker after the table is lost, and
the whole body is indexed as one run-on line hanging off a table cell.

### Fix

Derive both scopes from each element's **ancestors** instead of from
flags that nothing can clear, and drop the two dead branches plus the
two that become redundant.

The scopes are resolved in one up-front pass into `id`-keyed maps
(`table_scope`, `href_scope`) and looked up in O(1) per element. Probing
per element with `find_parent` instead is O(depth) each, which measured
12–13× slower on table-heavy pages and up to 103× on deeply nested
markup; the map version costs a depth-independent 1.13–1.35× over
`main`. Numbers and method are in the round-2 comment below.

This also changes one adjacent behaviour worth calling out explicitly: a
link **inside** a table cell now renders as markdown, where before it
rendered as plain text. That previous behaviour was not by design — it
only held when no link preceded the table. With a link before the table,
`main` stamps the stale href onto every cell:

```
main:   '[pre](http://STALE.com)\n\t[cellA](http://STALE.com)\t[cellB](http://STALE.com)'
branch: '[pre](http://STALE.com)\n\tcellA\tcellB'
```

Those cells are not links. Both symptoms are the same sticky-state bug,
so they are fixed together rather than left half-done.

### Testing

`test/unit_test/data_source/test_html_utils.py` is new —
`format_document_soup` had no test coverage. 11 tests: 8 fail on `main`
and pass on this branch, 3 are controls that pass on both (the table
itself still separates rows and cells, anchor text is still linkified,
the default `strip` strategy still strips).

Representative failures on `main`:

```
assert '\nAfter' in 'Before\n\tA\tB After'
assert '\n- item1' in 'Before\n\tA\tB item1 item2'
assert 'see [link](http://x.com) [ after](http://x.com)' == 'see [link](http://x.com) after'
assert '[next paragraph]' not in '[link](http://x.com)\n[next paragraph](http://x.com)'
```

Reverting each clause of the fix independently keeps the anchors honest:
reverting only the table clause fails exactly the 4 table tests and
leaves the link tests green; reverting only the link clause fails
exactly the 3 link tests and leaves the table tests green.
(`test_link_inside_a_table_cell_is_linkified` needs both clauses broken
to fail, so it appears in neither single-clause revert — it is covered
by the 8-fail run against `main`.)

Full `test/unit_test/data_source/` suite: **3 failed, 199 passed**, and
the failure set is byte-identical to clean `main` (**3 failed, 188
passed**) — the 3 are `TestSSRFValidation::*`, which resolve
`api.example.com` against real DNS and are unrelated to this change.
`ruff check` and `ruff format --check` are clean on both touched files.

---

This PR was drafted with AI assistance (Claude). I reviewed the change,
independently reproduced both symptoms against `main`, and take
responsibility for it.
This commit is contained in:
Aari
2026-08-11 22:28:40 +08:00
committed by GitHub
parent 492d6d81a9
commit a0e091e750
2 changed files with 106 additions and 14 deletions

View File

@@ -76,12 +76,24 @@ def format_document_soup(document: bs4.BeautifulSoup, table_cell_separator: str
text = ""
list_element_start = False
verbatim_output = 0
in_table = False
last_added_newline = False
link_href: str | None = None
# ``descendants`` yields opening tags only, so a flag set on <table>/<a> would
# never clear. Precompute scope by registering each <table>/<a> descendant;
# this avoids O(depth) ancestor walks while keeping lookups O(1).
table_scope = {id(d) for table in document.find_all("table") for d in table.descendants}
href_scope = {}
for anchor in document.find_all("a"): # document order, so a nested <a> wins over its parent
href_value = anchor.get("href", None)
# mostly for typing, having multiple hrefs is not valid HTML
link_href = href_value[0] if isinstance(href_value, list) else href_value
href_scope.update((id(d), link_href) for d in anchor.descendants)
for e in document.descendants:
verbatim_output -= 1
in_table = id(e) in table_scope
link_href = href_scope.get(id(e))
if isinstance(e, bs4.element.NavigableString):
if isinstance(e, (bs4.element.Comment, bs4.element.Doctype)):
continue
@@ -109,26 +121,15 @@ def format_document_soup(document: bs4.BeautifulSoup, table_cell_separator: str
list_element_start = False
elif isinstance(e, bs4.element.Tag):
# table is standard HTML element
if e.name == "table":
in_table = True
# TR is for rows
elif e.name == "tr" and in_table:
if e.name == "tr" and in_table:
text += "\n"
# td for data cell, th for header
elif e.name in ["td", "th"] and in_table:
text += table_cell_separator
elif e.name == "/table":
in_table = False
elif in_table:
# don't handle other cases while in table
pass
elif e.name == "a":
href_value = e.get("href", None)
# mostly for typing, having multiple hrefs is not valid HTML
link_href = href_value[0] if isinstance(href_value, list) else href_value
elif e.name == "/a":
link_href = None
elif e.name in ["p", "div"]:
if not list_element_start:
text += "\n"

View File

@@ -0,0 +1,91 @@
#
# Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import bs4
import pytest
from common.data_source import html_utils
from common.data_source.config import HtmlBasedConnectorTransformLinksStrategy
from common.data_source.html_utils import format_document_soup
def _fmt(html: str) -> str:
return format_document_soup(bs4.BeautifulSoup(html, "html.parser"))
@pytest.fixture
def markdown_links(monkeypatch):
"""``format_element_text`` only renders links under the markdown strategy."""
monkeypatch.setattr(
html_utils,
"HTML_BASED_CONNECTOR_TRANSFORM_LINKS_STRATEGY",
HtmlBasedConnectorTransformLinksStrategy.MARKDOWN,
)
TABLE = "<table><tr><td>A</td><td>B</td></tr></table>"
def test_paragraph_after_table_keeps_its_newline():
assert "\nAfter" in _fmt(f"<p>Before</p>{TABLE}<p>After</p>")
def test_list_after_table_keeps_hyphen_markers():
assert "\n- item1" in _fmt(f"<p>Before</p>{TABLE}<ul><li>item1</li><li>item2</li></ul>")
def test_block_after_table_starts_on_a_new_line():
# The div must not be folded onto the last table row.
assert "\nTrailing" in _fmt(f"{TABLE}<div>Trailing</div>")
def test_table_still_separates_rows_and_cells():
# Control: the table itself must keep working — rows on newlines, cells tab-separated.
assert _fmt("<table><tr><td>A</td><td>B</td></tr><tr><td>C</td><td>D</td></tr></table>") == "A\tB\n\tC\tD"
def test_content_after_table_matches_the_same_content_without_a_table(markdown_links):
tail = "<p>After</p><ul><li>item1</li></ul>"
with_table = _fmt(f"{TABLE}{tail}")
without_table = _fmt(tail)
assert with_table.endswith(without_table.lstrip("\n"))
def test_text_after_link_in_same_paragraph_is_not_linkified(markdown_links):
assert _fmt('<p>see <a href="http://x.com">link</a> after</p>') == "see [link](http://x.com) after"
def test_paragraph_after_link_is_not_linkified(markdown_links):
out = _fmt('<p><a href="http://x.com">link</a></p><p>next paragraph</p>')
assert "[next paragraph]" not in out
def test_link_inside_anchor_is_still_linkified(markdown_links):
# The anchor text itself must keep its markdown link.
assert _fmt('<p><a href="http://x.com">link</a></p>') == "[link](http://x.com)"
def test_table_cells_do_not_inherit_a_preceding_links_href(markdown_links):
out = _fmt(f'<p><a href="http://x.com">pre</a></p>{TABLE}')
assert "[A](http://x.com)" not in out
def test_link_inside_a_table_cell_is_linkified(markdown_links):
assert _fmt('<table><tr><td><a href="http://x.com">cell</a></td></tr></table>') == "[cell](http://x.com)"
def test_link_is_stripped_under_the_default_strategy():
# Default strategy is STRIP: no markdown link syntax at all.
assert _fmt('<p>see <a href="http://x.com">link</a> after</p>') == "see link after"