mirror of
https://github.com/lancedb/lancedb.git
synced 2025-12-23 05:19:58 +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
43 lines
1.2 KiB
TypeScript
43 lines
1.2 KiB
TypeScript
// SPDX-License-Identifier: Apache-2.0
|
|
// SPDX-FileCopyrightText: Copyright The LanceDB Authors
|
|
import { expect, test } from "@jest/globals";
|
|
import * as lancedb from "@lancedb/lancedb";
|
|
import { withTempDirectory } from "./util.ts";
|
|
|
|
test("filtering examples", async () => {
|
|
await withTempDirectory(async (databaseDir) => {
|
|
const db = await lancedb.connect(databaseDir);
|
|
|
|
const data = Array.from({ length: 10_000 }, (_, i) => ({
|
|
vector: Array(1536).fill(i),
|
|
id: i,
|
|
item: `item ${i}`,
|
|
strId: `${i}`,
|
|
}));
|
|
|
|
const tbl = await db.createTable("myVectors", data, { mode: "overwrite" });
|
|
|
|
// --8<-- [start:search]
|
|
const _result = await tbl
|
|
.search(Array(1536).fill(0.5))
|
|
.limit(1)
|
|
.where("id = 10")
|
|
.toArray();
|
|
// --8<-- [end:search]
|
|
|
|
// --8<-- [start:vec_search]
|
|
const result = await (
|
|
tbl.search(Array(1536).fill(0)) as lancedb.VectorQuery
|
|
)
|
|
.where("(item IN ('item 0', 'item 2')) AND (id > 10)")
|
|
.postfilter()
|
|
.toArray();
|
|
// --8<-- [end:vec_search]
|
|
expect(result.length).toBe(0);
|
|
|
|
// --8<-- [start:sql_search]
|
|
await tbl.query().where("id = 10").limit(10).toArray();
|
|
// --8<-- [end:sql_search]
|
|
});
|
|
});
|