Source code for langchain_community.cross_encoders.fake

from difflib import SequenceMatcher
from typing import List, Tuple

from langchain_core.pydantic_v1 import BaseModel

from langchain_community.cross_encoders.base import BaseCrossEncoder


[docs]class FakeCrossEncoder(BaseCrossEncoder, BaseModel):
    """Fake cross encoder model."""

[docs]    def score(self, text_pairs: List[Tuple[str, str]]) -> List[float]:
        scores = list(
            map(
                lambda pair: SequenceMatcher(None, pair[0], pair[1]).ratio(), text_pairs
            )
        )
        return scores