diff --git a/libs/core/langchain_core/cross_encoders.py b/libs/core/langchain_core/cross_encoders.py new file mode 100644 index 00000000000..7872daab213 --- /dev/null +++ b/libs/core/langchain_core/cross_encoders.py @@ -0,0 +1,18 @@ +"""Cross Encoder interface.""" + +from abc import ABC, abstractmethod + + +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. + """ diff --git a/libs/langchain/langchain_classic/retrievers/document_compressors/cross_encoder.py b/libs/langchain/langchain_classic/retrievers/document_compressors/cross_encoder.py index 7a26ceb5d4f..f8391d0188e 100644 --- a/libs/langchain/langchain_classic/retrievers/document_compressors/cross_encoder.py +++ b/libs/langchain/langchain_classic/retrievers/document_compressors/cross_encoder.py @@ -1,16 +1,3 @@ -from abc import ABC, abstractmethod +from langchain_core.cross_encoders 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"]