@anvia/lancedb 1.0.0-rc.2 → 1.0.0-rc.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.d.ts CHANGED
@@ -1,5 +1,5 @@
1
1
  import { VectorMetadata } from '@anvia/core/embeddings';
2
- import { VectorMetric, VectorStore, VectorStoreUpsertOptions, VectorSearchRequest, VectorSearchResult, VectorFilter } from '@anvia/core/vector-store';
2
+ import { VectorMetric, VectorStore, VectorStoreUpsertOptions, VectorSearchRequest, VectorSearchResult } from '@anvia/core/vector-store';
3
3
 
4
4
  type LanceDBTableLike = {
5
5
  add(rows: Record<string, unknown>[], options?: Record<string, unknown>): Promise<unknown>;
@@ -58,6 +58,4 @@ declare class LanceDBVectorClient {
58
58
  private assertOpen;
59
59
  }
60
60
 
61
- declare function filterToLanceExpr(filter: VectorFilter | undefined): string | undefined;
62
-
63
- export { type LanceDBConnectionLike, type LanceDBTableLike, LanceDBVectorClient, type LanceDBVectorClientOptions, LanceDBVectorStore, type LanceDBVectorStoreOptions, filterToLanceExpr };
61
+ export { type LanceDBConnectionLike, type LanceDBTableLike, LanceDBVectorClient, type LanceDBVectorClientOptions, LanceDBVectorStore, type LanceDBVectorStoreOptions };
package/dist/index.js CHANGED
@@ -92,59 +92,8 @@ function metadataFromColumns(record) {
92
92
  return Object.keys(metadata).length === 0 ? void 0 : metadata;
93
93
  }
94
94
 
95
- // src/filters.ts
96
- function filterToLanceExpr(filter) {
97
- if (filter === void 0) {
98
- return void 0;
99
- }
100
- return translateFilter(filter);
101
- }
102
- function translateFilter(filter) {
103
- switch (filter.type) {
104
- case "eq": {
105
- if (filter.value === null) {
106
- const key = escapeSql(filter.key);
107
- return `json_contains(${metadataColumn}, '${key}') AND json_get_json(${metadataColumn}, '${key}') = 'null'`;
108
- }
109
- const value = metadataValue(filter.key, filter.value);
110
- if (typeof filter.value === "string") {
111
- return `${value} = '${escapeSql(filter.value)}'`;
112
- }
113
- if (typeof filter.value === "boolean") {
114
- return `${value} = ${filter.value ? "TRUE" : "FALSE"}`;
115
- }
116
- return `${value} = ${filter.value}`;
117
- }
118
- case "gt": {
119
- assertNumericValue(filter.value, filter.type);
120
- return `${metadataValue(filter.key, filter.value)} > ${filter.value}`;
121
- }
122
- case "lt": {
123
- assertNumericValue(filter.value, filter.type);
124
- return `${metadataValue(filter.key, filter.value)} < ${filter.value}`;
125
- }
126
- case "and":
127
- return filter.filters.map((f) => `(${translateFilter(f)})`).join(" AND ");
128
- case "or":
129
- return filter.filters.map((f) => `(${translateFilter(f)})`).join(" OR ");
130
- }
131
- }
132
- function metadataValue(key, value) {
133
- const escapedKey = escapeSql(key);
134
- if (typeof value === "string") return `json_get_str(${metadataColumn}, '${escapedKey}')`;
135
- if (typeof value === "boolean") return `json_get_bool(${metadataColumn}, '${escapedKey}')`;
136
- return `json_get_float(${metadataColumn}, '${escapedKey}')`;
137
- }
138
- function assertNumericValue(value, operator) {
139
- if (typeof value !== "number") {
140
- throw new TypeError(`LanceDB ${operator} filters require numeric metadata values`);
141
- }
142
- }
143
- function escapeSql(value) {
144
- return value.replace(/'/g, "''");
145
- }
146
-
147
95
  // src/store.ts
96
+ import { matchesVectorFilter } from "@anvia/core/vector-store";
148
97
  var LanceDBVectorStore = class {
149
98
  constructor(owner, options) {
150
99
  this.owner = owner;
@@ -209,9 +158,7 @@ var LanceDBVectorStore = class {
209
158
  const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);
210
159
  let candidateLimit = request.topK;
211
160
  for (; ; ) {
212
- let query = table.search(request.vector).distanceType(metricName(this.options.metric)).limit(candidateLimit);
213
- const filter = filterToLanceExpr(request.filter);
214
- if (filter !== void 0) query = query.where(filter);
161
+ const query = table.search(request.vector).distanceType(metricName(this.options.metric)).limit(candidateLimit);
215
162
  const response = await query.toArray();
216
163
  throwIfAborted(request.abortSignal);
217
164
  const unfiltered = parseQueryResults(response, void 0, this.options.metric);
@@ -225,7 +172,7 @@ var LanceDBVectorStore = class {
225
172
  }
226
173
  };
227
174
  function selectResults(results, request) {
228
- return results.filter((result) => request.minScore === void 0 || result.score >= request.minScore).sort((left, right) => right.score - left.score).slice(0, request.topK);
175
+ return results.filter((result) => matchesVectorFilter(result.metadata, request.filter)).filter((result) => request.minScore === void 0 || result.score >= request.minScore).sort((left, right) => right.score - left.score).slice(0, request.topK);
229
176
  }
230
177
  function hasResultBelowMinimum(results, minScore) {
231
178
  return minScore !== void 0 && results.some((result) => result.score < minScore);
@@ -313,7 +260,6 @@ var LanceDBVectorClient = class {
313
260
  };
314
261
  export {
315
262
  LanceDBVectorClient,
316
- LanceDBVectorStore,
317
- filterToLanceExpr
263
+ LanceDBVectorStore
318
264
  };
319
265
  //# sourceMappingURL=index.js.map
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/helpers.ts","../src/types.ts","../src/filters.ts","../src/store.ts","../src/client.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\nimport type { EmbeddedDocument, VectorMetadata } from \"@anvia/core/embeddings\";\nimport type { VectorSearchResult } from \"@anvia/core/vector-store\";\nimport type { LanceDBVectorStoreOptions } from \"./types.js\";\nimport {\n documentColumn,\n documentIdColumn,\n metadataColumn,\n reservedColumnPrefix,\n rowIdColumn,\n vectorColumn,\n} from \"./types.js\";\n\nexport function assertNoReservedMetadata(metadata: VectorMetadata | undefined): void {\n for (const key of Object.keys(metadata ?? {})) {\n if (key.startsWith(reservedColumnPrefix)) {\n throw new Error(`Metadata key ${key} is reserved for Anvia LanceDB columns`);\n }\n }\n}\n\nexport function rowId(id: string): string {\n return createHash(\"sha256\").update(id).digest(\"hex\").slice(0, 32);\n}\n\nexport function serializeDocument(document: unknown): string {\n return typeof document === \"string\" ? document : JSON.stringify(document);\n}\n\nexport function parseDocument<T>(document: unknown): T {\n if (document === null || document === undefined) {\n return \"\" as T;\n }\n if (typeof document !== \"string\") {\n return document as T;\n }\n try {\n return JSON.parse(document) as T;\n } catch {\n return document as T;\n }\n}\n\nexport function lanceRows<T, Metadata extends VectorMetadata>(\n document: EmbeddedDocument<T, Metadata>,\n): Array<Record<string, unknown>> {\n if (document.embeddings.length === 0) {\n throw new Error(`Document ${document.id} has no embeddings`);\n }\n assertNoReservedMetadata(document.metadata);\n\n return document.embeddings.map((embedding, index) => {\n const logicalId =\n document.embeddings.length === 1 ? document.id : `${document.id}#embedding:${index}`;\n const row: Record<string, unknown> = {\n [rowIdColumn]: rowId(logicalId),\n [documentIdColumn]: document.id,\n [documentColumn]: serializeDocument(document.document),\n [metadataColumn]: JSON.stringify(document.metadata ?? {}),\n [vectorColumn]: embedding.vector,\n };\n return row;\n });\n}\n\nexport function parseQueryResults<T, Metadata extends VectorMetadata>(\n response: unknown[],\n minScore: number | undefined,\n metric: LanceDBVectorStoreOptions[\"metric\"],\n): Array<VectorSearchResult<T, Metadata>> {\n const byId = new Map<string, VectorSearchResult<T, Metadata>>();\n\n for (const row of response) {\n const record = row as Record<string, unknown>;\n const distance = typeof record._distance === \"number\" ? record._distance : 0;\n const score = metric === undefined || metric === \"cosine\" ? 1 - distance : -distance;\n\n if (minScore !== undefined && score < minScore) {\n continue;\n }\n\n const id = String(record[documentIdColumn] ?? \"\");\n const result: VectorSearchResult<T, Metadata> = {\n id,\n score,\n document: parseDocument(record[documentColumn]),\n };\n const metadata = metadataFromColumns<Metadata>(record);\n if (metadata !== undefined) {\n result.metadata = metadata;\n }\n const current = byId.get(id);\n if (current === undefined || result.score > current.score) {\n byId.set(id, result);\n }\n }\n\n return [...byId.values()];\n}\n\nexport async function defaultLanceDBConnection(\n uri?: string,\n): Promise<import(\"./types.js\").LanceDBConnectionLike> {\n const lancedb = await import(\"@lancedb/lancedb\");\n return lancedb.connect(\n uri ?? \"~/.anvia/lancedb\",\n ) as unknown as import(\"./types.js\").LanceDBConnectionLike;\n}\n\nfunction metadataFromColumns<Metadata extends VectorMetadata>(\n record: Record<string, unknown>,\n): Metadata | undefined {\n const raw = record[metadataColumn];\n if (raw === null || raw === undefined) return undefined;\n const metadata = (typeof raw === \"string\" ? JSON.parse(raw) : raw) as Metadata;\n return Object.keys(metadata).length === 0 ? undefined : metadata;\n}\n","import type { VectorMetric } from \"@anvia/core/vector-store\";\n\nexport const documentIdColumn = \"__anvia_document_id\";\nexport const documentColumn = \"__anvia_document\";\nexport const metadataColumn = \"__anvia_metadata\";\nexport const vectorColumn = \"__anvia_vector\";\nexport const rowIdColumn = \"__anvia_id\";\nexport const reservedColumnPrefix = \"__anvia_\";\n\nexport type LanceDBTableLike = {\n add(rows: Record<string, unknown>[], options?: Record<string, unknown>): Promise<unknown>;\n search(vector: number[]): LanceDBQueryLike;\n countRows(): Promise<number>;\n delete(filter: string): Promise<unknown>;\n schema?(): Promise<{ fields?: Array<{ name?: string; type?: unknown }> }>;\n};\n\nexport type LanceDBQueryLike = {\n limit(n: number): LanceDBQueryLike;\n where(predicate: string): LanceDBQueryLike;\n distanceType(metric: \"cosine\" | \"l2\" | \"dot\"): LanceDBQueryLike;\n toArray(): Promise<unknown[]>;\n};\n\nexport type LanceDBConnectionLike = {\n openTable(name: string): Promise<LanceDBTableLike>;\n tableNames(): Promise<string[]>;\n createTable(name: string, data: Record<string, unknown>[]): Promise<LanceDBTableLike>;\n createEmptyTable?(\n name: string,\n schema: unknown,\n options?: Record<string, unknown>,\n ): Promise<LanceDBTableLike>;\n close?(): Promise<unknown> | unknown;\n};\n\nexport type LanceDBVectorClientOptions = {\n client?: LanceDBConnectionLike | undefined;\n uri?: string | undefined;\n};\n\nexport type LanceDBVectorStoreOptions = {\n tableName: string;\n dimensions: number;\n metric?: VectorMetric | undefined;\n};\n","import type { VectorFilter } from \"@anvia/core/vector-store\";\nimport { metadataColumn } from \"./types.js\";\n\nexport function filterToLanceExpr(filter: VectorFilter | undefined): string | undefined {\n if (filter === undefined) {\n return undefined;\n }\n\n return translateFilter(filter);\n}\n\nfunction translateFilter(filter: VectorFilter): string {\n switch (filter.type) {\n case \"eq\": {\n if (filter.value === null) {\n const key = escapeSql(filter.key);\n return `json_contains(${metadataColumn}, '${key}') AND json_get_json(${metadataColumn}, '${key}') = 'null'`;\n }\n const value = metadataValue(filter.key, filter.value);\n if (typeof filter.value === \"string\") {\n return `${value} = '${escapeSql(filter.value)}'`;\n }\n if (typeof filter.value === \"boolean\") {\n return `${value} = ${filter.value ? \"TRUE\" : \"FALSE\"}`;\n }\n return `${value} = ${filter.value}`;\n }\n case \"gt\": {\n assertNumericValue(filter.value, filter.type);\n return `${metadataValue(filter.key, filter.value)} > ${filter.value}`;\n }\n case \"lt\": {\n assertNumericValue(filter.value, filter.type);\n return `${metadataValue(filter.key, filter.value)} < ${filter.value}`;\n }\n case \"and\":\n return filter.filters.map((f) => `(${translateFilter(f)})`).join(\" AND \");\n case \"or\":\n return filter.filters.map((f) => `(${translateFilter(f)})`).join(\" OR \");\n }\n}\n\nfunction metadataValue(key: string, value: string | number | boolean | null): string {\n const escapedKey = escapeSql(key);\n if (typeof value === \"string\") return `json_get_str(${metadataColumn}, '${escapedKey}')`;\n if (typeof value === \"boolean\") return `json_get_bool(${metadataColumn}, '${escapedKey}')`;\n return `json_get_float(${metadataColumn}, '${escapedKey}')`;\n}\n\nfunction assertNumericValue(value: unknown, operator: \"gt\" | \"lt\"): asserts value is number {\n if (typeof value !== \"number\") {\n throw new TypeError(`LanceDB ${operator} filters require numeric metadata values`);\n }\n}\n\nfunction escapeSql(value: string): string {\n return value.replace(/'/g, \"''\");\n}\n","import type { EmbeddedDocument, VectorMetadata } from \"@anvia/core/embeddings\";\nimport type {\n VectorSearchRequest,\n VectorSearchResult,\n VectorStore,\n VectorStoreUpsertOptions,\n} from \"@anvia/core/vector-store\";\nimport type { LanceDBVectorClient } from \"./client.js\";\nimport { filterToLanceExpr } from \"./filters.js\";\nimport { lanceRows, parseQueryResults } from \"./helpers.js\";\nimport {\n documentColumn,\n documentIdColumn,\n type LanceDBVectorStoreOptions,\n metadataColumn,\n rowIdColumn,\n vectorColumn,\n} from \"./types.js\";\n\nexport class LanceDBVectorStore<T, Metadata extends VectorMetadata = VectorMetadata>\n implements VectorStore<T, Metadata>\n{\n constructor(\n private readonly owner: LanceDBVectorClient,\n readonly options: LanceDBVectorStoreOptions,\n ) {\n assertDimensions(options.dimensions);\n }\n\n async ensure(): Promise<void> {\n const connection = await this.owner.nativeClient();\n if (!(await connection.tableNames()).includes(this.options.tableName)) {\n if (connection.createEmptyTable === undefined) {\n throw new TypeError(\"LanceDB provisioning requires createEmptyTable(...).\");\n }\n const { Field, FixedSizeList, Float32, Schema, Utf8 } = await import(\"apache-arrow\");\n await connection.createEmptyTable(\n this.options.tableName,\n new Schema([\n new Field(rowIdColumn, new Utf8(), false),\n new Field(documentIdColumn, new Utf8(), false),\n new Field(documentColumn, new Utf8(), false),\n new Field(metadataColumn, new Utf8(), false),\n new Field(\n vectorColumn,\n new FixedSizeList(this.options.dimensions, new Field(\"item\", new Float32(), true)),\n false,\n ),\n ]),\n );\n }\n await this.validate();\n }\n\n async validate(): Promise<void> {\n const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);\n if (table.schema === undefined) return;\n const schema = await table.schema();\n const vectorField = schema.fields?.find((field) => field.name === vectorColumn);\n if (vectorField === undefined) {\n throw new Error(`LanceDB table ${this.options.tableName} is missing ${vectorColumn}`);\n }\n if (!schema.fields?.some((field) => field.name === metadataColumn)) {\n throw new Error(`LanceDB table ${this.options.tableName} is missing ${metadataColumn}`);\n }\n const listSize = (vectorField.type as { listSize?: unknown } | undefined)?.listSize;\n if (typeof listSize === \"number\" && listSize !== this.options.dimensions) {\n throw new Error(\n `LanceDB table ${this.options.tableName} has ${listSize} dimensions; expected ${this.options.dimensions}`,\n );\n }\n }\n\n async upsert(options: VectorStoreUpsertOptions<T, Metadata>): Promise<void> {\n validateDocuments(options.documents, this.options.dimensions);\n if (options.documents.length === 0) return;\n const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);\n const ids = options.documents.map((document) => sqlString(document.id)).join(\", \");\n await table.delete(`${documentIdColumn} IN (${ids})`);\n const rows = options.documents.flatMap((document) => lanceRows(document));\n if (rows.length > 0) await table.add(rows, options.providerOptions);\n }\n\n async search(request: VectorSearchRequest): Promise<Array<VectorSearchResult<T, Metadata>>> {\n validateSearchRequest(request, this.options.dimensions);\n const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);\n let candidateLimit = request.topK;\n for (;;) {\n let query = table\n .search(request.vector)\n .distanceType(metricName(this.options.metric))\n .limit(candidateLimit);\n const filter = filterToLanceExpr(request.filter);\n if (filter !== undefined) query = query.where(filter);\n const response = await query.toArray();\n throwIfAborted(request.abortSignal);\n const unfiltered = parseQueryResults<T, Metadata>(response, undefined, this.options.metric);\n const results = selectResults(unfiltered, request);\n if (\n results.length >= request.topK ||\n response.length < candidateLimit ||\n hasResultBelowMinimum(unfiltered, request.minScore)\n )\n return results;\n const nextLimit = nextCandidateLimit(candidateLimit);\n if (nextLimit === candidateLimit) return results;\n candidateLimit = nextLimit;\n }\n }\n}\n\nfunction selectResults<T, Metadata extends VectorMetadata>(\n results: Array<VectorSearchResult<T, Metadata>>,\n request: VectorSearchRequest,\n): Array<VectorSearchResult<T, Metadata>> {\n return results\n .filter((result) => request.minScore === undefined || result.score >= request.minScore)\n .sort((left, right) => right.score - left.score)\n .slice(0, request.topK);\n}\n\nfunction hasResultBelowMinimum<T, Metadata extends VectorMetadata>(\n results: Array<VectorSearchResult<T, Metadata>>,\n minScore: number | undefined,\n): boolean {\n return minScore !== undefined && results.some((result) => result.score < minScore);\n}\n\nfunction nextCandidateLimit(current: number): number {\n return current > Math.floor(Number.MAX_SAFE_INTEGER / 2) ? current : current * 2;\n}\n\nfunction sqlString(value: string): string {\n return `'${value.replaceAll(\"'\", \"''\")}'`;\n}\n\nfunction throwIfAborted(signal: AbortSignal | undefined): void {\n if (signal?.aborted) {\n const error = new Error(\"The operation was aborted.\");\n error.name = \"AbortError\";\n throw error;\n }\n}\n\nfunction assertDimensions(dimensions: number): void {\n if (!Number.isSafeInteger(dimensions) || dimensions < 1)\n throw new RangeError(\"Vector dimensions must be a positive safe integer.\");\n}\n\nfunction validateDocuments<T, Metadata extends VectorMetadata>(\n documents: Array<EmbeddedDocument<T, Metadata>>,\n dimensions: number,\n): void {\n const ids = new Set<string>();\n for (const document of documents) {\n if (ids.has(document.id)) throw new TypeError(`Duplicate vector document id: ${document.id}`);\n ids.add(document.id);\n if (document.embeddings.length === 0)\n throw new TypeError(`Vector document ${document.id} must contain at least one embedding.`);\n for (const embedding of document.embeddings) {\n if (embedding.vector.length !== dimensions)\n throw new Error(`Vector dimension mismatch for document ${document.id}`);\n if (!embedding.vector.every(Number.isFinite))\n throw new Error(`Vector for document ${document.id} must contain only finite numbers`);\n }\n }\n}\n\nfunction metricName(metric: LanceDBVectorStoreOptions[\"metric\"]): \"cosine\" | \"l2\" | \"dot\" {\n if (metric === \"euclidean\") return \"l2\";\n if (metric === \"dotProduct\") return \"dot\";\n return \"cosine\";\n}\n\nfunction validateSearchRequest(request: VectorSearchRequest, dimensions: number): void {\n throwIfAborted(request.abortSignal);\n if (!Number.isSafeInteger(request.topK) || request.topK < 1)\n throw new RangeError(\"Vector search topK must be a positive safe integer.\");\n if (request.minScore !== undefined && !Number.isFinite(request.minScore))\n throw new RangeError(\"Vector search minScore must be a finite number.\");\n if (request.vector.length !== dimensions) {\n throw new Error(\n `Vector dimension mismatch: expected ${dimensions} dimensions but received ${request.vector.length} for query`,\n );\n }\n if (!request.vector.every(Number.isFinite))\n throw new Error(\"Vector search query must contain only finite numbers\");\n}\n","import type { VectorMetadata } from \"@anvia/core/embeddings\";\nimport { defaultLanceDBConnection } from \"./helpers.js\";\nimport { LanceDBVectorStore } from \"./store.js\";\nimport type {\n LanceDBConnectionLike,\n LanceDBVectorClientOptions,\n LanceDBVectorStoreOptions,\n} from \"./types.js\";\n\nexport class LanceDBVectorClient {\n private readonly injected: LanceDBConnectionLike | undefined;\n private clientPromise: Promise<LanceDBConnectionLike> | undefined;\n private closed = false;\n\n constructor(private readonly options: LanceDBVectorClientOptions = {}) {\n this.injected = options.client;\n }\n\n vectorStore<T, Metadata extends VectorMetadata = VectorMetadata>(\n options: LanceDBVectorStoreOptions,\n ): LanceDBVectorStore<T, Metadata> {\n this.assertOpen();\n return new LanceDBVectorStore<T, Metadata>(this, options);\n }\n\n nativeClient(): Promise<LanceDBConnectionLike> {\n this.assertOpen();\n this.clientPromise ??=\n this.injected === undefined\n ? defaultLanceDBConnection(this.options.uri)\n : Promise.resolve(this.injected);\n return this.clientPromise;\n }\n\n async close(): Promise<void> {\n if (this.closed) return;\n this.closed = true;\n if (this.injected !== undefined || this.clientPromise === undefined) return;\n await (await this.clientPromise).close?.();\n }\n\n private assertOpen(): void {\n if (this.closed) throw new Error(\"LanceDBVectorClient is closed.\");\n }\n}\n"],"mappings":";AAAA,SAAS,kBAAkB;;;ACEpB,IAAM,mBAAmB;AACzB,IAAM,iBAAiB;AACvB,IAAM,iBAAiB;AACvB,IAAM,eAAe;AACrB,IAAM,cAAc;AACpB,IAAM,uBAAuB;;;ADM7B,SAAS,yBAAyB,UAA4C;AACnF,aAAW,OAAO,OAAO,KAAK,YAAY,CAAC,CAAC,GAAG;AAC7C,QAAI,IAAI,WAAW,oBAAoB,GAAG;AACxC,YAAM,IAAI,MAAM,gBAAgB,GAAG,wCAAwC;AAAA,IAC7E;AAAA,EACF;AACF;AAEO,SAAS,MAAM,IAAoB;AACxC,SAAO,WAAW,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,KAAK,EAAE,MAAM,GAAG,EAAE;AAClE;AAEO,SAAS,kBAAkB,UAA2B;AAC3D,SAAO,OAAO,aAAa,WAAW,WAAW,KAAK,UAAU,QAAQ;AAC1E;AAEO,SAAS,cAAiB,UAAsB;AACrD,MAAI,aAAa,QAAQ,aAAa,QAAW;AAC/C,WAAO;AAAA,EACT;AACA,MAAI,OAAO,aAAa,UAAU;AAChC,WAAO;AAAA,EACT;AACA,MAAI;AACF,WAAO,KAAK,MAAM,QAAQ;AAAA,EAC5B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEO,SAAS,UACd,UACgC;AAChC,MAAI,SAAS,WAAW,WAAW,GAAG;AACpC,UAAM,IAAI,MAAM,YAAY,SAAS,EAAE,oBAAoB;AAAA,EAC7D;AACA,2BAAyB,SAAS,QAAQ;AAE1C,SAAO,SAAS,WAAW,IAAI,CAAC,WAAW,UAAU;AACnD,UAAM,YACJ,SAAS,WAAW,WAAW,IAAI,SAAS,KAAK,GAAG,SAAS,EAAE,cAAc,KAAK;AACpF,UAAM,MAA+B;AAAA,MACnC,CAAC,WAAW,GAAG,MAAM,SAAS;AAAA,MAC9B,CAAC,gBAAgB,GAAG,SAAS;AAAA,MAC7B,CAAC,cAAc,GAAG,kBAAkB,SAAS,QAAQ;AAAA,MACrD,CAAC,cAAc,GAAG,KAAK,UAAU,SAAS,YAAY,CAAC,CAAC;AAAA,MACxD,CAAC,YAAY,GAAG,UAAU;AAAA,IAC5B;AACA,WAAO;AAAA,EACT,CAAC;AACH;AAEO,SAAS,kBACd,UACA,UACA,QACwC;AACxC,QAAM,OAAO,oBAAI,IAA6C;AAE9D,aAAW,OAAO,UAAU;AAC1B,UAAM,SAAS;AACf,UAAM,WAAW,OAAO,OAAO,cAAc,WAAW,OAAO,YAAY;AAC3E,UAAM,QAAQ,WAAW,UAAa,WAAW,WAAW,IAAI,WAAW,CAAC;AAE5E,QAAI,aAAa,UAAa,QAAQ,UAAU;AAC9C;AAAA,IACF;AAEA,UAAM,KAAK,OAAO,OAAO,gBAAgB,KAAK,EAAE;AAChD,UAAM,SAA0C;AAAA,MAC9C;AAAA,MACA;AAAA,MACA,UAAU,cAAc,OAAO,cAAc,CAAC;AAAA,IAChD;AACA,UAAM,WAAW,oBAA8B,MAAM;AACrD,QAAI,aAAa,QAAW;AAC1B,aAAO,WAAW;AAAA,IACpB;AACA,UAAM,UAAU,KAAK,IAAI,EAAE;AAC3B,QAAI,YAAY,UAAa,OAAO,QAAQ,QAAQ,OAAO;AACzD,WAAK,IAAI,IAAI,MAAM;AAAA,IACrB;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,KAAK,OAAO,CAAC;AAC1B;AAEA,eAAsB,yBACpB,KACqD;AACrD,QAAM,UAAU,MAAM,OAAO,kBAAkB;AAC/C,SAAO,QAAQ;AAAA,IACb,OAAO;AAAA,EACT;AACF;AAEA,SAAS,oBACP,QACsB;AACtB,QAAM,MAAM,OAAO,cAAc;AACjC,MAAI,QAAQ,QAAQ,QAAQ,OAAW,QAAO;AAC9C,QAAM,WAAY,OAAO,QAAQ,WAAW,KAAK,MAAM,GAAG,IAAI;AAC9D,SAAO,OAAO,KAAK,QAAQ,EAAE,WAAW,IAAI,SAAY;AAC1D;;;AEjHO,SAAS,kBAAkB,QAAsD;AACtF,MAAI,WAAW,QAAW;AACxB,WAAO;AAAA,EACT;AAEA,SAAO,gBAAgB,MAAM;AAC/B;AAEA,SAAS,gBAAgB,QAA8B;AACrD,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK,MAAM;AACT,UAAI,OAAO,UAAU,MAAM;AACzB,cAAM,MAAM,UAAU,OAAO,GAAG;AAChC,eAAO,iBAAiB,cAAc,MAAM,GAAG,wBAAwB,cAAc,MAAM,GAAG;AAAA,MAChG;AACA,YAAM,QAAQ,cAAc,OAAO,KAAK,OAAO,KAAK;AACpD,UAAI,OAAO,OAAO,UAAU,UAAU;AACpC,eAAO,GAAG,KAAK,OAAO,UAAU,OAAO,KAAK,CAAC;AAAA,MAC/C;AACA,UAAI,OAAO,OAAO,UAAU,WAAW;AACrC,eAAO,GAAG,KAAK,MAAM,OAAO,QAAQ,SAAS,OAAO;AAAA,MACtD;AACA,aAAO,GAAG,KAAK,MAAM,OAAO,KAAK;AAAA,IACnC;AAAA,IACA,KAAK,MAAM;AACT,yBAAmB,OAAO,OAAO,OAAO,IAAI;AAC5C,aAAO,GAAG,cAAc,OAAO,KAAK,OAAO,KAAK,CAAC,MAAM,OAAO,KAAK;AAAA,IACrE;AAAA,IACA,KAAK,MAAM;AACT,yBAAmB,OAAO,OAAO,OAAO,IAAI;AAC5C,aAAO,GAAG,cAAc,OAAO,KAAK,OAAO,KAAK,CAAC,MAAM,OAAO,KAAK;AAAA,IACrE;AAAA,IACA,KAAK;AACH,aAAO,OAAO,QAAQ,IAAI,CAAC,MAAM,IAAI,gBAAgB,CAAC,CAAC,GAAG,EAAE,KAAK,OAAO;AAAA,IAC1E,KAAK;AACH,aAAO,OAAO,QAAQ,IAAI,CAAC,MAAM,IAAI,gBAAgB,CAAC,CAAC,GAAG,EAAE,KAAK,MAAM;AAAA,EAC3E;AACF;AAEA,SAAS,cAAc,KAAa,OAAiD;AACnF,QAAM,aAAa,UAAU,GAAG;AAChC,MAAI,OAAO,UAAU,SAAU,QAAO,gBAAgB,cAAc,MAAM,UAAU;AACpF,MAAI,OAAO,UAAU,UAAW,QAAO,iBAAiB,cAAc,MAAM,UAAU;AACtF,SAAO,kBAAkB,cAAc,MAAM,UAAU;AACzD;AAEA,SAAS,mBAAmB,OAAgB,UAAgD;AAC1F,MAAI,OAAO,UAAU,UAAU;AAC7B,UAAM,IAAI,UAAU,WAAW,QAAQ,0CAA0C;AAAA,EACnF;AACF;AAEA,SAAS,UAAU,OAAuB;AACxC,SAAO,MAAM,QAAQ,MAAM,IAAI;AACjC;;;ACtCO,IAAM,qBAAN,MAEP;AAAA,EACE,YACmB,OACR,SACT;AAFiB;AACR;AAET,qBAAiB,QAAQ,UAAU;AAAA,EACrC;AAAA,EAJmB;AAAA,EACR;AAAA,EAKX,MAAM,SAAwB;AAC5B,UAAM,aAAa,MAAM,KAAK,MAAM,aAAa;AACjD,QAAI,EAAE,MAAM,WAAW,WAAW,GAAG,SAAS,KAAK,QAAQ,SAAS,GAAG;AACrE,UAAI,WAAW,qBAAqB,QAAW;AAC7C,cAAM,IAAI,UAAU,sDAAsD;AAAA,MAC5E;AACA,YAAM,EAAE,OAAO,eAAe,SAAS,QAAQ,KAAK,IAAI,MAAM,OAAO,cAAc;AACnF,YAAM,WAAW;AAAA,QACf,KAAK,QAAQ;AAAA,QACb,IAAI,OAAO;AAAA,UACT,IAAI,MAAM,aAAa,IAAI,KAAK,GAAG,KAAK;AAAA,UACxC,IAAI,MAAM,kBAAkB,IAAI,KAAK,GAAG,KAAK;AAAA,UAC7C,IAAI,MAAM,gBAAgB,IAAI,KAAK,GAAG,KAAK;AAAA,UAC3C,IAAI,MAAM,gBAAgB,IAAI,KAAK,GAAG,KAAK;AAAA,UAC3C,IAAI;AAAA,YACF;AAAA,YACA,IAAI,cAAc,KAAK,QAAQ,YAAY,IAAI,MAAM,QAAQ,IAAI,QAAQ,GAAG,IAAI,CAAC;AAAA,YACjF;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AACA,UAAM,KAAK,SAAS;AAAA,EACtB;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,QAAQ,OAAO,MAAM,KAAK,MAAM,aAAa,GAAG,UAAU,KAAK,QAAQ,SAAS;AACtF,QAAI,MAAM,WAAW,OAAW;AAChC,UAAM,SAAS,MAAM,MAAM,OAAO;AAClC,UAAM,cAAc,OAAO,QAAQ,KAAK,CAAC,UAAU,MAAM,SAAS,YAAY;AAC9E,QAAI,gBAAgB,QAAW;AAC7B,YAAM,IAAI,MAAM,iBAAiB,KAAK,QAAQ,SAAS,eAAe,YAAY,EAAE;AAAA,IACtF;AACA,QAAI,CAAC,OAAO,QAAQ,KAAK,CAAC,UAAU,MAAM,SAAS,cAAc,GAAG;AAClE,YAAM,IAAI,MAAM,iBAAiB,KAAK,QAAQ,SAAS,eAAe,cAAc,EAAE;AAAA,IACxF;AACA,UAAM,WAAY,YAAY,MAA6C;AAC3E,QAAI,OAAO,aAAa,YAAY,aAAa,KAAK,QAAQ,YAAY;AACxE,YAAM,IAAI;AAAA,QACR,iBAAiB,KAAK,QAAQ,SAAS,QAAQ,QAAQ,yBAAyB,KAAK,QAAQ,UAAU;AAAA,MACzG;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,SAA+D;AAC1E,sBAAkB,QAAQ,WAAW,KAAK,QAAQ,UAAU;AAC5D,QAAI,QAAQ,UAAU,WAAW,EAAG;AACpC,UAAM,QAAQ,OAAO,MAAM,KAAK,MAAM,aAAa,GAAG,UAAU,KAAK,QAAQ,SAAS;AACtF,UAAM,MAAM,QAAQ,UAAU,IAAI,CAAC,aAAa,UAAU,SAAS,EAAE,CAAC,EAAE,KAAK,IAAI;AACjF,UAAM,MAAM,OAAO,GAAG,gBAAgB,QAAQ,GAAG,GAAG;AACpD,UAAM,OAAO,QAAQ,UAAU,QAAQ,CAAC,aAAa,UAAU,QAAQ,CAAC;AACxE,QAAI,KAAK,SAAS,EAAG,OAAM,MAAM,IAAI,MAAM,QAAQ,eAAe;AAAA,EACpE;AAAA,EAEA,MAAM,OAAO,SAA+E;AAC1F,0BAAsB,SAAS,KAAK,QAAQ,UAAU;AACtD,UAAM,QAAQ,OAAO,MAAM,KAAK,MAAM,aAAa,GAAG,UAAU,KAAK,QAAQ,SAAS;AACtF,QAAI,iBAAiB,QAAQ;AAC7B,eAAS;AACP,UAAI,QAAQ,MACT,OAAO,QAAQ,MAAM,EACrB,aAAa,WAAW,KAAK,QAAQ,MAAM,CAAC,EAC5C,MAAM,cAAc;AACvB,YAAM,SAAS,kBAAkB,QAAQ,MAAM;AAC/C,UAAI,WAAW,OAAW,SAAQ,MAAM,MAAM,MAAM;AACpD,YAAM,WAAW,MAAM,MAAM,QAAQ;AACrC,qBAAe,QAAQ,WAAW;AAClC,YAAM,aAAa,kBAA+B,UAAU,QAAW,KAAK,QAAQ,MAAM;AAC1F,YAAM,UAAU,cAAc,YAAY,OAAO;AACjD,UACE,QAAQ,UAAU,QAAQ,QAC1B,SAAS,SAAS,kBAClB,sBAAsB,YAAY,QAAQ,QAAQ;AAElD,eAAO;AACT,YAAM,YAAY,mBAAmB,cAAc;AACnD,UAAI,cAAc,eAAgB,QAAO;AACzC,uBAAiB;AAAA,IACnB;AAAA,EACF;AACF;AAEA,SAAS,cACP,SACA,SACwC;AACxC,SAAO,QACJ,OAAO,CAAC,WAAW,QAAQ,aAAa,UAAa,OAAO,SAAS,QAAQ,QAAQ,EACrF,KAAK,CAAC,MAAM,UAAU,MAAM,QAAQ,KAAK,KAAK,EAC9C,MAAM,GAAG,QAAQ,IAAI;AAC1B;AAEA,SAAS,sBACP,SACA,UACS;AACT,SAAO,aAAa,UAAa,QAAQ,KAAK,CAAC,WAAW,OAAO,QAAQ,QAAQ;AACnF;AAEA,SAAS,mBAAmB,SAAyB;AACnD,SAAO,UAAU,KAAK,MAAM,OAAO,mBAAmB,CAAC,IAAI,UAAU,UAAU;AACjF;AAEA,SAAS,UAAU,OAAuB;AACxC,SAAO,IAAI,MAAM,WAAW,KAAK,IAAI,CAAC;AACxC;AAEA,SAAS,eAAe,QAAuC;AAC7D,MAAI,QAAQ,SAAS;AACnB,UAAM,QAAQ,IAAI,MAAM,4BAA4B;AACpD,UAAM,OAAO;AACb,UAAM;AAAA,EACR;AACF;AAEA,SAAS,iBAAiB,YAA0B;AAClD,MAAI,CAAC,OAAO,cAAc,UAAU,KAAK,aAAa;AACpD,UAAM,IAAI,WAAW,oDAAoD;AAC7E;AAEA,SAAS,kBACP,WACA,YACM;AACN,QAAM,MAAM,oBAAI,IAAY;AAC5B,aAAW,YAAY,WAAW;AAChC,QAAI,IAAI,IAAI,SAAS,EAAE,EAAG,OAAM,IAAI,UAAU,iCAAiC,SAAS,EAAE,EAAE;AAC5F,QAAI,IAAI,SAAS,EAAE;AACnB,QAAI,SAAS,WAAW,WAAW;AACjC,YAAM,IAAI,UAAU,mBAAmB,SAAS,EAAE,uCAAuC;AAC3F,eAAW,aAAa,SAAS,YAAY;AAC3C,UAAI,UAAU,OAAO,WAAW;AAC9B,cAAM,IAAI,MAAM,0CAA0C,SAAS,EAAE,EAAE;AACzE,UAAI,CAAC,UAAU,OAAO,MAAM,OAAO,QAAQ;AACzC,cAAM,IAAI,MAAM,uBAAuB,SAAS,EAAE,mCAAmC;AAAA,IACzF;AAAA,EACF;AACF;AAEA,SAAS,WAAW,QAAsE;AACxF,MAAI,WAAW,YAAa,QAAO;AACnC,MAAI,WAAW,aAAc,QAAO;AACpC,SAAO;AACT;AAEA,SAAS,sBAAsB,SAA8B,YAA0B;AACrF,iBAAe,QAAQ,WAAW;AAClC,MAAI,CAAC,OAAO,cAAc,QAAQ,IAAI,KAAK,QAAQ,OAAO;AACxD,UAAM,IAAI,WAAW,qDAAqD;AAC5E,MAAI,QAAQ,aAAa,UAAa,CAAC,OAAO,SAAS,QAAQ,QAAQ;AACrE,UAAM,IAAI,WAAW,iDAAiD;AACxE,MAAI,QAAQ,OAAO,WAAW,YAAY;AACxC,UAAM,IAAI;AAAA,MACR,uCAAuC,UAAU,4BAA4B,QAAQ,OAAO,MAAM;AAAA,IACpG;AAAA,EACF;AACA,MAAI,CAAC,QAAQ,OAAO,MAAM,OAAO,QAAQ;AACvC,UAAM,IAAI,MAAM,sDAAsD;AAC1E;;;AClLO,IAAM,sBAAN,MAA0B;AAAA,EAK/B,YAA6B,UAAsC,CAAC,GAAG;AAA1C;AAC3B,SAAK,WAAW,QAAQ;AAAA,EAC1B;AAAA,EAF6B;AAAA,EAJZ;AAAA,EACT;AAAA,EACA,SAAS;AAAA,EAMjB,YACE,SACiC;AACjC,SAAK,WAAW;AAChB,WAAO,IAAI,mBAAgC,MAAM,OAAO;AAAA,EAC1D;AAAA,EAEA,eAA+C;AAC7C,SAAK,WAAW;AAChB,SAAK,kBACH,KAAK,aAAa,SACd,yBAAyB,KAAK,QAAQ,GAAG,IACzC,QAAQ,QAAQ,KAAK,QAAQ;AACnC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAM,QAAuB;AAC3B,QAAI,KAAK,OAAQ;AACjB,SAAK,SAAS;AACd,QAAI,KAAK,aAAa,UAAa,KAAK,kBAAkB,OAAW;AACrE,WAAO,MAAM,KAAK,eAAe,QAAQ;AAAA,EAC3C;AAAA,EAEQ,aAAmB;AACzB,QAAI,KAAK,OAAQ,OAAM,IAAI,MAAM,gCAAgC;AAAA,EACnE;AACF;","names":[]}
1
+ {"version":3,"sources":["../src/helpers.ts","../src/types.ts","../src/store.ts","../src/client.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\nimport type { EmbeddedDocument, VectorMetadata } from \"@anvia/core/embeddings\";\nimport type { VectorSearchResult } from \"@anvia/core/vector-store\";\nimport type { LanceDBVectorStoreOptions } from \"./types.js\";\nimport {\n documentColumn,\n documentIdColumn,\n metadataColumn,\n reservedColumnPrefix,\n rowIdColumn,\n vectorColumn,\n} from \"./types.js\";\n\nexport function assertNoReservedMetadata(metadata: VectorMetadata | undefined): void {\n for (const key of Object.keys(metadata ?? {})) {\n if (key.startsWith(reservedColumnPrefix)) {\n throw new Error(`Metadata key ${key} is reserved for Anvia LanceDB columns`);\n }\n }\n}\n\nexport function rowId(id: string): string {\n return createHash(\"sha256\").update(id).digest(\"hex\").slice(0, 32);\n}\n\nexport function serializeDocument(document: unknown): string {\n return typeof document === \"string\" ? document : JSON.stringify(document);\n}\n\nexport function parseDocument<T>(document: unknown): T {\n if (document === null || document === undefined) {\n return \"\" as T;\n }\n if (typeof document !== \"string\") {\n return document as T;\n }\n try {\n return JSON.parse(document) as T;\n } catch {\n return document as T;\n }\n}\n\nexport function lanceRows<T, Metadata extends VectorMetadata>(\n document: EmbeddedDocument<T, Metadata>,\n): Array<Record<string, unknown>> {\n if (document.embeddings.length === 0) {\n throw new Error(`Document ${document.id} has no embeddings`);\n }\n assertNoReservedMetadata(document.metadata);\n\n return document.embeddings.map((embedding, index) => {\n const logicalId =\n document.embeddings.length === 1 ? document.id : `${document.id}#embedding:${index}`;\n const row: Record<string, unknown> = {\n [rowIdColumn]: rowId(logicalId),\n [documentIdColumn]: document.id,\n [documentColumn]: serializeDocument(document.document),\n [metadataColumn]: JSON.stringify(document.metadata ?? {}),\n [vectorColumn]: embedding.vector,\n };\n return row;\n });\n}\n\nexport function parseQueryResults<T, Metadata extends VectorMetadata>(\n response: unknown[],\n minScore: number | undefined,\n metric: LanceDBVectorStoreOptions[\"metric\"],\n): Array<VectorSearchResult<T, Metadata>> {\n const byId = new Map<string, VectorSearchResult<T, Metadata>>();\n\n for (const row of response) {\n const record = row as Record<string, unknown>;\n const distance = typeof record._distance === \"number\" ? record._distance : 0;\n const score = metric === undefined || metric === \"cosine\" ? 1 - distance : -distance;\n\n if (minScore !== undefined && score < minScore) {\n continue;\n }\n\n const id = String(record[documentIdColumn] ?? \"\");\n const result: VectorSearchResult<T, Metadata> = {\n id,\n score,\n document: parseDocument(record[documentColumn]),\n };\n const metadata = metadataFromColumns<Metadata>(record);\n if (metadata !== undefined) {\n result.metadata = metadata;\n }\n const current = byId.get(id);\n if (current === undefined || result.score > current.score) {\n byId.set(id, result);\n }\n }\n\n return [...byId.values()];\n}\n\nexport async function defaultLanceDBConnection(\n uri?: string,\n): Promise<import(\"./types.js\").LanceDBConnectionLike> {\n const lancedb = await import(\"@lancedb/lancedb\");\n return lancedb.connect(\n uri ?? \"~/.anvia/lancedb\",\n ) as unknown as import(\"./types.js\").LanceDBConnectionLike;\n}\n\nfunction metadataFromColumns<Metadata extends VectorMetadata>(\n record: Record<string, unknown>,\n): Metadata | undefined {\n const raw = record[metadataColumn];\n if (raw === null || raw === undefined) return undefined;\n const metadata = (typeof raw === \"string\" ? JSON.parse(raw) : raw) as Metadata;\n return Object.keys(metadata).length === 0 ? undefined : metadata;\n}\n","import type { VectorMetric } from \"@anvia/core/vector-store\";\n\nexport const documentIdColumn = \"__anvia_document_id\";\nexport const documentColumn = \"__anvia_document\";\nexport const metadataColumn = \"__anvia_metadata\";\nexport const vectorColumn = \"__anvia_vector\";\nexport const rowIdColumn = \"__anvia_id\";\nexport const reservedColumnPrefix = \"__anvia_\";\n\nexport type LanceDBTableLike = {\n add(rows: Record<string, unknown>[], options?: Record<string, unknown>): Promise<unknown>;\n search(vector: number[]): LanceDBQueryLike;\n countRows(): Promise<number>;\n delete(filter: string): Promise<unknown>;\n schema?(): Promise<{ fields?: Array<{ name?: string; type?: unknown }> }>;\n};\n\nexport type LanceDBQueryLike = {\n limit(n: number): LanceDBQueryLike;\n where(predicate: string): LanceDBQueryLike;\n distanceType(metric: \"cosine\" | \"l2\" | \"dot\"): LanceDBQueryLike;\n toArray(): Promise<unknown[]>;\n};\n\nexport type LanceDBConnectionLike = {\n openTable(name: string): Promise<LanceDBTableLike>;\n tableNames(): Promise<string[]>;\n createTable(name: string, data: Record<string, unknown>[]): Promise<LanceDBTableLike>;\n createEmptyTable?(\n name: string,\n schema: unknown,\n options?: Record<string, unknown>,\n ): Promise<LanceDBTableLike>;\n close?(): Promise<unknown> | unknown;\n};\n\nexport type LanceDBVectorClientOptions = {\n client?: LanceDBConnectionLike | undefined;\n uri?: string | undefined;\n};\n\nexport type LanceDBVectorStoreOptions = {\n tableName: string;\n dimensions: number;\n metric?: VectorMetric | undefined;\n};\n","import type { EmbeddedDocument, VectorMetadata } from \"@anvia/core/embeddings\";\nimport type {\n VectorSearchRequest,\n VectorSearchResult,\n VectorStore,\n VectorStoreUpsertOptions,\n} from \"@anvia/core/vector-store\";\nimport { matchesVectorFilter } from \"@anvia/core/vector-store\";\nimport type { LanceDBVectorClient } from \"./client.js\";\nimport { lanceRows, parseQueryResults } from \"./helpers.js\";\nimport {\n documentColumn,\n documentIdColumn,\n type LanceDBVectorStoreOptions,\n metadataColumn,\n rowIdColumn,\n vectorColumn,\n} from \"./types.js\";\n\nexport class LanceDBVectorStore<T, Metadata extends VectorMetadata = VectorMetadata>\n implements VectorStore<T, Metadata>\n{\n constructor(\n private readonly owner: LanceDBVectorClient,\n readonly options: LanceDBVectorStoreOptions,\n ) {\n assertDimensions(options.dimensions);\n }\n\n async ensure(): Promise<void> {\n const connection = await this.owner.nativeClient();\n if (!(await connection.tableNames()).includes(this.options.tableName)) {\n if (connection.createEmptyTable === undefined) {\n throw new TypeError(\"LanceDB provisioning requires createEmptyTable(...).\");\n }\n const { Field, FixedSizeList, Float32, Schema, Utf8 } = await import(\"apache-arrow\");\n await connection.createEmptyTable(\n this.options.tableName,\n new Schema([\n new Field(rowIdColumn, new Utf8(), false),\n new Field(documentIdColumn, new Utf8(), false),\n new Field(documentColumn, new Utf8(), false),\n new Field(metadataColumn, new Utf8(), false),\n new Field(\n vectorColumn,\n new FixedSizeList(this.options.dimensions, new Field(\"item\", new Float32(), true)),\n false,\n ),\n ]),\n );\n }\n await this.validate();\n }\n\n async validate(): Promise<void> {\n const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);\n if (table.schema === undefined) return;\n const schema = await table.schema();\n const vectorField = schema.fields?.find((field) => field.name === vectorColumn);\n if (vectorField === undefined) {\n throw new Error(`LanceDB table ${this.options.tableName} is missing ${vectorColumn}`);\n }\n if (!schema.fields?.some((field) => field.name === metadataColumn)) {\n throw new Error(`LanceDB table ${this.options.tableName} is missing ${metadataColumn}`);\n }\n const listSize = (vectorField.type as { listSize?: unknown } | undefined)?.listSize;\n if (typeof listSize === \"number\" && listSize !== this.options.dimensions) {\n throw new Error(\n `LanceDB table ${this.options.tableName} has ${listSize} dimensions; expected ${this.options.dimensions}`,\n );\n }\n }\n\n async upsert(options: VectorStoreUpsertOptions<T, Metadata>): Promise<void> {\n validateDocuments(options.documents, this.options.dimensions);\n if (options.documents.length === 0) return;\n const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);\n const ids = options.documents.map((document) => sqlString(document.id)).join(\", \");\n await table.delete(`${documentIdColumn} IN (${ids})`);\n const rows = options.documents.flatMap((document) => lanceRows(document));\n if (rows.length > 0) await table.add(rows, options.providerOptions);\n }\n\n async search(request: VectorSearchRequest): Promise<Array<VectorSearchResult<T, Metadata>>> {\n validateSearchRequest(request, this.options.dimensions);\n const table = await (await this.owner.nativeClient()).openTable(this.options.tableName);\n let candidateLimit = request.topK;\n for (;;) {\n const query = table\n .search(request.vector)\n .distanceType(metricName(this.options.metric))\n .limit(candidateLimit);\n const response = await query.toArray();\n throwIfAborted(request.abortSignal);\n const unfiltered = parseQueryResults<T, Metadata>(response, undefined, this.options.metric);\n const results = selectResults(unfiltered, request);\n if (\n results.length >= request.topK ||\n response.length < candidateLimit ||\n hasResultBelowMinimum(unfiltered, request.minScore)\n )\n return results;\n const nextLimit = nextCandidateLimit(candidateLimit);\n if (nextLimit === candidateLimit) return results;\n candidateLimit = nextLimit;\n }\n }\n}\n\nfunction selectResults<T, Metadata extends VectorMetadata>(\n results: Array<VectorSearchResult<T, Metadata>>,\n request: VectorSearchRequest,\n): Array<VectorSearchResult<T, Metadata>> {\n return results\n .filter((result) => matchesVectorFilter(result.metadata, request.filter))\n .filter((result) => request.minScore === undefined || result.score >= request.minScore)\n .sort((left, right) => right.score - left.score)\n .slice(0, request.topK);\n}\n\nfunction hasResultBelowMinimum<T, Metadata extends VectorMetadata>(\n results: Array<VectorSearchResult<T, Metadata>>,\n minScore: number | undefined,\n): boolean {\n return minScore !== undefined && results.some((result) => result.score < minScore);\n}\n\nfunction nextCandidateLimit(current: number): number {\n return current > Math.floor(Number.MAX_SAFE_INTEGER / 2) ? current : current * 2;\n}\n\nfunction sqlString(value: string): string {\n return `'${value.replaceAll(\"'\", \"''\")}'`;\n}\n\nfunction throwIfAborted(signal: AbortSignal | undefined): void {\n if (signal?.aborted) {\n const error = new Error(\"The operation was aborted.\");\n error.name = \"AbortError\";\n throw error;\n }\n}\n\nfunction assertDimensions(dimensions: number): void {\n if (!Number.isSafeInteger(dimensions) || dimensions < 1)\n throw new RangeError(\"Vector dimensions must be a positive safe integer.\");\n}\n\nfunction validateDocuments<T, Metadata extends VectorMetadata>(\n documents: Array<EmbeddedDocument<T, Metadata>>,\n dimensions: number,\n): void {\n const ids = new Set<string>();\n for (const document of documents) {\n if (ids.has(document.id)) throw new TypeError(`Duplicate vector document id: ${document.id}`);\n ids.add(document.id);\n if (document.embeddings.length === 0)\n throw new TypeError(`Vector document ${document.id} must contain at least one embedding.`);\n for (const embedding of document.embeddings) {\n if (embedding.vector.length !== dimensions)\n throw new Error(`Vector dimension mismatch for document ${document.id}`);\n if (!embedding.vector.every(Number.isFinite))\n throw new Error(`Vector for document ${document.id} must contain only finite numbers`);\n }\n }\n}\n\nfunction metricName(metric: LanceDBVectorStoreOptions[\"metric\"]): \"cosine\" | \"l2\" | \"dot\" {\n if (metric === \"euclidean\") return \"l2\";\n if (metric === \"dotProduct\") return \"dot\";\n return \"cosine\";\n}\n\nfunction validateSearchRequest(request: VectorSearchRequest, dimensions: number): void {\n throwIfAborted(request.abortSignal);\n if (!Number.isSafeInteger(request.topK) || request.topK < 1)\n throw new RangeError(\"Vector search topK must be a positive safe integer.\");\n if (request.minScore !== undefined && !Number.isFinite(request.minScore))\n throw new RangeError(\"Vector search minScore must be a finite number.\");\n if (request.vector.length !== dimensions) {\n throw new Error(\n `Vector dimension mismatch: expected ${dimensions} dimensions but received ${request.vector.length} for query`,\n );\n }\n if (!request.vector.every(Number.isFinite))\n throw new Error(\"Vector search query must contain only finite numbers\");\n}\n","import type { VectorMetadata } from \"@anvia/core/embeddings\";\nimport { defaultLanceDBConnection } from \"./helpers.js\";\nimport { LanceDBVectorStore } from \"./store.js\";\nimport type {\n LanceDBConnectionLike,\n LanceDBVectorClientOptions,\n LanceDBVectorStoreOptions,\n} from \"./types.js\";\n\nexport class LanceDBVectorClient {\n private readonly injected: LanceDBConnectionLike | undefined;\n private clientPromise: Promise<LanceDBConnectionLike> | undefined;\n private closed = false;\n\n constructor(private readonly options: LanceDBVectorClientOptions = {}) {\n this.injected = options.client;\n }\n\n vectorStore<T, Metadata extends VectorMetadata = VectorMetadata>(\n options: LanceDBVectorStoreOptions,\n ): LanceDBVectorStore<T, Metadata> {\n this.assertOpen();\n return new LanceDBVectorStore<T, Metadata>(this, options);\n }\n\n nativeClient(): Promise<LanceDBConnectionLike> {\n this.assertOpen();\n this.clientPromise ??=\n this.injected === undefined\n ? defaultLanceDBConnection(this.options.uri)\n : Promise.resolve(this.injected);\n return this.clientPromise;\n }\n\n async close(): Promise<void> {\n if (this.closed) return;\n this.closed = true;\n if (this.injected !== undefined || this.clientPromise === undefined) return;\n await (await this.clientPromise).close?.();\n }\n\n private assertOpen(): void {\n if (this.closed) throw new Error(\"LanceDBVectorClient is closed.\");\n }\n}\n"],"mappings":";AAAA,SAAS,kBAAkB;;;ACEpB,IAAM,mBAAmB;AACzB,IAAM,iBAAiB;AACvB,IAAM,iBAAiB;AACvB,IAAM,eAAe;AACrB,IAAM,cAAc;AACpB,IAAM,uBAAuB;;;ADM7B,SAAS,yBAAyB,UAA4C;AACnF,aAAW,OAAO,OAAO,KAAK,YAAY,CAAC,CAAC,GAAG;AAC7C,QAAI,IAAI,WAAW,oBAAoB,GAAG;AACxC,YAAM,IAAI,MAAM,gBAAgB,GAAG,wCAAwC;AAAA,IAC7E;AAAA,EACF;AACF;AAEO,SAAS,MAAM,IAAoB;AACxC,SAAO,WAAW,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,KAAK,EAAE,MAAM,GAAG,EAAE;AAClE;AAEO,SAAS,kBAAkB,UAA2B;AAC3D,SAAO,OAAO,aAAa,WAAW,WAAW,KAAK,UAAU,QAAQ;AAC1E;AAEO,SAAS,cAAiB,UAAsB;AACrD,MAAI,aAAa,QAAQ,aAAa,QAAW;AAC/C,WAAO;AAAA,EACT;AACA,MAAI,OAAO,aAAa,UAAU;AAChC,WAAO;AAAA,EACT;AACA,MAAI;AACF,WAAO,KAAK,MAAM,QAAQ;AAAA,EAC5B,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEO,SAAS,UACd,UACgC;AAChC,MAAI,SAAS,WAAW,WAAW,GAAG;AACpC,UAAM,IAAI,MAAM,YAAY,SAAS,EAAE,oBAAoB;AAAA,EAC7D;AACA,2BAAyB,SAAS,QAAQ;AAE1C,SAAO,SAAS,WAAW,IAAI,CAAC,WAAW,UAAU;AACnD,UAAM,YACJ,SAAS,WAAW,WAAW,IAAI,SAAS,KAAK,GAAG,SAAS,EAAE,cAAc,KAAK;AACpF,UAAM,MAA+B;AAAA,MACnC,CAAC,WAAW,GAAG,MAAM,SAAS;AAAA,MAC9B,CAAC,gBAAgB,GAAG,SAAS;AAAA,MAC7B,CAAC,cAAc,GAAG,kBAAkB,SAAS,QAAQ;AAAA,MACrD,CAAC,cAAc,GAAG,KAAK,UAAU,SAAS,YAAY,CAAC,CAAC;AAAA,MACxD,CAAC,YAAY,GAAG,UAAU;AAAA,IAC5B;AACA,WAAO;AAAA,EACT,CAAC;AACH;AAEO,SAAS,kBACd,UACA,UACA,QACwC;AACxC,QAAM,OAAO,oBAAI,IAA6C;AAE9D,aAAW,OAAO,UAAU;AAC1B,UAAM,SAAS;AACf,UAAM,WAAW,OAAO,OAAO,cAAc,WAAW,OAAO,YAAY;AAC3E,UAAM,QAAQ,WAAW,UAAa,WAAW,WAAW,IAAI,WAAW,CAAC;AAE5E,QAAI,aAAa,UAAa,QAAQ,UAAU;AAC9C;AAAA,IACF;AAEA,UAAM,KAAK,OAAO,OAAO,gBAAgB,KAAK,EAAE;AAChD,UAAM,SAA0C;AAAA,MAC9C;AAAA,MACA;AAAA,MACA,UAAU,cAAc,OAAO,cAAc,CAAC;AAAA,IAChD;AACA,UAAM,WAAW,oBAA8B,MAAM;AACrD,QAAI,aAAa,QAAW;AAC1B,aAAO,WAAW;AAAA,IACpB;AACA,UAAM,UAAU,KAAK,IAAI,EAAE;AAC3B,QAAI,YAAY,UAAa,OAAO,QAAQ,QAAQ,OAAO;AACzD,WAAK,IAAI,IAAI,MAAM;AAAA,IACrB;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,KAAK,OAAO,CAAC;AAC1B;AAEA,eAAsB,yBACpB,KACqD;AACrD,QAAM,UAAU,MAAM,OAAO,kBAAkB;AAC/C,SAAO,QAAQ;AAAA,IACb,OAAO;AAAA,EACT;AACF;AAEA,SAAS,oBACP,QACsB;AACtB,QAAM,MAAM,OAAO,cAAc;AACjC,MAAI,QAAQ,QAAQ,QAAQ,OAAW,QAAO;AAC9C,QAAM,WAAY,OAAO,QAAQ,WAAW,KAAK,MAAM,GAAG,IAAI;AAC9D,SAAO,OAAO,KAAK,QAAQ,EAAE,WAAW,IAAI,SAAY;AAC1D;;;AE7GA,SAAS,2BAA2B;AAY7B,IAAM,qBAAN,MAEP;AAAA,EACE,YACmB,OACR,SACT;AAFiB;AACR;AAET,qBAAiB,QAAQ,UAAU;AAAA,EACrC;AAAA,EAJmB;AAAA,EACR;AAAA,EAKX,MAAM,SAAwB;AAC5B,UAAM,aAAa,MAAM,KAAK,MAAM,aAAa;AACjD,QAAI,EAAE,MAAM,WAAW,WAAW,GAAG,SAAS,KAAK,QAAQ,SAAS,GAAG;AACrE,UAAI,WAAW,qBAAqB,QAAW;AAC7C,cAAM,IAAI,UAAU,sDAAsD;AAAA,MAC5E;AACA,YAAM,EAAE,OAAO,eAAe,SAAS,QAAQ,KAAK,IAAI,MAAM,OAAO,cAAc;AACnF,YAAM,WAAW;AAAA,QACf,KAAK,QAAQ;AAAA,QACb,IAAI,OAAO;AAAA,UACT,IAAI,MAAM,aAAa,IAAI,KAAK,GAAG,KAAK;AAAA,UACxC,IAAI,MAAM,kBAAkB,IAAI,KAAK,GAAG,KAAK;AAAA,UAC7C,IAAI,MAAM,gBAAgB,IAAI,KAAK,GAAG,KAAK;AAAA,UAC3C,IAAI,MAAM,gBAAgB,IAAI,KAAK,GAAG,KAAK;AAAA,UAC3C,IAAI;AAAA,YACF;AAAA,YACA,IAAI,cAAc,KAAK,QAAQ,YAAY,IAAI,MAAM,QAAQ,IAAI,QAAQ,GAAG,IAAI,CAAC;AAAA,YACjF;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AACA,UAAM,KAAK,SAAS;AAAA,EACtB;AAAA,EAEA,MAAM,WAA0B;AAC9B,UAAM,QAAQ,OAAO,MAAM,KAAK,MAAM,aAAa,GAAG,UAAU,KAAK,QAAQ,SAAS;AACtF,QAAI,MAAM,WAAW,OAAW;AAChC,UAAM,SAAS,MAAM,MAAM,OAAO;AAClC,UAAM,cAAc,OAAO,QAAQ,KAAK,CAAC,UAAU,MAAM,SAAS,YAAY;AAC9E,QAAI,gBAAgB,QAAW;AAC7B,YAAM,IAAI,MAAM,iBAAiB,KAAK,QAAQ,SAAS,eAAe,YAAY,EAAE;AAAA,IACtF;AACA,QAAI,CAAC,OAAO,QAAQ,KAAK,CAAC,UAAU,MAAM,SAAS,cAAc,GAAG;AAClE,YAAM,IAAI,MAAM,iBAAiB,KAAK,QAAQ,SAAS,eAAe,cAAc,EAAE;AAAA,IACxF;AACA,UAAM,WAAY,YAAY,MAA6C;AAC3E,QAAI,OAAO,aAAa,YAAY,aAAa,KAAK,QAAQ,YAAY;AACxE,YAAM,IAAI;AAAA,QACR,iBAAiB,KAAK,QAAQ,SAAS,QAAQ,QAAQ,yBAAyB,KAAK,QAAQ,UAAU;AAAA,MACzG;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,SAA+D;AAC1E,sBAAkB,QAAQ,WAAW,KAAK,QAAQ,UAAU;AAC5D,QAAI,QAAQ,UAAU,WAAW,EAAG;AACpC,UAAM,QAAQ,OAAO,MAAM,KAAK,MAAM,aAAa,GAAG,UAAU,KAAK,QAAQ,SAAS;AACtF,UAAM,MAAM,QAAQ,UAAU,IAAI,CAAC,aAAa,UAAU,SAAS,EAAE,CAAC,EAAE,KAAK,IAAI;AACjF,UAAM,MAAM,OAAO,GAAG,gBAAgB,QAAQ,GAAG,GAAG;AACpD,UAAM,OAAO,QAAQ,UAAU,QAAQ,CAAC,aAAa,UAAU,QAAQ,CAAC;AACxE,QAAI,KAAK,SAAS,EAAG,OAAM,MAAM,IAAI,MAAM,QAAQ,eAAe;AAAA,EACpE;AAAA,EAEA,MAAM,OAAO,SAA+E;AAC1F,0BAAsB,SAAS,KAAK,QAAQ,UAAU;AACtD,UAAM,QAAQ,OAAO,MAAM,KAAK,MAAM,aAAa,GAAG,UAAU,KAAK,QAAQ,SAAS;AACtF,QAAI,iBAAiB,QAAQ;AAC7B,eAAS;AACP,YAAM,QAAQ,MACX,OAAO,QAAQ,MAAM,EACrB,aAAa,WAAW,KAAK,QAAQ,MAAM,CAAC,EAC5C,MAAM,cAAc;AACvB,YAAM,WAAW,MAAM,MAAM,QAAQ;AACrC,qBAAe,QAAQ,WAAW;AAClC,YAAM,aAAa,kBAA+B,UAAU,QAAW,KAAK,QAAQ,MAAM;AAC1F,YAAM,UAAU,cAAc,YAAY,OAAO;AACjD,UACE,QAAQ,UAAU,QAAQ,QAC1B,SAAS,SAAS,kBAClB,sBAAsB,YAAY,QAAQ,QAAQ;AAElD,eAAO;AACT,YAAM,YAAY,mBAAmB,cAAc;AACnD,UAAI,cAAc,eAAgB,QAAO;AACzC,uBAAiB;AAAA,IACnB;AAAA,EACF;AACF;AAEA,SAAS,cACP,SACA,SACwC;AACxC,SAAO,QACJ,OAAO,CAAC,WAAW,oBAAoB,OAAO,UAAU,QAAQ,MAAM,CAAC,EACvE,OAAO,CAAC,WAAW,QAAQ,aAAa,UAAa,OAAO,SAAS,QAAQ,QAAQ,EACrF,KAAK,CAAC,MAAM,UAAU,MAAM,QAAQ,KAAK,KAAK,EAC9C,MAAM,GAAG,QAAQ,IAAI;AAC1B;AAEA,SAAS,sBACP,SACA,UACS;AACT,SAAO,aAAa,UAAa,QAAQ,KAAK,CAAC,WAAW,OAAO,QAAQ,QAAQ;AACnF;AAEA,SAAS,mBAAmB,SAAyB;AACnD,SAAO,UAAU,KAAK,MAAM,OAAO,mBAAmB,CAAC,IAAI,UAAU,UAAU;AACjF;AAEA,SAAS,UAAU,OAAuB;AACxC,SAAO,IAAI,MAAM,WAAW,KAAK,IAAI,CAAC;AACxC;AAEA,SAAS,eAAe,QAAuC;AAC7D,MAAI,QAAQ,SAAS;AACnB,UAAM,QAAQ,IAAI,MAAM,4BAA4B;AACpD,UAAM,OAAO;AACb,UAAM;AAAA,EACR;AACF;AAEA,SAAS,iBAAiB,YAA0B;AAClD,MAAI,CAAC,OAAO,cAAc,UAAU,KAAK,aAAa;AACpD,UAAM,IAAI,WAAW,oDAAoD;AAC7E;AAEA,SAAS,kBACP,WACA,YACM;AACN,QAAM,MAAM,oBAAI,IAAY;AAC5B,aAAW,YAAY,WAAW;AAChC,QAAI,IAAI,IAAI,SAAS,EAAE,EAAG,OAAM,IAAI,UAAU,iCAAiC,SAAS,EAAE,EAAE;AAC5F,QAAI,IAAI,SAAS,EAAE;AACnB,QAAI,SAAS,WAAW,WAAW;AACjC,YAAM,IAAI,UAAU,mBAAmB,SAAS,EAAE,uCAAuC;AAC3F,eAAW,aAAa,SAAS,YAAY;AAC3C,UAAI,UAAU,OAAO,WAAW;AAC9B,cAAM,IAAI,MAAM,0CAA0C,SAAS,EAAE,EAAE;AACzE,UAAI,CAAC,UAAU,OAAO,MAAM,OAAO,QAAQ;AACzC,cAAM,IAAI,MAAM,uBAAuB,SAAS,EAAE,mCAAmC;AAAA,IACzF;AAAA,EACF;AACF;AAEA,SAAS,WAAW,QAAsE;AACxF,MAAI,WAAW,YAAa,QAAO;AACnC,MAAI,WAAW,aAAc,QAAO;AACpC,SAAO;AACT;AAEA,SAAS,sBAAsB,SAA8B,YAA0B;AACrF,iBAAe,QAAQ,WAAW;AAClC,MAAI,CAAC,OAAO,cAAc,QAAQ,IAAI,KAAK,QAAQ,OAAO;AACxD,UAAM,IAAI,WAAW,qDAAqD;AAC5E,MAAI,QAAQ,aAAa,UAAa,CAAC,OAAO,SAAS,QAAQ,QAAQ;AACrE,UAAM,IAAI,WAAW,iDAAiD;AACxE,MAAI,QAAQ,OAAO,WAAW,YAAY;AACxC,UAAM,IAAI;AAAA,MACR,uCAAuC,UAAU,4BAA4B,QAAQ,OAAO,MAAM;AAAA,IACpG;AAAA,EACF;AACA,MAAI,CAAC,QAAQ,OAAO,MAAM,OAAO,QAAQ;AACvC,UAAM,IAAI,MAAM,sDAAsD;AAC1E;;;ACjLO,IAAM,sBAAN,MAA0B;AAAA,EAK/B,YAA6B,UAAsC,CAAC,GAAG;AAA1C;AAC3B,SAAK,WAAW,QAAQ;AAAA,EAC1B;AAAA,EAF6B;AAAA,EAJZ;AAAA,EACT;AAAA,EACA,SAAS;AAAA,EAMjB,YACE,SACiC;AACjC,SAAK,WAAW;AAChB,WAAO,IAAI,mBAAgC,MAAM,OAAO;AAAA,EAC1D;AAAA,EAEA,eAA+C;AAC7C,SAAK,WAAW;AAChB,SAAK,kBACH,KAAK,aAAa,SACd,yBAAyB,KAAK,QAAQ,GAAG,IACzC,QAAQ,QAAQ,KAAK,QAAQ;AACnC,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAM,QAAuB;AAC3B,QAAI,KAAK,OAAQ;AACjB,SAAK,SAAS;AACd,QAAI,KAAK,aAAa,UAAa,KAAK,kBAAkB,OAAW;AACrE,WAAO,MAAM,KAAK,eAAe,QAAQ;AAAA,EAC3C;AAAA,EAEQ,aAAmB;AACzB,QAAI,KAAK,OAAQ,OAAM,IAAI,MAAM,gCAAgC;AAAA,EACnE;AACF;","names":[]}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@anvia/lancedb",
3
- "version": "1.0.0-rc.2",
3
+ "version": "1.0.0-rc.4",
4
4
  "description": "LanceDB vector store adapter for Anvia.",
5
5
  "author": "anvia",
6
6
  "maintainer": "Indra Zulfi",
@@ -34,10 +34,10 @@
34
34
  "tsup": "^8.5.0",
35
35
  "typescript": "^5.9.3",
36
36
  "vitest": "^4.0.8",
37
- "@anvia/core": "1.0.0-rc.2"
37
+ "@anvia/core": "1.0.0-rc.4"
38
38
  },
39
39
  "peerDependencies": {
40
- "@anvia/core": "1.0.0-rc.2"
40
+ "@anvia/core": "1.0.0-rc.4"
41
41
  },
42
42
  "scripts": {
43
43
  "build": "tsup src/index.ts --format esm --dts --sourcemap --clean",