Skip to content

Commit e730f42

Browse files
fix(python_code_splitter): preserve function identity in secondary splits
1 parent a8a6495 commit e730f42

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

haystack/components/preprocessors/python_code_splitter.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -556,7 +556,8 @@ def _secondary_split(self, unit: _CodeUnit, parent_doc: Document) -> list[Docume
556556
meta["secondary_split"] = True
557557
meta["secondary_split_index"] = idx
558558
meta["secondary_split_total"] = len(intermediate)
559-
results.append(Document(content=piece.content or "", meta=meta))
559+
content = piece.content or "" if idx == 0 else f"# {qualified_name}\n{piece.content or ''}"
560+
results.append(Document(content=content, meta=meta))
560561
return results
561562

562563
@component.output_types(documents=list[Document])

0 commit comments

Comments
 (0)