From 6394ebcfb8cc269cdc534fe332ca7fb68003739b Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 2 May 2024 14:25:30 -0400 Subject: [PATCH] Updating community BaseCrossEncoder to reference langhcain --- .../langchain_community/cross_encoders/base.py | 18 ++---------------- 1 file changed, 2 insertions(+), 16 deletions(-) diff --git a/libs/community/langchain_community/cross_encoders/base.py b/libs/community/langchain_community/cross_encoders/base.py index 98fa0568980..8f46e0f0dc3 100644 --- a/libs/community/langchain_community/cross_encoders/base.py +++ b/libs/community/langchain_community/cross_encoders/base.py @@ -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"]