diff --git a/python/lancedb/remote/__init__.py b/python/lancedb/remote/__init__.py index d8c37464..c6e15c83 100644 --- a/python/lancedb/remote/__init__.py +++ b/python/lancedb/remote/__init__.py @@ -18,6 +18,8 @@ import attrs import pyarrow as pa from pydantic import BaseModel +from lancedb.common import VECTOR_COLUMN_NAME + __all__ = ["LanceDBClient", "VectorQuery", "VectorQueryResult"] @@ -43,6 +45,8 @@ class VectorQuery(BaseModel): refine_factor: Optional[int] = None + vector_column: str = VECTOR_COLUMN_NAME + @attrs.define class VectorQueryResult: