fix: rerank attribute unknown (#1554)

PR fixes #1550
This commit is contained in:
Gagan Bhullar
2024-08-22 00:16:36 -06:00
committed by GitHub
parent 758c82858f
commit 6eb7ccfdee

View File

@@ -457,6 +457,22 @@ class LanceQueryBuilder(ABC):
},
).explain_plan(verbose)
@abstractmethod
def rerank(self, reranker: Reranker) -> LanceQueryBuilder:
"""Rerank the results using the specified reranker.
Parameters
----------
reranker: Reranker
The reranker to use.
Returns
-------
The LanceQueryBuilder object.
"""
raise NotImplementedError
class LanceVectorQueryBuilder(LanceQueryBuilder):
"""
@@ -841,6 +857,21 @@ class LanceEmptyQueryBuilder(LanceQueryBuilder):
limit=self._limit,
)
def rerank(self, reranker: Reranker) -> LanceEmptyQueryBuilder:
"""Rerank the results using the specified reranker.
Parameters
----------
reranker: Reranker
The reranker to use.
Returns
-------
LanceEmptyQueryBuilder
The LanceQueryBuilder object.
"""
raise NotImplementedError("Reranking is not yet supported.")
class LanceHybridQueryBuilder(LanceQueryBuilder):
"""