mirror of
https://github.com/lancedb/lancedb.git
synced 2025-12-26 06:39:57 +00:00
0fd8a50bd7
This is done as setup for a PR that will fix the OpenAI dependency issue. * [x] FTS examples * [x] Setup mock openai * [x] Ran `npm audit fix` * [x] sentences embeddings test * [x] Double check formatting of docs examples
58 lines
1.6 KiB
TypeScript
58 lines
1.6 KiB
TypeScript
// SPDX-License-Identifier: Apache-2.0
|
|
// SPDX-FileCopyrightText: Copyright The LanceDB Authors
|
|
import { expect, test } from "@jest/globals";
|
|
// --8<-- [start:import]
|
|
import * as lancedb from "@lancedb/lancedb";
|
|
import { VectorQuery } from "@lancedb/lancedb";
|
|
// --8<-- [end:import]
|
|
import { withTempDirectory } from "./util.ts";
|
|
|
|
test("ann index examples", async () => {
|
|
await withTempDirectory(async (databaseDir) => {
|
|
// --8<-- [start:ingest]
|
|
const db = await lancedb.connect(databaseDir);
|
|
|
|
const data = Array.from({ length: 5_000 }, (_, i) => ({
|
|
vector: Array(128).fill(i),
|
|
id: `${i}`,
|
|
content: "",
|
|
longId: `${i}`,
|
|
}));
|
|
|
|
const table = await db.createTable("my_vectors", data, {
|
|
mode: "overwrite",
|
|
});
|
|
await table.createIndex("vector", {
|
|
config: lancedb.Index.ivfPq({
|
|
numPartitions: 10,
|
|
numSubVectors: 16,
|
|
}),
|
|
});
|
|
// --8<-- [end:ingest]
|
|
|
|
// --8<-- [start:search1]
|
|
const search = table.search(Array(128).fill(1.2)).limit(2) as VectorQuery;
|
|
const results1 = await search.nprobes(20).refineFactor(10).toArray();
|
|
// --8<-- [end:search1]
|
|
expect(results1.length).toBe(2);
|
|
|
|
// --8<-- [start:search2]
|
|
const results2 = await table
|
|
.search(Array(128).fill(1.2))
|
|
.where("id != '1141'")
|
|
.limit(2)
|
|
.toArray();
|
|
// --8<-- [end:search2]
|
|
expect(results2.length).toBe(2);
|
|
|
|
// --8<-- [start:search3]
|
|
const results3 = await table
|
|
.search(Array(128).fill(1.2))
|
|
.select(["id"])
|
|
.limit(2)
|
|
.toArray();
|
|
// --8<-- [end:search3]
|
|
expect(results3.length).toBe(2);
|
|
});
|
|
}, 100_000);
|