MetaGPT

1
"""Base Ranker."""
2

3
from abc import abstractmethod
4
from typing import Optional
5

6
from llama_index.core.postprocessor.types import BaseNodePostprocessor
7
from llama_index.core.schema import NodeWithScore, QueryBundle
8

9

10
class RAGRanker(BaseNodePostprocessor):
11
    """inherit from llama_index"""
12

13
    @abstractmethod
14
    def _postprocess_nodes(
15
        self,
16
        nodes: list[NodeWithScore],
17
        query_bundle: Optional[QueryBundle] = None,
18
    ) -> list[NodeWithScore]:
19
        """postprocess nodes."""
20