blob: b96b70a56fc5601d639ff7f9d8d0ccd6aea69fff (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
from abc import abstractmethod
from rag.generator.prompt import Prompt
class AbstractReranker(type):
_instances = {}
def __call__(cls, *args, **kwargs):
if cls not in cls._instances:
instance = super().__call__(*args, **kwargs)
cls._instances[cls] = instance
return cls._instances[cls]
@abstractmethod
def rank(self, prompt: Prompt) -> Prompt:
return prompt
|
