rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-08 12:52:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Specify and Index Column for Vector Search (#217)

Browse Source
This commit is contained in:
Philip Kung
2023-06-26 16:11:08 -07:00
committed by GitHub
parent e850df56f1
commit 313e66c4c5
4 changed files with 97 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/lancedb/query.py
View File

@@ -45,7 +45,12 @@ class LanceQueryBuilder:
0 6 [0.4, 0.4] 0.0
"""
def __init__(self, table: "lancedb.table.LanceTable", query: np.ndarray):
def __init__(
self,
table: "lancedb.table.LanceTable",
query: np.ndarray,
vector_column_name: str = VECTOR_COLUMN_NAME,
):
self._metric = "L2"
self._nprobes = 20
self._refine_factor = None
@@ -54,6 +59,7 @@ class LanceQueryBuilder:
self._limit = 10
self._columns = None
self._where = None
self._vector_column_name = vector_column_name
def limit(self, limit: int) -> LanceQueryBuilder:
"""Set the maximum number of results to return.
@@ -195,7 +201,7 @@ class LanceQueryBuilder:
columns=self._columns,
filter=self._where,
nearest={
"column": VECTOR_COLUMN_NAME,
"column": self._vector_column_name,
"q": self._query,
"k": self._limit,
"metric": self._metric,