presubmit 0.1.0

package/vendor/lancedb/dist/index.js

@@ -0,0 +1,176 @@
+"use strict";
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: Copyright The LanceDB Authors
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.packBits = exports.rerankers = exports.Scannable = exports.PermutationBuilder = exports.permutationBuilder = exports.embedding = exports.MergeInsertBuilder = exports.OAuthFlowType = exports.OAuthHeaderProvider = exports.StaticHeaderProvider = exports.HeaderProvider = exports.Branches = 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.AutoQuery = exports.Session = exports.Job = exports.Connection = exports.VectorColumnOptions = exports.MakeArrowTableOptions = exports.makeArrowTable = exports.BranchContents = exports.TagContents = exports.Tags = exports.instrumentLanceDbMetrics = exports.NativeJsHeaderProvider = exports.MaterializedView = void 0;
+exports.tokenize = tokenize;
+exports.connect = connect;
+exports.connectNamespace = connectNamespace;
+const connection_1 = require("./connection");
+const native_js_1 = require("./native.js");
+// Re-export native header provider for use with connectWithHeaderProvider
+var materialized_view_1 = require("./materialized_view");
+Object.defineProperty(exports, "MaterializedView", { enumerable: true, get: function () { return materialized_view_1.MaterializedView; } });
+var native_js_2 = require("./native.js");
+Object.defineProperty(exports, "NativeJsHeaderProvider", { enumerable: true, get: function () { return native_js_2.JsHeaderProvider; } });
+// OpenTelemetry metrics bridge. Only the high-level entry point is public; the
+// underlying recorder/catalog/snapshot functions remain internal plumbing that
+// `otel.ts` consumes from the native module.
+var otel_1 = require("./otel");
+Object.defineProperty(exports, "instrumentLanceDbMetrics", { enumerable: true, get: function () { return otel_1.instrumentLanceDbMetrics; } });
+var native_js_3 = require("./native.js");
+Object.defineProperty(exports, "Tags", { enumerable: true, get: function () { return native_js_3.Tags; } });
+Object.defineProperty(exports, "TagContents", { enumerable: true, get: function () { return native_js_3.TagContents; } });
+Object.defineProperty(exports, "BranchContents", { enumerable: true, get: function () { return native_js_3.BranchContents; } });
+var arrow_1 = require("./arrow");
+Object.defineProperty(exports, "makeArrowTable", { enumerable: true, get: function () { return arrow_1.makeArrowTable; } });
+Object.defineProperty(exports, "MakeArrowTableOptions", { enumerable: true, get: function () { return arrow_1.MakeArrowTableOptions; } });
+Object.defineProperty(exports, "VectorColumnOptions", { enumerable: true, get: function () { return arrow_1.VectorColumnOptions; } });
+var connection_2 = require("./connection");
+Object.defineProperty(exports, "Connection", { enumerable: true, get: function () { return connection_2.Connection; } });
+var native_js_4 = require("./native.js");
+Object.defineProperty(exports, "Job", { enumerable: true, get: function () { return native_js_4.Job; } });
+Object.defineProperty(exports, "Session", { enumerable: true, get: function () { return native_js_4.Session; } });
+var query_1 = require("./query");
+Object.defineProperty(exports, "AutoQuery", { enumerable: true, get: function () { return query_1.AutoQuery; } });
+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; } });
+Object.defineProperty(exports, "BoostQuery", { enumerable: true, get: function () { return query_1.BoostQuery; } });
+Object.defineProperty(exports, "MultiMatchQuery", { enumerable: true, get: function () { return query_1.MultiMatchQuery; } });
+Object.defineProperty(exports, "BooleanQuery", { enumerable: true, get: function () { return query_1.BooleanQuery; } });
+Object.defineProperty(exports, "FullTextQueryType", { enumerable: true, get: function () { return query_1.FullTextQueryType; } });
+Object.defineProperty(exports, "Operator", { enumerable: true, get: function () { return query_1.Operator; } });
+Object.defineProperty(exports, "Occur", { enumerable: true, get: function () { return query_1.Occur; } });
+var indices_1 = require("./indices");
+Object.defineProperty(exports, "Index", { enumerable: true, get: function () { return indices_1.Index; } });
+var table_1 = require("./table");
+Object.defineProperty(exports, "Table", { enumerable: true, get: function () { return table_1.Table; } });
+Object.defineProperty(exports, "Branches", { enumerable: true, get: function () { return table_1.Branches; } });
+var header_1 = require("./header");
+Object.defineProperty(exports, "HeaderProvider", { enumerable: true, get: function () { return header_1.HeaderProvider; } });
+Object.defineProperty(exports, "StaticHeaderProvider", { enumerable: true, get: function () { return header_1.StaticHeaderProvider; } });
+Object.defineProperty(exports, "OAuthHeaderProvider", { enumerable: true, get: function () { return header_1.OAuthHeaderProvider; } });
+var oauth_1 = require("./oauth");
+Object.defineProperty(exports, "OAuthFlowType", { enumerable: true, get: function () { return oauth_1.OAuthFlowType; } });
+var merge_1 = require("./merge");
+Object.defineProperty(exports, "MergeInsertBuilder", { enumerable: true, get: function () { return merge_1.MergeInsertBuilder; } });
+exports.embedding = require("./embedding");
+var permutation_1 = require("./permutation");
+Object.defineProperty(exports, "permutationBuilder", { enumerable: true, get: function () { return permutation_1.permutationBuilder; } });
+Object.defineProperty(exports, "PermutationBuilder", { enumerable: true, get: function () { return permutation_1.PermutationBuilder; } });
+var scannable_1 = require("./scannable");
+Object.defineProperty(exports, "Scannable", { enumerable: true, get: function () { return scannable_1.Scannable; } });
+exports.rerankers = require("./rerankers");
+var util_1 = require("./util");
+Object.defineProperty(exports, "packBits", { enumerable: true, get: function () { return util_1.packBits; } });
+/**
+ * Tokenize a full-text search query using an explicit tokenizer.
+ *
+ * This does not require a table or FTS index. The tokenizer options match
+ * {@link Index.fts}.
+ */
+async function tokenize(query, options) {
+    return await (0, native_js_1.tokenize)(query, options?.baseTokenizer, options?.language, options?.maxTokenLength, options?.lowercase, options?.stem, options?.removeStopWords, options?.customStopWords, options?.asciiFolding, options?.ngramMinLength, options?.ngramMaxLength, options?.prefixOnly);
+}
+async function connect(uriOrOptions, optionsOrSession, sessionOrHeaderProvider, headerProvider) {
+    let uri;
+    let finalOptions = {};
+    let finalHeaderProvider;
+    if (typeof uriOrOptions !== "string") {
+        // First overload: connect(options)
+        const { uri: uri_, ...opts } = uriOrOptions;
+        uri = uri_;
+        finalOptions = opts;
+    }
+    else {
+        // Second overload: connect(uri, options?, session?, headerProvider?)
+        uri = uriOrOptions;
+        // Handle optionsOrSession parameter
+        if (optionsOrSession && "inner" in optionsOrSession) {
+            // Second param is session, so no options provided
+            finalOptions = {};
+        }
+        else {
+            // Second param is options
+            finalOptions = optionsOrSession || {};
+        }
+        // Handle sessionOrHeaderProvider parameter
+        if (sessionOrHeaderProvider &&
+            (typeof sessionOrHeaderProvider === "function" ||
+                "getHeaders" in sessionOrHeaderProvider)) {
+            // Third param is header provider
+            finalHeaderProvider = sessionOrHeaderProvider;
+        }
+        else {
+            // Third param is session, header provider is fourth param
+            finalHeaderProvider = headerProvider;
+        }
+    }
+    if (!uri) {
+        throw new Error("uri is required");
+    }
+    finalOptions = finalOptions ?? {};
+    finalOptions.storageOptions = (0, connection_1.cleanseStorageOptions)(finalOptions.storageOptions);
+    // Create native header provider if one was provided
+    let nativeProvider;
+    if (finalHeaderProvider) {
+        if (typeof finalHeaderProvider === "function") {
+            nativeProvider = new native_js_1.JsHeaderProvider(async () => finalHeaderProvider());
+        }
+        else if (finalHeaderProvider &&
+            typeof finalHeaderProvider.getHeaders === "function") {
+            nativeProvider = new native_js_1.JsHeaderProvider(async () => finalHeaderProvider.getHeaders());
+        }
+    }
+    const nativeConn = await native_js_1.Connection.new(uri, finalOptions, nativeProvider);
+    return new connection_1.LocalConnection(nativeConn);
+}
+function dirConfigToProperties(config) {
+    // Spread the whole input so that unknown keys (e.g. a raw `manifest_enabled`
+    // passed via the dynamic-impl path) flow through instead of being dropped.
+    // Typed transformations layer on top.
+    const { manifestEnabled, extraProperties, ...rest } = config;
+    const properties = {
+        ...(extraProperties ?? {}),
+        ...rest,
+    };
+    if (manifestEnabled !== undefined) {
+        properties.manifest_enabled = String(manifestEnabled);
+    }
+    return properties;
+}
+function restConfigToProperties(config) {
+    const { headers, extraProperties, ...rest } = config;
+    const properties = {
+        ...(extraProperties ?? {}),
+        ...rest,
+    };
+    if (headers) {
+        for (const [name, value] of Object.entries(headers)) {
+            properties[`headers.${name}`] = value;
+        }
+    }
+    return properties;
+}
+async function connectNamespace(implName, configOrProperties, options) {
+    let properties;
+    if (implName === "dir") {
+        properties = dirConfigToProperties(configOrProperties);
+    }
+    else if (implName === "rest") {
+        properties = restConfigToProperties(configOrProperties);
+    }
+    else {
+        properties = configOrProperties;
+    }
+    const finalOptions = (options ??
+        {});
+    finalOptions.storageOptions = (0, connection_1.cleanseStorageOptions)(finalOptions.storageOptions);
+    const nativeConn = await native_js_1.Connection.newWithNamespace(implName, properties, finalOptions);
+    return new connection_1.LocalConnection(nativeConn);
+}