diff --git a/python/python/lancedb/query.py b/python/python/lancedb/query.py index 6c3c71bd..271028d9 100644 --- a/python/python/lancedb/query.py +++ b/python/python/lancedb/query.py @@ -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): """