From bccbd7492cb9610cd530aa764aff9f0fe464ac60 Mon Sep 17 00:00:00 2001
From: buua436 <sz_buua@foxmail.com>
Date: Tue, 4 Aug 2026 13:24:36 +0800
Subject: [PATCH] fix: preserve raptor tree titles (#17772)

---
 rag/advanced_rag/knowlege_compile/raptor.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/rag/advanced_rag/knowlege_compile/raptor.py b/rag/advanced_rag/knowlege_compile/raptor.py
index 48723a6844..587eb1f24b 100644
--- 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: