Merge branch 'master' into add-metadata

This commit is contained in:
Tanushree
2026-03-12 17:49:32 -07:00
committed by GitHub
2 changed files with 20 additions and 15 deletions

View File

@@ -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.
"""

View File

@@ -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"]