context_builder: 라운드로빈 다양화 제거 → cross-source cross-encoder reranker. source 무관 적합도 순 top-N. pipeline 에서 query 전달
Browse files- 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,
|