diff --git a/python/python/lancedb/query.py b/python/python/lancedb/query.py index 95b691b2..75d30270 100644 --- a/python/python/lancedb/query.py +++ b/python/python/lancedb/query.py @@ -481,6 +481,7 @@ class LanceQueryBuilder(ABC): >>> plan = table.search(query).explain_plan(True) >>> print(plan) # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE ProjectionExec: expr=[vector@0 as vector, _distance@2 as _distance] + GlobalLimitExec: skip=0, fetch=10 FilterExec: _distance@2 IS NOT NULL SortExec: TopK(fetch=10), expr=[_distance@2 ASC NULLS LAST], preserve_partitioning=[false] KNNVectorDistance: metric=l2 @@ -500,7 +501,16 @@ class LanceQueryBuilder(ABC): nearest={ "column": self._vector_column, "q": self._query, + "k": self._limit, + "metric": self._metric, + "nprobes": self._nprobes, + "refine_factor": self._refine_factor, }, + prefilter=self._prefilter, + filter=self._str_query, + limit=self._limit, + with_row_id=self._with_row_id, + offset=self._offset, ).explain_plan(verbose) def vector(self, vector: Union[np.ndarray, list]) -> LanceQueryBuilder: