From 3472cb2c6dc81696063793bb869d266168c30282 Mon Sep 17 00:00:00 2001 From: suhaniiz Date: Wed, 8 Jul 2026 21:30:38 +0530 Subject: [PATCH] fix: resolve KeyError when text_key is missing in documents #816 --- backend/app/rag/reranker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/app/rag/reranker.py b/backend/app/rag/reranker.py index f43116ba..ba5adefc 100644 --- a/backend/app/rag/reranker.py +++ b/backend/app/rag/reranker.py @@ -67,10 +67,10 @@ def rerank( model = self._load_model() - # Prepare query-document pairs - pairs = [(query, doc[text_key]) for doc in documents] + # Safely extract text, fallback to an empty string if key is missing to avoid crashes + pairs = [(query, doc.get(text_key, "")) for doc in documents] - # Get relevance scores + # Get relevance scores (utilizing batch_size to prevent OOM errors) scores = model.predict(pairs) # Pair scores with documents and sort in descending order