feat: expose use_index in LanceDB OSS

2026-01-15 00:02:59 +00:00 · 2023-10-13 11:17:10 -07:00
parent 683824f1e9
commit f9ccefb032
8 changed files with 71 additions and 3 deletions
--- a/python/tests/test_query.py
+++ b/python/tests/test_query.py
@@ -46,6 +46,7 @@ class MockTable:
                "metric": query.metric,
                "nprobes": query.nprobes,
                "refine_factor": query.refine_factor,
+                "use_index": query.use_index,
            },
        )

@@ -84,10 +85,12 @@ def test_cast(table):
    assert r0.float_field == 1.0


-def test_query_builder(table):
+@pytest.mark.parametrize("use_index", [True, False])
+def test_query_builder(table, use_index: bool):
    rs = (
        LanceVectorQueryBuilder(table, [0, 0], "vector")
        .limit(1)
+        .use_index(use_index)
        .select(["id"])
        .to_list()
    )