|
@@ -818,7 +818,9 @@ class RerankCompressor(BaseDocumentCompressor):
|
|
|
)
|
|
|
scores = util.cos_sim(query_embedding, document_embedding)[0]
|
|
|
|
|
|
- docs_with_scores = list(zip(documents, scores.tolist()))
|
|
|
+ docs_with_scores = list(
|
|
|
+ zip(documents, scores.tolist() if not isinstance(scores, list) else scores)
|
|
|
+ )
|
|
|
if self.r_score:
|
|
|
docs_with_scores = [
|
|
|
(d, s) for d, s in docs_with_scores if s >= self.r_score
|