Rifqi Hafizuddin commited on
Commit ·
3e7924d
1
Parent(s): 73b7fe3
fix sorted ranking so model uses overall sorted retrieved chunks
Browse files
src/rag/retrievers/schema.py
CHANGED
|
@@ -393,7 +393,7 @@ class SchemaRetriever(BaseRetriever):
|
|
| 393 |
db_ranked = self._rank_db_tables(db_tbl_results, db_col_results, fts_results, top_k=k)
|
| 394 |
tabular_ranked = self._rank_tabular_sheets(sheet_results, tabular_results, top_k=k)
|
| 395 |
|
| 396 |
-
results = db_ranked + tabular_ranked
|
| 397 |
logger.info(
|
| 398 |
"schema retrieval",
|
| 399 |
count=len(results),
|
|
|
|
| 393 |
db_ranked = self._rank_db_tables(db_tbl_results, db_col_results, fts_results, top_k=k)
|
| 394 |
tabular_ranked = self._rank_tabular_sheets(sheet_results, tabular_results, top_k=k)
|
| 395 |
|
| 396 |
+
results = sorted(db_ranked + tabular_ranked, key=lambda r: r.score, reverse=True)
|
| 397 |
logger.info(
|
| 398 |
"schema retrieval",
|
| 399 |
count=len(results),
|