scvcoder commited on
Commit
ba52b21
·
verified ·
1 Parent(s): 30fab83

context_builder: 라운드로빈 다양화 제거 → cross-source cross-encoder reranker. source 무관 적합도 순 top-N. pipeline 에서 query 전달

Browse files
Files changed (1) hide show
  1. src/kpaa/pipeline.py +1 -1
src/kpaa/pipeline.py CHANGED
@@ -78,7 +78,7 @@ async def build_context(
78
  )
79
  raw = await retrieve(plan, client=client, on_progress=on_progress)
80
  ranked = ranker.rank(raw)
81
- block = context_builder.build(ranked)
82
  return RetrievalResult(
83
  plan=plan,
84
  excerpts=ranked,
 
78
  )
79
  raw = await retrieve(plan, client=client, on_progress=on_progress)
80
  ranked = ranker.rank(raw)
81
+ block = context_builder.build(ranked, query=plan.query)
82
  return RetrievalResult(
83
  plan=plan,
84
  excerpts=ranked,