diff --git a/backend/infra/impl/document/parser/builtin/py_parser_protocol.go b/backend/infra/impl/document/parser/builtin/py_parser_protocol.go index a622d010..6477acc6 100644 --- a/backend/infra/impl/document/parser/builtin/py_parser_protocol.go +++ b/backend/infra/impl/document/parser/builtin/py_parser_protocol.go @@ -249,6 +249,11 @@ func formatTablesInDocument(input []*schema.Document) (output []*schema.Document values = append(values, col.Name) } write(values) + if colOnly, err := document.GetDocumentColumnsOnly(doc); err != nil { + return nil, err + } else if colOnly { + break + } } data, err := document.GetDocumentColumnData(doc)