@lancedb/lancedb 0.21.2 → 0.21.4-beta.0

package/dist/arrow.js

@@ -56,6 +56,13 @@ exports.dataTypeToJson = dataTypeToJson;
 const apache_arrow_1 = require("apache-arrow");
 const registry_1 = require("./embedding/registry");
 const sanitize_1 = require("./sanitize");
+/**
+ * Check if a field name indicates a vector column.
+ */
+function nameSuggestsVectorColumn(fieldName) {
+    const nameLower = fieldName.toLowerCase();
+    return nameLower.includes("vector") || nameLower.includes("embedding");
+}
 __exportStar(require("apache-arrow"), exports);
 function isMultiVector(value) {
     return Array.isArray(value) && isIntoVector(value[0]);
@@ -494,10 +501,18 @@ function inferType(value, path, opts) {
             return undefined;
         }
         // Try to automatically detect embedding columns.
-        if (valueType instanceof apache_arrow_1.Float && path[path.length - 1] === "vector") {
-            // We default to Float32 for vectors.
-            const child = new apache_arrow_1.Field("item", new apache_arrow_1.Float32(), true);
-            return new apache_arrow_1.FixedSizeList(value.length, child);
+        if (nameSuggestsVectorColumn(path[path.length - 1])) {
+            // Check if value is a Uint8Array for integer vector type determination
+            if (value instanceof Uint8Array) {
+                // For integer vectors, we default to Uint8 (matching Python implementation)
+                const child = new apache_arrow_1.Field("item", new apache_arrow_1.Uint8(), true);
+                return new apache_arrow_1.FixedSizeList(value.length, child);
+            }
+            else {
+                // For float vectors, we default to Float32
+                const child = new apache_arrow_1.Field("item", new apache_arrow_1.Float32(), true);
+                return new apache_arrow_1.FixedSizeList(value.length, child);
+            }
         }
         else {
             const child = new apache_arrow_1.Field("item", valueType, true);

package/dist/index.d.ts

@@ -4,7 +4,7 @@ export { AddColumnsSql, ConnectionOptions, IndexStatistics, IndexConfig, ClientC
 export { makeArrowTable, MakeArrowTableOptions, Data, VectorColumnOptions, } from "./arrow";
 export { Connection, CreateTableOptions, TableNamesOptions, OpenTableOptions, } from "./connection";
 export { Session } from "./native.js";
-export { ExecutableQuery, Query, QueryBase, VectorQuery, QueryExecutionOptions, FullTextSearchOptions, RecordBatchIterator, FullTextQuery, MatchQuery, PhraseQuery, BoostQuery, MultiMatchQuery, BooleanQuery, FullTextQueryType, Operator, Occur, } from "./query";
+export { ExecutableQuery, Query, QueryBase, VectorQuery, TakeQuery, QueryExecutionOptions, FullTextSearchOptions, RecordBatchIterator, FullTextQuery, MatchQuery, PhraseQuery, BoostQuery, MultiMatchQuery, BooleanQuery, FullTextQueryType, Operator, Occur, } from "./query";
 export { Index, IndexOptions, IvfPqOptions, IvfFlatOptions, HnswPqOptions, HnswSqOptions, FtsOptions, } from "./indices";
 export { Table, AddDataOptions, UpdateOptions, OptimizeOptions, Version, ColumnAlteration, } from "./table";
 export { MergeInsertBuilder, WriteExecutionOptions } from "./merge";

package/dist/index.js

@@ -2,7 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 // SPDX-FileCopyrightText: Copyright The LanceDB Authors
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.packBits = exports.rerankers = exports.embedding = exports.MergeInsertBuilder = exports.Table = exports.Index = exports.Occur = exports.Operator = exports.FullTextQueryType = exports.BooleanQuery = exports.MultiMatchQuery = exports.BoostQuery = exports.PhraseQuery = exports.MatchQuery = exports.RecordBatchIterator = exports.VectorQuery = exports.QueryBase = exports.Query = exports.Session = exports.Connection = exports.VectorColumnOptions = exports.MakeArrowTableOptions = exports.makeArrowTable = exports.TagContents = exports.Tags = void 0;
+exports.packBits = exports.rerankers = exports.embedding = exports.MergeInsertBuilder = exports.Table = exports.Index = exports.Occur = exports.Operator = exports.FullTextQueryType = exports.BooleanQuery = exports.MultiMatchQuery = exports.BoostQuery = exports.PhraseQuery = exports.MatchQuery = exports.RecordBatchIterator = exports.TakeQuery = exports.VectorQuery = exports.QueryBase = exports.Query = exports.Session = exports.Connection = exports.VectorColumnOptions = exports.MakeArrowTableOptions = exports.makeArrowTable = exports.TagContents = exports.Tags = void 0;
 exports.connect = connect;
 const connection_1 = require("./connection");
 const native_js_1 = require("./native.js");
@@ -21,6 +21,7 @@ var query_1 = require("./query");
 Object.defineProperty(exports, "Query", { enumerable: true, get: function () { return query_1.Query; } });
 Object.defineProperty(exports, "QueryBase", { enumerable: true, get: function () { return query_1.QueryBase; } });
 Object.defineProperty(exports, "VectorQuery", { enumerable: true, get: function () { return query_1.VectorQuery; } });
+Object.defineProperty(exports, "TakeQuery", { enumerable: true, get: function () { return query_1.TakeQuery; } });
 Object.defineProperty(exports, "RecordBatchIterator", { enumerable: true, get: function () { return query_1.RecordBatchIterator; } });
 Object.defineProperty(exports, "MatchQuery", { enumerable: true, get: function () { return query_1.MatchQuery; } });
 Object.defineProperty(exports, "PhraseQuery", { enumerable: true, get: function () { return query_1.PhraseQuery; } });

package/dist/indices.d.ts

@@ -585,5 +585,25 @@ export interface IndexOptions {
      * The default is true
      */
     replace?: boolean;
+    /**
+     * Timeout in seconds to wait for index creation to complete.
+     *
+     * If not specified, the method will return immediately after starting the index creation.
+     */
     waitTimeoutSeconds?: number;
+    /**
+     * Optional custom name for the index.
+     *
+     * If not provided, a default name will be generated based on the column name.
+     */
+    name?: string;
+    /**
+     * Whether to train the index with existing data.
+     *
+     * If true (default), the index will be trained with existing data in the table.
+     * If false, the index will be created empty and populated as new data is added.
+     *
+     * Note: This option is only supported for scalar indices. Vector indices always train.
+     */
+    train?: boolean;
 }

package/dist/native.d.ts

@@ -5,6 +5,14 @@
 
 /** Timeout configuration for remote HTTP client. */
 export interface TimeoutConfig {
+  /**
+   * The overall timeout for the entire request in seconds. This includes
+   * connection, send, and read time. If the entire request doesn't complete
+   * within this time, it will fail. Default is None (no overall timeout).
+   * This can also be set via the environment variable `LANCE_CLIENT_TIMEOUT`,
+   * as an integer number of seconds.
+   */
+  timeout?: number
   /**
    * The timeout for establishing a connection in seconds. Default is 120
    * seconds (2 minutes). This can also be set via the environment variable
@@ -385,6 +393,14 @@ export class VectorQuery {
   explainPlan(verbose: boolean): Promise<string>
   analyzePlan(): Promise<string>
 }
+export class TakeQuery {
+  select(columns: Array<[string, string]>): void
+  selectColumns(columns: Array<string>): void
+  withRowId(): void
+  execute(maxBatchLength?: number | undefined | null, timeoutMs?: number | undefined | null): Promise<RecordBatchIterator>
+  explainPlan(verbose: boolean): Promise<string>
+  analyzePlan(): Promise<string>
+}
 export class JsFullTextQuery {
   static matchQuery(query: string, column: string, boost: number, fuzziness: number | undefined | null, maxExpansions: number, operator: string, prefixLength: number): JsFullTextQuery
   static phraseQuery(query: string, column: string, slop: number): JsFullTextQuery
@@ -452,13 +468,15 @@ export class Table {
   add(buf: Buffer, mode: string): Promise<AddResult>
   countRows(filter?: string | undefined | null): Promise<number>
   delete(predicate: string): Promise<DeleteResult>
-  createIndex(index: Index | undefined | null, column: string, replace?: boolean | undefined | null, waitTimeoutS?: number | undefined | null): Promise<void>
+  createIndex(index: Index | undefined | null, column: string, replace?: boolean | undefined | null, waitTimeoutS?: number | undefined | null, name?: string | undefined | null, train?: boolean | undefined | null): Promise<void>
   dropIndex(indexName: string): Promise<void>
   prewarmIndex(indexName: string): Promise<void>
   waitForIndex(indexNames: Array<string>, timeoutS: number): Promise<void>
   stats(): Promise<TableStatistics>
   update(onlyIf: string | undefined | null, columns: Array<[string, string]>): Promise<UpdateResult>
   query(): Query
+  takeOffsets(offsets: Array<number>): TakeQuery
+  takeRowIds(rowIds: Array<number>): TakeQuery
   vectorSearch(vector: Float32Array): VectorQuery
   addColumns(transforms: Array<AddColumnsSql>): Promise<AddColumnsResult>
   alterColumns(alterations: Array<ColumnAlteration>): Promise<AlterColumnsResult>

package/dist/native.js

@@ -319,13 +319,14 @@ if (!nativeBinding) {
     }
     throw new Error(`Failed to load native binding`);
 }
-const { Connection, Index, RecordBatchIterator, NativeMergeInsertBuilder, Query, VectorQuery, JsFullTextQuery, Reranker, RrfReranker, Session, Table, TagContents, Tags } = nativeBinding;
+const { Connection, Index, RecordBatchIterator, NativeMergeInsertBuilder, Query, VectorQuery, TakeQuery, JsFullTextQuery, Reranker, RrfReranker, Session, Table, TagContents, Tags } = nativeBinding;
 module.exports.Connection = Connection;
 module.exports.Index = Index;
 module.exports.RecordBatchIterator = RecordBatchIterator;
 module.exports.NativeMergeInsertBuilder = NativeMergeInsertBuilder;
 module.exports.Query = Query;
 module.exports.VectorQuery = VectorQuery;
+module.exports.TakeQuery = TakeQuery;
 module.exports.JsFullTextQuery = JsFullTextQuery;
 module.exports.Reranker = Reranker;
 module.exports.RrfReranker = RrfReranker;

package/dist/query.d.ts

@@ -1,6 +1,6 @@
 import { Table as ArrowTable, type IntoVector, RecordBatch } from "./arrow";
 import { type IvfPqOptions } from "./indices";
-import { JsFullTextQuery, RecordBatchIterator as NativeBatchIterator, Query as NativeQuery, Table as NativeTable, VectorQuery as NativeVectorQuery } from "./native";
+import { JsFullTextQuery, RecordBatchIterator as NativeBatchIterator, Query as NativeQuery, Table as NativeTable, TakeQuery as NativeTakeQuery, VectorQuery as NativeVectorQuery } from "./native";
 import { Reranker } from "./rerankers";
 export declare class RecordBatchIterator implements AsyncIterator<RecordBatch> {
     private promisedInner?;
@@ -43,7 +43,7 @@ export interface FullTextSearchOptions {
  *
  * @hideconstructor
  */
-export declare class QueryBase<NativeQueryType extends NativeQuery | NativeVectorQuery> implements AsyncIterable<RecordBatch> {
+export declare class QueryBase<NativeQueryType extends NativeQuery | NativeVectorQuery | NativeTakeQuery> implements AsyncIterable<RecordBatch> {
     protected inner: NativeQueryType | Promise<NativeQueryType>;
     /**
      * @hidden
@@ -53,26 +53,6 @@ export declare class QueryBase<NativeQueryType extends NativeQuery | NativeVecto
      * @hidden
      */
     protected doCall(fn: (inner: NativeQueryType) => void): void;
-    /**
-     * A filter statement to be applied to this query.
-     *
-     * The filter should be supplied as an SQL query string.  For example:
-     * @example
-     * x > 10
-     * y > 0 AND y < 100
-     * x > 5 OR y = 'test'
-     *
-     * Filtering performance can often be improved by creating a scalar index
-     * on the filter column(s).
-     */
-    where(predicate: string): this;
-    /**
-     * A filter statement to be applied to this query.
-     * @see where
-     * @deprecated Use `where` instead
-     */
-    filter(predicate: string): this;
-    fullTextSearch(query: string | FullTextQuery, options?: Partial<FullTextSearchOptions>): this;
     /**
      * Return only the specified columns.
      *
@@ -104,21 +84,6 @@ export declare class QueryBase<NativeQueryType extends NativeQuery | NativeVecto
      * object insertion order is easy to get wrong and `Map` is more foolproof.
      */
     select(columns: string[] | Map<string, string> | Record<string, string> | string): this;
-    /**
-     * Set the maximum number of results to return.
-     *
-     * By default, a plain search has no limit.  If this method is not
-     * called then every valid row from the table will be returned.
-     */
-    limit(limit: number): this;
-    offset(offset: number): this;
-    /**
-     * Skip searching un-indexed data. This can make search faster, but will miss
-     * any data that is not yet indexed.
-     *
-     * Use {@link Table#optimize} to index all un-indexed data.
-     */
-    fastSearch(): this;
     /**
      * Whether to return the row id in the results.
      *
@@ -197,6 +162,49 @@ export declare class QueryBase<NativeQueryType extends NativeQuery | NativeVecto
      */
     analyzePlan(): Promise<string>;
 }
+export declare class StandardQueryBase<NativeQueryType extends NativeQuery | NativeVectorQuery> extends QueryBase<NativeQueryType> implements ExecutableQuery {
+    constructor(inner: NativeQueryType | Promise<NativeQueryType>);
+    /**
+     * A filter statement to be applied to this query.
+     *
+     * The filter should be supplied as an SQL query string.  For example:
+     * @example
+     * x > 10
+     * y > 0 AND y < 100
+     * x > 5 OR y = 'test'
+     *
+     * Filtering performance can often be improved by creating a scalar index
+     * on the filter column(s).
+     */
+    where(predicate: string): this;
+    /**
+     * A filter statement to be applied to this query.
+     * @see where
+     * @deprecated Use `where` instead
+     */
+    filter(predicate: string): this;
+    fullTextSearch(query: string | FullTextQuery, options?: Partial<FullTextSearchOptions>): this;
+    /**
+     * Set the maximum number of results to return.
+     *
+     * By default, a plain search has no limit.  If this method is not
+     * called then every valid row from the table will be returned.
+     */
+    limit(limit: number): this;
+    /**
+     * Set the number of rows to skip before returning results.
+     *
+     * This is useful for pagination.
+     */
+    offset(offset: number): this;
+    /**
+     * Skip searching un-indexed data. This can make search faster, but will miss
+     * any data that is not yet indexed.
+     *
+     * Use {@link Table#optimize} to index all un-indexed data.
+     */
+    fastSearch(): this;
+}
 /**
  * An interface for a query that can be executed
  *
@@ -213,7 +221,7 @@ export interface ExecutableQuery {
  *
  * @hideconstructor
  */
-export declare class VectorQuery extends QueryBase<NativeVectorQuery> {
+export declare class VectorQuery extends StandardQueryBase<NativeVectorQuery> {
     /**
      * @hidden
      */
@@ -364,13 +372,21 @@ export declare class VectorQuery extends QueryBase<NativeVectorQuery> {
     addQueryVector(vector: IntoVector): VectorQuery;
     rerank(reranker: Reranker): VectorQuery;
 }
+/**
+ * A query that returns a subset of the rows in the table.
+ *
+ * @hideconstructor
+ */
+export declare class TakeQuery extends QueryBase<NativeTakeQuery> {
+    constructor(inner: NativeTakeQuery);
+}
 /** A builder for LanceDB queries.
  *
  * @see {@link Table#query}, {@link Table#search}
  *
  * @hideconstructor
  */
-export declare class Query extends QueryBase<NativeQuery> {
+export declare class Query extends StandardQueryBase<NativeQuery> {
     /**
      * @hidden
      */

package/dist/query.js

@@ -2,7 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 // SPDX-FileCopyrightText: Copyright The LanceDB Authors
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BooleanQuery = exports.MultiMatchQuery = exports.BoostQuery = exports.PhraseQuery = exports.MatchQuery = exports.Occur = exports.Operator = exports.FullTextQueryType = exports.Query = exports.VectorQuery = exports.QueryBase = exports.RecordBatchIterator = void 0;
+exports.BooleanQuery = exports.MultiMatchQuery = exports.BoostQuery = exports.PhraseQuery = exports.MatchQuery = exports.Occur = exports.Operator = exports.FullTextQueryType = exports.Query = exports.TakeQuery = exports.VectorQuery = exports.StandardQueryBase = exports.QueryBase = exports.RecordBatchIterator = void 0;
 exports.instanceOfFullTextQuery = instanceOfFullTextQuery;
 const arrow_1 = require("./arrow");
 const native_1 = require("./native");
@@ -77,53 +77,6 @@ class QueryBase {
             fn(this.inner);
         }
     }
-    /**
-     * A filter statement to be applied to this query.
-     *
-     * The filter should be supplied as an SQL query string.  For example:
-     * @example
-     * x > 10
-     * y > 0 AND y < 100
-     * x > 5 OR y = 'test'
-     *
-     * Filtering performance can often be improved by creating a scalar index
-     * on the filter column(s).
-     */
-    where(predicate) {
-        this.doCall((inner) => inner.onlyIf(predicate));
-        return this;
-    }
-    /**
-     * A filter statement to be applied to this query.
-     * @see where
-     * @deprecated Use `where` instead
-     */
-    filter(predicate) {
-        return this.where(predicate);
-    }
-    fullTextSearch(query, options) {
-        let columns = null;
-        if (options) {
-            if (typeof options.columns === "string") {
-                columns = [options.columns];
-            }
-            else if (Array.isArray(options.columns)) {
-                columns = options.columns;
-            }
-        }
-        this.doCall((inner) => {
-            if (typeof query === "string") {
-                inner.fullTextSearch({
-                    query: query,
-                    columns: columns,
-                });
-            }
-            else {
-                inner.fullTextSearch({ query: query.inner });
-            }
-        });
-        return this;
-    }
     /**
      * Return only the specified columns.
      *
@@ -179,30 +132,6 @@ class QueryBase {
         }
         return this;
     }
-    /**
-     * Set the maximum number of results to return.
-     *
-     * By default, a plain search has no limit.  If this method is not
-     * called then every valid row from the table will be returned.
-     */
-    limit(limit) {
-        this.doCall((inner) => inner.limit(limit));
-        return this;
-    }
-    offset(offset) {
-        this.doCall((inner) => inner.offset(offset));
-        return this;
-    }
-    /**
-     * Skip searching un-indexed data. This can make search faster, but will miss
-     * any data that is not yet indexed.
-     *
-     * Use {@link Table#optimize} to index all un-indexed data.
-     */
-    fastSearch() {
-        this.doCall((inner) => inner.fastSearch());
-        return this;
-    }
     /**
      * Whether to return the row id in the results.
      *
@@ -329,6 +258,88 @@ class QueryBase {
     }
 }
 exports.QueryBase = QueryBase;
+class StandardQueryBase extends QueryBase {
+    constructor(inner) {
+        super(inner);
+    }
+    /**
+     * A filter statement to be applied to this query.
+     *
+     * The filter should be supplied as an SQL query string.  For example:
+     * @example
+     * x > 10
+     * y > 0 AND y < 100
+     * x > 5 OR y = 'test'
+     *
+     * Filtering performance can often be improved by creating a scalar index
+     * on the filter column(s).
+     */
+    where(predicate) {
+        this.doCall((inner) => inner.onlyIf(predicate));
+        return this;
+    }
+    /**
+     * A filter statement to be applied to this query.
+     * @see where
+     * @deprecated Use `where` instead
+     */
+    filter(predicate) {
+        return this.where(predicate);
+    }
+    fullTextSearch(query, options) {
+        let columns = null;
+        if (options) {
+            if (typeof options.columns === "string") {
+                columns = [options.columns];
+            }
+            else if (Array.isArray(options.columns)) {
+                columns = options.columns;
+            }
+        }
+        this.doCall((inner) => {
+            if (typeof query === "string") {
+                inner.fullTextSearch({
+                    query: query,
+                    columns: columns,
+                });
+            }
+            else {
+                inner.fullTextSearch({ query: query.inner });
+            }
+        });
+        return this;
+    }
+    /**
+     * Set the maximum number of results to return.
+     *
+     * By default, a plain search has no limit.  If this method is not
+     * called then every valid row from the table will be returned.
+     */
+    limit(limit) {
+        this.doCall((inner) => inner.limit(limit));
+        return this;
+    }
+    /**
+     * Set the number of rows to skip before returning results.
+     *
+     * This is useful for pagination.
+     */
+    offset(offset) {
+        this.doCall((inner) => inner.offset(offset));
+        return this;
+    }
+    /**
+     * Skip searching un-indexed data. This can make search faster, but will miss
+     * any data that is not yet indexed.
+     *
+     * Use {@link Table#optimize} to index all un-indexed data.
+     */
+    fastSearch() {
+        this.doCall((inner) => inner.fastSearch());
+        return this;
+    }
+}
+exports.StandardQueryBase = StandardQueryBase;
 /**
  * A builder used to construct a vector search
  *
@@ -338,7 +349,7 @@ exports.QueryBase = QueryBase;
  *
  * @hideconstructor
  */
-class VectorQuery extends QueryBase {
+class VectorQuery extends StandardQueryBase {
     /**
      * @hidden
      */
@@ -575,13 +586,24 @@ class VectorQuery extends QueryBase {
     }
 }
 exports.VectorQuery = VectorQuery;
+/**
+ * A query that returns a subset of the rows in the table.
+ *
+ * @hideconstructor
+ */
+class TakeQuery extends QueryBase {
+    constructor(inner) {
+        super(inner);
+    }
+}
+exports.TakeQuery = TakeQuery;
 /** A builder for LanceDB queries.
  *
  * @see {@link Table#query}, {@link Table#search}
  *
  * @hideconstructor
  */
-class Query extends QueryBase {
+class Query extends StandardQueryBase {
     /**
      * @hidden
      */

package/dist/table.d.ts

@@ -2,7 +2,7 @@ import { Table as ArrowTable, Data, DataType, IntoVector, MultiVector, Schema }
 import { IndexOptions } from "./indices";
 import { MergeInsertBuilder } from "./merge";
 import { AddColumnsResult, AddColumnsSql, AddResult, AlterColumnsResult, DeleteResult, DropColumnsResult, IndexConfig, IndexStatistics, OptimizeStats, TableStatistics, Tags, UpdateResult, Table as _NativeTable } from "./native";
-import { FullTextQuery, Query, VectorQuery } from "./query";
+import { FullTextQuery, Query, TakeQuery, VectorQuery } from "./query";
 import { IntoSql } from "./util";
 export { IndexConfig } from "./native";
 /**
@@ -269,6 +269,18 @@ export declare abstract class Table {
      * @returns {Query} A builder that can be used to parameterize the query
      */
     abstract query(): Query;
+    /**
+     * Create a query that returns a subset of the rows in the table.
+     * @param offsets The offsets of the rows to return.
+     * @returns A builder that can be used to parameterize the query.
+     */
+    abstract takeOffsets(offsets: number[]): TakeQuery;
+    /**
+     * Create a query that returns a subset of the rows in the table.
+     * @param rowIds The row ids of the rows to return.
+     * @returns A builder that can be used to parameterize the query.
+     */
+    abstract takeRowIds(rowIds: number[]): TakeQuery;
     /**
      * Create a search query to find the nearest neighbors
      * of the given query
@@ -462,6 +474,8 @@ export declare class LocalTable extends Table {
     dropIndex(name: string): Promise<void>;
     prewarmIndex(name: string): Promise<void>;
     waitForIndex(indexNames: string[], timeoutSeconds: number): Promise<void>;
+    takeOffsets(offsets: number[]): TakeQuery;
+    takeRowIds(rowIds: number[]): TakeQuery;
     query(): Query;
     search(query: string | IntoVector | MultiVector | FullTextQuery, queryType?: string, ftsColumns?: string | string[]): VectorQuery | Query;
     vectorSearch(vector: IntoVector | MultiVector): VectorQuery;

package/dist/table.js

@@ -123,7 +123,7 @@ class LocalTable extends Table {
         // Bit of a hack to get around the fact that TS has no package-scope.
         // biome-ignore lint/suspicious/noExplicitAny: skip
         const nativeIndex = options?.config?.inner;
-        await this.inner.createIndex(nativeIndex, column, options?.replace, options?.waitTimeoutSeconds);
+        await this.inner.createIndex(nativeIndex, column, options?.replace, options?.waitTimeoutSeconds, options?.name, options?.train);
     }
     async dropIndex(name) {
         await this.inner.dropIndex(name);
@@ -134,6 +134,12 @@ class LocalTable extends Table {
     async waitForIndex(indexNames, timeoutSeconds) {
         await this.inner.waitForIndex(indexNames, timeoutSeconds);
     }
+    takeOffsets(offsets) {
+        return new query_1.TakeQuery(this.inner.takeOffsets(offsets));
+    }
+    takeRowIds(rowIds) {
+        return new query_1.TakeQuery(this.inner.takeRowIds(rowIds));
+    }
     query() {
         return new query_1.Query(this.inner);
     }

package/package.json

@@ -11,7 +11,7 @@
     "ann"
   ],
   "private": false,
-  "version": "0.21.2",
+  "version": "0.21.4-beta.0",
   "main": "dist/index.js",
   "exports": {
     ".": "./dist/index.js",
@@ -36,7 +36,7 @@
       ]
     }
   },
-  "license": "Apache 2.0",
+  "license": "Apache-2.0",
   "devDependencies": {
     "@aws-sdk/client-dynamodb": "^3.33.0",
     "@aws-sdk/client-kms": "^3.33.0",
@@ -100,14 +100,14 @@
     "reflect-metadata": "^0.2.2"
   },
   "optionalDependencies": {
-    "@lancedb/lancedb-darwin-x64": "0.21.2",
-    "@lancedb/lancedb-darwin-arm64": "0.21.2",
-    "@lancedb/lancedb-linux-x64-gnu": "0.21.2",
-    "@lancedb/lancedb-linux-arm64-gnu": "0.21.2",
-    "@lancedb/lancedb-linux-x64-musl": "0.21.2",
-    "@lancedb/lancedb-linux-arm64-musl": "0.21.2",
-    "@lancedb/lancedb-win32-x64-msvc": "0.21.2",
-    "@lancedb/lancedb-win32-arm64-msvc": "0.21.2"
+    "@lancedb/lancedb-darwin-x64": "0.21.4-beta.0",
+    "@lancedb/lancedb-darwin-arm64": "0.21.4-beta.0",
+    "@lancedb/lancedb-linux-x64-gnu": "0.21.4-beta.0",
+    "@lancedb/lancedb-linux-arm64-gnu": "0.21.4-beta.0",
+    "@lancedb/lancedb-linux-x64-musl": "0.21.4-beta.0",
+    "@lancedb/lancedb-linux-arm64-musl": "0.21.4-beta.0",
+    "@lancedb/lancedb-win32-x64-msvc": "0.21.4-beta.0",
+    "@lancedb/lancedb-win32-arm64-msvc": "0.21.4-beta.0"
   },
   "peerDependencies": {
     "apache-arrow": ">=15.0.0 <=18.1.0"