open-webui/backend/open_webui/retrieval/models/base_reranker.py

9 lines
219 B
Python

from abc import ABC, abstractmethod
from typing import Optional, List, Tuple
class BaseReranker(ABC):
@abstractmethod
def predict(self, sentences: List[Tuple[str, str]]) -> Optional[List[float]]:
pass