Skip to content

Commit 97cd7d1

Browse files
committed
Add the default (no-op) reranker for AAP
1 parent dfefca7 commit 97cd7d1

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

ols/customize/aap/reranker.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
"""Reranker for post-processing the Vector DB search results."""
2+
3+
import logging
4+
5+
from llama_index.core.schema import NodeWithScore
6+
7+
logger = logging.getLogger(__name__)
8+
9+
10+
def rerank(retrieved_nodes: list[NodeWithScore]) -> list[NodeWithScore]:
11+
"""Rerank Vector DB search results."""
12+
message = f"reranker.rerank() is called with {len(retrieved_nodes)} result(s)."
13+
logger.debug(message)
14+
return retrieved_nodes

0 commit comments

Comments
 (0)