Updating community BaseCrossEncoder to reference langhcain

This commit is contained in:
Eugene Yurtsev
2024-05-02 14:25:30 -04:00
parent ce7aa2ff74
commit 6394ebcfb8

View File

@@ -1,17 +1,3 @@
from abc import ABC, abstractmethod
from typing import List, Tuple
from langchain.retrievers.document_compressors.cross_encoder import BaseCrossEncoder
class BaseCrossEncoder(ABC):
"""Interface for cross encoder models."""
@abstractmethod
def score(self, text_pairs: List[Tuple[str, str]]) -> List[float]:
"""Score pairs' similarity.
Args:
text_pairs: List of pairs of texts.
Returns:
List of scores.
"""
__all__ = ["BaseCrossEncoder"]