fix: preserve raptor tree titles (#17772)

2026-08-05 15:20:30 +08:00 · 2026-08-04 13:24:36 +08:00
parent 79a57d2126
commit bccbd7492c
1 changed files with 3 additions and 1 deletions
--- a/rag/advanced_rag/knowlege_compile/raptor.py
+++ b/rag/advanced_rag/knowlege_compile/raptor.py
@@ -231,12 +231,14 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval:
                    "",
                    cnt,
                )
+                cnt = str(cnt or "").strip()
                logging.debug(f"SUM: {cnt}")

                self._check_task_canceled(task_id, "before embedding")

                embds = await self._embedding_encode(cnt)
-                return cnt.split("\n")[0], cnt, embds
+                title = cnt.splitlines()[0].strip() if cnt else ""
+                return title, cnt, embds
        except TaskCanceledException:
            raise
        except Exception as exc: