polyguard-rl/app/models/retrieval/reranker.py

"""Result reranker."""

from __future__ import annotations


def rerank(results: list[dict], query: str) -> list[dict]:
    qlen = max(1, len(query.split()))
    return sorted(results, key=lambda item: abs(len(item.get("text", "")) - qlen * 24))