马一丁

Use chapterId/title when building chapter overview

@@ -190,10 +190,10 @@ class GraphRAGQueryNode(BaseNode): @@ -190,10 +190,10 @@ class GraphRAGQueryNode(BaseNode):
190 # 获取搜索词样例 190 # 获取搜索词样例
191 sample_queries = query_engine.get_sample_search_queries(20) 191 sample_queries = query_engine.get_sample_search_queries(20)
192 192
193 - # 获取章节概览 193 + # 获取章节概览(word_plan 中的 chapters 使用 chapterId 而非 id)
194 chapters = context.get('chapters', []) 194 chapters = context.get('chapters', [])
195 chapters_text = "\n".join([ 195 chapters_text = "\n".join([
196 - f"- {c.get('id', '')}: {c.get('title', '')}" 196 + f"- {c.get('chapterId', c.get('id', ''))}: {c.get('title', c.get('display', ''))}"
197 for c in chapters[:10] 197 for c in chapters[:10]
198 ]) 198 ])
199 199