From 32515ace74f8ef118adc0bc9cb79ea7c48da371b Mon Sep 17 00:00:00 2001 From: Rob Meng Date: Fri, 15 Dec 2023 00:19:08 -0500 Subject: [PATCH] feat: pass vector column name to remote backend (#710) pass vector column name to remote as well. `vector_column` is already part of `Query` just declearing it as part to `remote.VectorQuery` as well --- python/lancedb/remote/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) 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: