Rifqi Hafizuddin commited on
Commit
3e7924d
·
1 Parent(s): 73b7fe3

fix sorted ranking so model uses overall sorted retrieved chunks

Browse files
Files changed (1) hide show
  1. src/rag/retrievers/schema.py +1 -1
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),