@nhtio/adk 1.20260604.0 → 1.20260607.0

package/batteries/vector/orama.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"orama.cjs","names":["#orama","#dbs","#collections","#dims","#coll","#getDb","#cosineLikeScore","#scoreVec","#project","#dot"],"sources":["../../../src/batteries/vector/orama/index.ts"],"sourcesContent":["/**\n * @module @nhtio/adk/batteries/vector/orama\n */\n\nimport { clamp01 } from '../helpers'\nimport { evaluateFilter } from '../filters'\nimport { BaseVectorStore } from '../contract'\nimport { validateRecords } from '../validation'\nimport { E_VECTOR_STORE_DRIVER_UNAVAILABLE, E_VECTOR_STORE_DIMENSION_MISMATCH } from '../exceptions'\nimport type { VectorMatch, VectorStoreCapabilities, BaseVectorStoreOptions } from '../types'\nimport type { SearchPlan, UpsertPlan, DeletePlan, CollectionSpec, Projection } from '../plan'\n\ninterface StoredRow {\n  id: string\n  vector?: number[]\n  document?: string\n  metadata?: Record<string, unknown>\n}\n\nexport interface OramaVectorStoreOptions extends BaseVectorStoreOptions {}\n\nexport class OramaVectorStore extends BaseVectorStore {\n  readonly capabilities: VectorStoreCapabilities = {\n    transactions: false,\n    namedVectors: false,\n    rename: true,\n    rawSql: false,\n    builtInEncoding: false,\n    // Strongly consistent: a write is visible on resolve, so the option is a no-op.\n    consistency: { configurable: false, default: 'strong', modes: ['strong'] },\n  }\n  #collections = new Map<string, Map<string, StoredRow>>()\n  #dims = new Map<string, number>()\n  #orama: any | null = null\n  #dbs = new Map<string, any>()\n\n  static isAvailable(): boolean {\n    return true\n  }\n  isAvailable(): boolean {\n    return true\n  }\n\n  async connect(): Promise<void> {\n    if (!this.#orama) {\n      try {\n        const mod = await import('@orama/orama')\n        this.#orama = mod\n      } catch {\n        throw new E_VECTOR_STORE_DRIVER_UNAVAILABLE(['@orama/orama'])\n      }\n    }\n  }\n\n  async close(): Promise<void> {\n    this.#dbs.clear()\n  }\n\n  async createCollection(spec: CollectionSpec, ifNotExists: boolean): Promise<void> {\n    if (this.#collections.has(spec.collection)) {\n      if (!ifNotExists) {\n        throw new Error(`Collection \"${spec.collection}\" already exists`)\n      }\n      return\n    }\n    this.#collections.set(spec.collection, new Map())\n    this.#dims.set(spec.collection, spec.vector.dimensions)\n    if (this.#orama) {\n      const db = this.#orama.create({\n        schema: {\n          embedding: `vector[${spec.vector.dimensions}]`,\n          metadataJson: 'string',\n        },\n      })\n      this.#dbs.set(spec.collection, db)\n    }\n  }\n\n  async dropCollection(collection: string, ifExists: boolean): Promise<void> {\n    if (!this.#collections.has(collection)) {\n      if (!ifExists) {\n        throw new Error(`Collection \"${collection}\" does not exist`)\n      }\n      return\n    }\n    this.#collections.delete(collection)\n    this.#dims.delete(collection)\n    this.#dbs.delete(collection)\n  }\n\n  async hasCollection(collection: string): Promise<boolean> {\n    return this.#collections.has(collection)\n  }\n\n  async renameCollection(from: string, to: string): Promise<void> {\n    const m = this.#collections.get(from)\n    if (!m) {\n      throw new Error(`Collection \"${from}\" does not exist`)\n    }\n    if (this.#collections.has(to)) {\n      throw new Error(`Collection \"${to}\" already exists`)\n    }\n    this.#collections.set(to, m)\n    this.#collections.delete(from)\n    const d = this.#dims.get(from)\n    if (d !== undefined) {\n      this.#dims.set(to, d)\n      this.#dims.delete(from)\n    }\n    const db = this.#dbs.get(from)\n    if (db) {\n      this.#dbs.set(to, db)\n      this.#dbs.delete(from)\n    }\n  }\n\n  #coll(collection: string): Map<string, StoredRow> {\n    let m = this.#collections.get(collection)\n    if (!m) {\n      m = new Map()\n      this.#collections.set(collection, m)\n    }\n    return m\n  }\n\n  #getDb(collection: string): any {\n    return this.#dbs.get(collection)\n  }\n\n  async executeUpsert(plan: UpsertPlan): Promise<void> {\n    if (plan.records.length === 0) return\n    validateRecords(plan.records)\n    const expected = this.#dims.get(plan.collection)\n    const m = this.#coll(plan.collection)\n    const db = this.#getDb(plan.collection)\n    for (const r of plan.records) {\n      let vector = r.vector\n      if (!vector && r.document) {\n        const [v] = await this.encode([r.document], 'document')\n        vector = v\n      }\n      if (vector && expected !== undefined && vector.length !== expected) {\n        throw new E_VECTOR_STORE_DIMENSION_MISMATCH([expected, vector.length])\n      }\n      const row: StoredRow = {\n        id: r.id,\n        vector,\n        document: r.document,\n        metadata: r.metadata as Record<string, unknown> | undefined,\n      }\n      m.set(r.id, row)\n      if (db) {\n        const doc: any = {\n          id: r.id,\n          metadataJson: JSON.stringify(row.metadata ?? {}),\n          embedding: vector,\n        }\n        await this.#orama.insert(db, doc)\n      }\n    }\n  }\n\n  async executeDelete(plan: DeletePlan): Promise<void> {\n    const m = this.#coll(plan.collection)\n    const db = this.#getDb(plan.collection)\n    if (plan.ids) {\n      for (const id of plan.ids) m.delete(id)\n      if (db) {\n        for (const id of plan.ids) await this.#orama.remove(db, id)\n      }\n      return\n    }\n    if (plan.filter) {\n      for (const [id, row] of [...m]) {\n        if (evaluateFilter(plan.filter!, (row.metadata ?? {}) as any)) {\n          m.delete(id)\n          if (db) await this.#orama.remove(db, id)\n        }\n      }\n      return\n    }\n    m.clear()\n    if (db) {\n      const allIds = this.#orama.select(db, { limit: 10000 })\n      if (allIds.hits) {\n        for (const h of allIds.hits) await this.#orama.remove(db, h.id)\n      }\n    }\n  }\n\n  async executeSearch(plan: SearchPlan): Promise<VectorMatch[]> {\n    const m = this.#coll(plan.collection)\n    const db = this.#getDb(plan.collection)\n    let rows = [...m.values()]\n    let scored: { row: StoredRow; score?: number }[]\n\n    if (plan.near && 'vector' in plan.near) {\n      if (db) {\n        const r = await this.#orama.search(db, {\n          mode: 'vector',\n          vector: { value: plan.near.vector, property: 'embedding' },\n          limit: plan.topK + (plan.offset ?? 0),\n          similarity: 0,\n        })\n        const ids = new Set(r.hits.map((h: any) => h.id))\n        const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]))\n        rows = r.hits.map((h: any) => idMap.get(h.id)).filter(Boolean) as StoredRow[]\n      } else {\n        rows = rows.filter(() => true)\n      }\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({\n        row,\n        score: this.#cosineLikeScore(plan.near as { vector: number[] }, row, plan),\n      }))\n      scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0))\n    } else if (plan.near && 'serverText' in plan.near) {\n      const [qv] = await this.encode([plan.near.serverText], 'query')\n      if (db) {\n        const r = await this.#orama.search(db, {\n          mode: 'vector',\n          vector: { value: qv, property: 'embedding' },\n          limit: plan.topK + (plan.offset ?? 0),\n          similarity: 0,\n        })\n        const ids = new Set(r.hits.map((h: any) => h.id))\n        const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]))\n        rows = r.hits.map((h: any) => idMap.get(h.id)).filter(Boolean) as StoredRow[]\n      }\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({ row, score: this.#scoreVec(qv, row.vector, plan) }))\n      scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0))\n    } else if (plan.near && 'id' in plan.near) {\n      const ref = m.get(plan.near.id)\n      const qv = ref?.vector\n      if (db && qv) {\n        const r = await this.#orama.search(db, {\n          mode: 'vector',\n          vector: { value: qv, property: 'embedding' },\n          limit: plan.topK + (plan.offset ?? 0),\n          similarity: 0,\n        })\n        const ids = new Set(r.hits.map((h: any) => h.id))\n        const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]))\n        rows = r.hits.map((h: any) => idMap.get(h.id)).filter(Boolean) as StoredRow[]\n      }\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({\n        row,\n        score: qv ? this.#scoreVec(qv, row.vector, plan) : undefined,\n      }))\n      scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0))\n    } else {\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({ row }))\n    }\n\n    const limited = scored.slice(plan.offset ?? 0, (plan.offset ?? 0) + plan.topK)\n    return limited.map(({ row, score }) => this.#project(row, score, plan.projection))\n  }\n\n  #scoreVec(q: number[], v: number[] | undefined, _plan: SearchPlan): number | undefined {\n    if (!v) return undefined\n    const metric = this.options.metric ?? 'cosine'\n    let raw: number\n    if (metric === 'cosine') {\n      const dot = this.#dot(q, v)\n      const na = Math.sqrt(this.#dot(q, q))\n      const nb = Math.sqrt(this.#dot(v, v))\n      raw = na && nb ? dot / (na * nb) : 0\n      return clamp01(raw)\n    }\n    if (metric === 'dot') {\n      raw = this.#dot(q, v)\n      return clamp01(raw)\n    }\n    let s = 0\n    for (let i = 0; i < Math.min(q.length, v.length); i++) {\n      const d = q[i] - v[i]\n      s += d * d\n    }\n    raw = Math.sqrt(s)\n    return clamp01(1 / (1 + raw))\n  }\n\n  #cosineLikeScore(\n    near: { vector: number[] },\n    row: StoredRow,\n    plan: SearchPlan\n  ): number | undefined {\n    return this.#scoreVec(near.vector, row.vector, plan)\n  }\n\n  #dot(a: number[], b: number[]): number {\n    let s = 0\n    for (let i = 0; i < Math.min(a.length, b.length); i++) s += a[i] * b[i]\n    return s\n  }\n\n  #project(row: StoredRow, score: number | undefined, p: Projection): VectorMatch {\n    const out: VectorMatch = {}\n    if (p.id) out.id = row.id\n    if (score !== undefined) out.score = score\n    if (p.vector && row.vector) out.vector = row.vector\n    if (p.document && row.document !== undefined) out.document = row.document\n    if (p.metadata && row.metadata !== undefined) out.metadata = row.metadata as any\n    return out\n  }\n}\n"],"mappings":";;;;;;;;;;;AAqBA,IAAa,mBAAb,cAAsC,kCAAA,gBAAgB;CACpD,eAAiD;EAC/C,cAAc;EACd,cAAc;EACd,QAAQ;EACR,QAAQ;EACR,iBAAiB;EAEjB,aAAa;GAAE,cAAc;GAAO,SAAS;GAAU,OAAO,CAAC,QAAQ;EAAE;CAC3E;CACA,+BAAe,IAAI,IAAoC;CACvD,wBAAQ,IAAI,IAAoB;CAChC,SAAqB;CACrB,uBAAO,IAAI,IAAiB;CAE5B,OAAO,cAAuB;EAC5B,OAAO;CACT;CACA,cAAuB;EACrB,OAAO;CACT;CAEA,MAAM,UAAyB;EAC7B,IAAI,CAAC,KAAKA,QACR,IAAI;GACF,MAAM,MAAM,MAAM,OAAO;GACzB,KAAKA,SAAS;EAChB,QAAQ;GACN,MAAM,IAAI,oCAAA,kCAAkC,CAAC,cAAc,CAAC;EAC9D;CAEJ;CAEA,MAAM,QAAuB;EAC3B,KAAKC,KAAK,MAAM;CAClB;CAEA,MAAM,iBAAiB,MAAsB,aAAqC;EAChF,IAAI,KAAKC,aAAa,IAAI,KAAK,UAAU,GAAG;GAC1C,IAAI,CAAC,aACH,MAAM,IAAI,MAAM,eAAe,KAAK,WAAW,iBAAiB;GAElE;EACF;EACA,KAAKA,aAAa,IAAI,KAAK,4BAAY,IAAI,IAAI,CAAC;EAChD,KAAKC,MAAM,IAAI,KAAK,YAAY,KAAK,OAAO,UAAU;EACtD,IAAI,KAAKH,QAAQ;GACf,MAAM,KAAK,KAAKA,OAAO,OAAO,EAC5B,QAAQ;IACN,WAAW,UAAU,KAAK,OAAO,WAAW;IAC5C,cAAc;GAChB,EACF,CAAC;GACD,KAAKC,KAAK,IAAI,KAAK,YAAY,EAAE;EACnC;CACF;CAEA,MAAM,eAAe,YAAoB,UAAkC;EACzE,IAAI,CAAC,KAAKC,aAAa,IAAI,UAAU,GAAG;GACtC,IAAI,CAAC,UACH,MAAM,IAAI,MAAM,eAAe,WAAW,iBAAiB;GAE7D;EACF;EACA,KAAKA,aAAa,OAAO,UAAU;EACnC,KAAKC,MAAM,OAAO,UAAU;EAC5B,KAAKF,KAAK,OAAO,UAAU;CAC7B;CAEA,MAAM,cAAc,YAAsC;EACxD,OAAO,KAAKC,aAAa,IAAI,UAAU;CACzC;CAEA,MAAM,iBAAiB,MAAc,IAA2B;EAC9D,MAAM,IAAI,KAAKA,aAAa,IAAI,IAAI;EACpC,IAAI,CAAC,GACH,MAAM,IAAI,MAAM,eAAe,KAAK,iBAAiB;EAEvD,IAAI,KAAKA,aAAa,IAAI,EAAE,GAC1B,MAAM,IAAI,MAAM,eAAe,GAAG,iBAAiB;EAErD,KAAKA,aAAa,IAAI,IAAI,CAAC;EAC3B,KAAKA,aAAa,OAAO,IAAI;EAC7B,MAAM,IAAI,KAAKC,MAAM,IAAI,IAAI;EAC7B,IAAI,MAAM,KAAA,GAAW;GACnB,KAAKA,MAAM,IAAI,IAAI,CAAC;GACpB,KAAKA,MAAM,OAAO,IAAI;EACxB;EACA,MAAM,KAAK,KAAKF,KAAK,IAAI,IAAI;EAC7B,IAAI,IAAI;GACN,KAAKA,KAAK,IAAI,IAAI,EAAE;GACpB,KAAKA,KAAK,OAAO,IAAI;EACvB;CACF;CAEA,MAAM,YAA4C;EAChD,IAAI,IAAI,KAAKC,aAAa,IAAI,UAAU;EACxC,IAAI,CAAC,GAAG;GACN,oBAAI,IAAI,IAAI;GACZ,KAAKA,aAAa,IAAI,YAAY,CAAC;EACrC;EACA,OAAO;CACT;CAEA,OAAO,YAAyB;EAC9B,OAAO,KAAKD,KAAK,IAAI,UAAU;CACjC;CAEA,MAAM,cAAc,MAAiC;EACnD,IAAI,KAAK,QAAQ,WAAW,GAAG;EAC/B,oCAAA,gBAAgB,KAAK,OAAO;EAC5B,MAAM,WAAW,KAAKE,MAAM,IAAI,KAAK,UAAU;EAC/C,MAAM,IAAI,KAAKC,MAAM,KAAK,UAAU;EACpC,MAAM,KAAK,KAAKC,OAAO,KAAK,UAAU;EACtC,KAAK,MAAM,KAAK,KAAK,SAAS;GAC5B,IAAI,SAAS,EAAE;GACf,IAAI,CAAC,UAAU,EAAE,UAAU;IACzB,MAAM,CAAC,KAAK,MAAM,KAAK,OAAO,CAAC,EAAE,QAAQ,GAAG,UAAU;IACtD,SAAS;GACX;GACA,IAAI,UAAU,aAAa,KAAA,KAAa,OAAO,WAAW,UACxD,MAAM,IAAI,oCAAA,kCAAkC,CAAC,UAAU,OAAO,MAAM,CAAC;GAEvE,MAAM,MAAiB;IACrB,IAAI,EAAE;IACN;IACA,UAAU,EAAE;IACZ,UAAU,EAAE;GACd;GACA,EAAE,IAAI,EAAE,IAAI,GAAG;GACf,IAAI,IAAI;IACN,MAAM,MAAW;KACf,IAAI,EAAE;KACN,cAAc,KAAK,UAAU,IAAI,YAAY,CAAC,CAAC;KAC/C,WAAW;IACb;IACA,MAAM,KAAKL,OAAO,OAAO,IAAI,GAAG;GAClC;EACF;CACF;CAEA,MAAM,cAAc,MAAiC;EACnD,MAAM,IAAI,KAAKI,MAAM,KAAK,UAAU;EACpC,MAAM,KAAK,KAAKC,OAAO,KAAK,UAAU;EACtC,IAAI,KAAK,KAAK;GACZ,KAAK,MAAM,MAAM,KAAK,KAAK,EAAE,OAAO,EAAE;GACtC,IAAI,IACF,KAAK,MAAM,MAAM,KAAK,KAAK,MAAM,KAAKL,OAAO,OAAO,IAAI,EAAE;GAE5D;EACF;EACA,IAAI,KAAK,QAAQ;GACf,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,GAAG,CAAC,GAC3B,IAAI,iCAAA,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,GAAG;IAC7D,EAAE,OAAO,EAAE;IACX,IAAI,IAAI,MAAM,KAAKA,OAAO,OAAO,IAAI,EAAE;GACzC;GAEF;EACF;EACA,EAAE,MAAM;EACR,IAAI,IAAI;GACN,MAAM,SAAS,KAAKA,OAAO,OAAO,IAAI,EAAE,OAAO,IAAM,CAAC;GACtD,IAAI,OAAO,MACT,KAAK,MAAM,KAAK,OAAO,MAAM,MAAM,KAAKA,OAAO,OAAO,IAAI,EAAE,EAAE;EAElE;CACF;CAEA,MAAM,cAAc,MAA0C;EAC5D,MAAM,IAAI,KAAKI,MAAM,KAAK,UAAU;EACpC,MAAM,KAAK,KAAKC,OAAO,KAAK,UAAU;EACtC,IAAI,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC;EACzB,IAAI;EAEJ,IAAI,KAAK,QAAQ,YAAY,KAAK,MAAM;GACtC,IAAI,IAAI;IACN,MAAM,IAAI,MAAM,KAAKL,OAAO,OAAO,IAAI;KACrC,MAAM;KACN,QAAQ;MAAE,OAAO,KAAK,KAAK;MAAQ,UAAU;KAAY;KACzD,OAAO,KAAK,QAAQ,KAAK,UAAU;KACnC,YAAY;IACd,CAAC;IACD,MAAM,MAAM,IAAI,IAAI,EAAE,KAAK,KAAK,MAAW,EAAE,EAAE,CAAC;IAChD,MAAM,QAAQ,IAAI,IAAI,KAAK,QAAQ,QAAQ,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACvF,OAAO,EAAE,KAAK,KAAK,MAAW,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,OAAO,OAAO;GAC/D,OACE,OAAO,KAAK,aAAa,IAAI;GAE/B,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,iCAAA,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS;IAC1B;IACA,OAAO,KAAKM,iBAAiB,KAAK,MAA8B,KAAK,IAAI;GAC3E,EAAE;GACF,OAAO,MAAM,GAAG,OAAO,EAAE,SAAS,MAAM,EAAE,SAAS,EAAE;EACvD,OAAO,IAAI,KAAK,QAAQ,gBAAgB,KAAK,MAAM;GACjD,MAAM,CAAC,MAAM,MAAM,KAAK,OAAO,CAAC,KAAK,KAAK,UAAU,GAAG,OAAO;GAC9D,IAAI,IAAI;IACN,MAAM,IAAI,MAAM,KAAKN,OAAO,OAAO,IAAI;KACrC,MAAM;KACN,QAAQ;MAAE,OAAO;MAAI,UAAU;KAAY;KAC3C,OAAO,KAAK,QAAQ,KAAK,UAAU;KACnC,YAAY;IACd,CAAC;IACD,MAAM,MAAM,IAAI,IAAI,EAAE,KAAK,KAAK,MAAW,EAAE,EAAE,CAAC;IAChD,MAAM,QAAQ,IAAI,IAAI,KAAK,QAAQ,QAAQ,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACvF,OAAO,EAAE,KAAK,KAAK,MAAW,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,OAAO,OAAO;GAC/D;GACA,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,iCAAA,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS;IAAE;IAAK,OAAO,KAAKO,UAAU,IAAI,IAAI,QAAQ,IAAI;GAAE,EAAE;GACjF,OAAO,MAAM,GAAG,OAAO,EAAE,SAAS,MAAM,EAAE,SAAS,EAAE;EACvD,OAAO,IAAI,KAAK,QAAQ,QAAQ,KAAK,MAAM;GAEzC,MAAM,KADM,EAAE,IAAI,KAAK,KAAK,EACjB,GAAK;GAChB,IAAI,MAAM,IAAI;IACZ,MAAM,IAAI,MAAM,KAAKP,OAAO,OAAO,IAAI;KACrC,MAAM;KACN,QAAQ;MAAE,OAAO;MAAI,UAAU;KAAY;KAC3C,OAAO,KAAK,QAAQ,KAAK,UAAU;KACnC,YAAY;IACd,CAAC;IACD,MAAM,MAAM,IAAI,IAAI,EAAE,KAAK,KAAK,MAAW,EAAE,EAAE,CAAC;IAChD,MAAM,QAAQ,IAAI,IAAI,KAAK,QAAQ,QAAQ,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACvF,OAAO,EAAE,KAAK,KAAK,MAAW,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,OAAO,OAAO;GAC/D;GACA,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,iCAAA,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS;IAC1B;IACA,OAAO,KAAK,KAAKO,UAAU,IAAI,IAAI,QAAQ,IAAI,IAAI,KAAA;GACrD,EAAE;GACF,OAAO,MAAM,GAAG,OAAO,EAAE,SAAS,MAAM,EAAE,SAAS,EAAE;EACvD,OAAO;GACL,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,iCAAA,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS,EAAE,IAAI,EAAE;EACtC;EAGA,OADgB,OAAO,MAAM,KAAK,UAAU,IAAI,KAAK,UAAU,KAAK,KAAK,IAClE,EAAQ,KAAK,EAAE,KAAK,YAAY,KAAKC,SAAS,KAAK,OAAO,KAAK,UAAU,CAAC;CACnF;CAEA,UAAU,GAAa,GAAyB,OAAuC;EACrF,IAAI,CAAC,GAAG,OAAO,KAAA;EACf,MAAM,SAAS,KAAK,QAAQ,UAAU;EACtC,IAAI;EACJ,IAAI,WAAW,UAAU;GACvB,MAAM,MAAM,KAAKC,KAAK,GAAG,CAAC;GAC1B,MAAM,KAAK,KAAK,KAAK,KAAKA,KAAK,GAAG,CAAC,CAAC;GACpC,MAAM,KAAK,KAAK,KAAK,KAAKA,KAAK,GAAG,CAAC,CAAC;GACpC,MAAM,MAAM,KAAK,OAAO,KAAK,MAAM;GACnC,OAAO,iCAAA,QAAQ,GAAG;EACpB;EACA,IAAI,WAAW,OAAO;GACpB,MAAM,KAAKA,KAAK,GAAG,CAAC;GACpB,OAAO,iCAAA,QAAQ,GAAG;EACpB;EACA,IAAI,IAAI;EACR,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,KAAK;GACrD,MAAM,IAAI,EAAE,KAAK,EAAE;GACnB,KAAK,IAAI;EACX;EACA,MAAM,KAAK,KAAK,CAAC;EACjB,OAAO,iCAAA,QAAQ,KAAK,IAAI,IAAI;CAC9B;CAEA,iBACE,MACA,KACA,MACoB;EACpB,OAAO,KAAKF,UAAU,KAAK,QAAQ,IAAI,QAAQ,IAAI;CACrD;CAEA,KAAK,GAAa,GAAqB;EACrC,IAAI,IAAI;EACR,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,KAAK,KAAK,EAAE,KAAK,EAAE;EACrE,OAAO;CACT;CAEA,SAAS,KAAgB,OAA2B,GAA4B;EAC9E,MAAM,MAAmB,CAAC;EAC1B,IAAI,EAAE,IAAI,IAAI,KAAK,IAAI;EACvB,IAAI,UAAU,KAAA,GAAW,IAAI,QAAQ;EACrC,IAAI,EAAE,UAAU,IAAI,QAAQ,IAAI,SAAS,IAAI;EAC7C,IAAI,EAAE,YAAY,IAAI,aAAa,KAAA,GAAW,IAAI,WAAW,IAAI;EACjE,IAAI,EAAE,YAAY,IAAI,aAAa,KAAA,GAAW,IAAI,WAAW,IAAI;EACjE,OAAO;CACT;AACF"}

package/batteries/vector/orama.mjs

@@ -0,0 +1,270 @@
+import { evaluateFilter } from "./filters.mjs";
+import { clamp01 } from "./helpers.mjs";
+import { E_VECTOR_STORE_DIMENSION_MISMATCH, E_VECTOR_STORE_DRIVER_UNAVAILABLE } from "./exceptions.mjs";
+import { BaseVectorStore } from "./contract.mjs";
+import { validateRecords } from "./validation.mjs";
+//#region src/batteries/vector/orama/index.ts
+/**
+* @module @nhtio/adk/batteries/vector/orama
+*/
+var OramaVectorStore = class extends BaseVectorStore {
+	capabilities = {
+		transactions: false,
+		namedVectors: false,
+		rename: true,
+		rawSql: false,
+		builtInEncoding: false,
+		consistency: {
+			configurable: false,
+			default: "strong",
+			modes: ["strong"]
+		}
+	};
+	#collections = /* @__PURE__ */ new Map();
+	#dims = /* @__PURE__ */ new Map();
+	#orama = null;
+	#dbs = /* @__PURE__ */ new Map();
+	static isAvailable() {
+		return true;
+	}
+	isAvailable() {
+		return true;
+	}
+	async connect() {
+		if (!this.#orama) try {
+			const mod = await import("@orama/orama");
+			this.#orama = mod;
+		} catch {
+			throw new E_VECTOR_STORE_DRIVER_UNAVAILABLE(["@orama/orama"]);
+		}
+	}
+	async close() {
+		this.#dbs.clear();
+	}
+	async createCollection(spec, ifNotExists) {
+		if (this.#collections.has(spec.collection)) {
+			if (!ifNotExists) throw new Error(`Collection "${spec.collection}" already exists`);
+			return;
+		}
+		this.#collections.set(spec.collection, /* @__PURE__ */ new Map());
+		this.#dims.set(spec.collection, spec.vector.dimensions);
+		if (this.#orama) {
+			const db = this.#orama.create({ schema: {
+				embedding: `vector[${spec.vector.dimensions}]`,
+				metadataJson: "string"
+			} });
+			this.#dbs.set(spec.collection, db);
+		}
+	}
+	async dropCollection(collection, ifExists) {
+		if (!this.#collections.has(collection)) {
+			if (!ifExists) throw new Error(`Collection "${collection}" does not exist`);
+			return;
+		}
+		this.#collections.delete(collection);
+		this.#dims.delete(collection);
+		this.#dbs.delete(collection);
+	}
+	async hasCollection(collection) {
+		return this.#collections.has(collection);
+	}
+	async renameCollection(from, to) {
+		const m = this.#collections.get(from);
+		if (!m) throw new Error(`Collection "${from}" does not exist`);
+		if (this.#collections.has(to)) throw new Error(`Collection "${to}" already exists`);
+		this.#collections.set(to, m);
+		this.#collections.delete(from);
+		const d = this.#dims.get(from);
+		if (d !== void 0) {
+			this.#dims.set(to, d);
+			this.#dims.delete(from);
+		}
+		const db = this.#dbs.get(from);
+		if (db) {
+			this.#dbs.set(to, db);
+			this.#dbs.delete(from);
+		}
+	}
+	#coll(collection) {
+		let m = this.#collections.get(collection);
+		if (!m) {
+			m = /* @__PURE__ */ new Map();
+			this.#collections.set(collection, m);
+		}
+		return m;
+	}
+	#getDb(collection) {
+		return this.#dbs.get(collection);
+	}
+	async executeUpsert(plan) {
+		if (plan.records.length === 0) return;
+		validateRecords(plan.records);
+		const expected = this.#dims.get(plan.collection);
+		const m = this.#coll(plan.collection);
+		const db = this.#getDb(plan.collection);
+		for (const r of plan.records) {
+			let vector = r.vector;
+			if (!vector && r.document) {
+				const [v] = await this.encode([r.document], "document");
+				vector = v;
+			}
+			if (vector && expected !== void 0 && vector.length !== expected) throw new E_VECTOR_STORE_DIMENSION_MISMATCH([expected, vector.length]);
+			const row = {
+				id: r.id,
+				vector,
+				document: r.document,
+				metadata: r.metadata
+			};
+			m.set(r.id, row);
+			if (db) {
+				const doc = {
+					id: r.id,
+					metadataJson: JSON.stringify(row.metadata ?? {}),
+					embedding: vector
+				};
+				await this.#orama.insert(db, doc);
+			}
+		}
+	}
+	async executeDelete(plan) {
+		const m = this.#coll(plan.collection);
+		const db = this.#getDb(plan.collection);
+		if (plan.ids) {
+			for (const id of plan.ids) m.delete(id);
+			if (db) for (const id of plan.ids) await this.#orama.remove(db, id);
+			return;
+		}
+		if (plan.filter) {
+			for (const [id, row] of [...m]) if (evaluateFilter(plan.filter, row.metadata ?? {})) {
+				m.delete(id);
+				if (db) await this.#orama.remove(db, id);
+			}
+			return;
+		}
+		m.clear();
+		if (db) {
+			const allIds = this.#orama.select(db, { limit: 1e4 });
+			if (allIds.hits) for (const h of allIds.hits) await this.#orama.remove(db, h.id);
+		}
+	}
+	async executeSearch(plan) {
+		const m = this.#coll(plan.collection);
+		const db = this.#getDb(plan.collection);
+		let rows = [...m.values()];
+		let scored;
+		if (plan.near && "vector" in plan.near) {
+			if (db) {
+				const r = await this.#orama.search(db, {
+					mode: "vector",
+					vector: {
+						value: plan.near.vector,
+						property: "embedding"
+					},
+					limit: plan.topK + (plan.offset ?? 0),
+					similarity: 0
+				});
+				const ids = new Set(r.hits.map((h) => h.id));
+				const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]));
+				rows = r.hits.map((h) => idMap.get(h.id)).filter(Boolean);
+			} else rows = rows.filter(() => true);
+			if (plan.filter) rows = rows.filter((row) => evaluateFilter(plan.filter, row.metadata ?? {}));
+			scored = rows.map((row) => ({
+				row,
+				score: this.#cosineLikeScore(plan.near, row, plan)
+			}));
+			scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0));
+		} else if (plan.near && "serverText" in plan.near) {
+			const [qv] = await this.encode([plan.near.serverText], "query");
+			if (db) {
+				const r = await this.#orama.search(db, {
+					mode: "vector",
+					vector: {
+						value: qv,
+						property: "embedding"
+					},
+					limit: plan.topK + (plan.offset ?? 0),
+					similarity: 0
+				});
+				const ids = new Set(r.hits.map((h) => h.id));
+				const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]));
+				rows = r.hits.map((h) => idMap.get(h.id)).filter(Boolean);
+			}
+			if (plan.filter) rows = rows.filter((row) => evaluateFilter(plan.filter, row.metadata ?? {}));
+			scored = rows.map((row) => ({
+				row,
+				score: this.#scoreVec(qv, row.vector, plan)
+			}));
+			scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0));
+		} else if (plan.near && "id" in plan.near) {
+			const qv = m.get(plan.near.id)?.vector;
+			if (db && qv) {
+				const r = await this.#orama.search(db, {
+					mode: "vector",
+					vector: {
+						value: qv,
+						property: "embedding"
+					},
+					limit: plan.topK + (plan.offset ?? 0),
+					similarity: 0
+				});
+				const ids = new Set(r.hits.map((h) => h.id));
+				const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]));
+				rows = r.hits.map((h) => idMap.get(h.id)).filter(Boolean);
+			}
+			if (plan.filter) rows = rows.filter((row) => evaluateFilter(plan.filter, row.metadata ?? {}));
+			scored = rows.map((row) => ({
+				row,
+				score: qv ? this.#scoreVec(qv, row.vector, plan) : void 0
+			}));
+			scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0));
+		} else {
+			if (plan.filter) rows = rows.filter((row) => evaluateFilter(plan.filter, row.metadata ?? {}));
+			scored = rows.map((row) => ({ row }));
+		}
+		return scored.slice(plan.offset ?? 0, (plan.offset ?? 0) + plan.topK).map(({ row, score }) => this.#project(row, score, plan.projection));
+	}
+	#scoreVec(q, v, _plan) {
+		if (!v) return void 0;
+		const metric = this.options.metric ?? "cosine";
+		let raw;
+		if (metric === "cosine") {
+			const dot = this.#dot(q, v);
+			const na = Math.sqrt(this.#dot(q, q));
+			const nb = Math.sqrt(this.#dot(v, v));
+			raw = na && nb ? dot / (na * nb) : 0;
+			return clamp01(raw);
+		}
+		if (metric === "dot") {
+			raw = this.#dot(q, v);
+			return clamp01(raw);
+		}
+		let s = 0;
+		for (let i = 0; i < Math.min(q.length, v.length); i++) {
+			const d = q[i] - v[i];
+			s += d * d;
+		}
+		raw = Math.sqrt(s);
+		return clamp01(1 / (1 + raw));
+	}
+	#cosineLikeScore(near, row, plan) {
+		return this.#scoreVec(near.vector, row.vector, plan);
+	}
+	#dot(a, b) {
+		let s = 0;
+		for (let i = 0; i < Math.min(a.length, b.length); i++) s += a[i] * b[i];
+		return s;
+	}
+	#project(row, score, p) {
+		const out = {};
+		if (p.id) out.id = row.id;
+		if (score !== void 0) out.score = score;
+		if (p.vector && row.vector) out.vector = row.vector;
+		if (p.document && row.document !== void 0) out.document = row.document;
+		if (p.metadata && row.metadata !== void 0) out.metadata = row.metadata;
+		return out;
+	}
+};
+//#endregion
+export { OramaVectorStore };
+
+//# sourceMappingURL=orama.mjs.map

package/batteries/vector/orama.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"orama.mjs","names":["#orama","#dbs","#collections","#dims","#coll","#getDb","#cosineLikeScore","#scoreVec","#project","#dot"],"sources":["../../../src/batteries/vector/orama/index.ts"],"sourcesContent":["/**\n * @module @nhtio/adk/batteries/vector/orama\n */\n\nimport { clamp01 } from '../helpers'\nimport { evaluateFilter } from '../filters'\nimport { BaseVectorStore } from '../contract'\nimport { validateRecords } from '../validation'\nimport { E_VECTOR_STORE_DRIVER_UNAVAILABLE, E_VECTOR_STORE_DIMENSION_MISMATCH } from '../exceptions'\nimport type { VectorMatch, VectorStoreCapabilities, BaseVectorStoreOptions } from '../types'\nimport type { SearchPlan, UpsertPlan, DeletePlan, CollectionSpec, Projection } from '../plan'\n\ninterface StoredRow {\n  id: string\n  vector?: number[]\n  document?: string\n  metadata?: Record<string, unknown>\n}\n\nexport interface OramaVectorStoreOptions extends BaseVectorStoreOptions {}\n\nexport class OramaVectorStore extends BaseVectorStore {\n  readonly capabilities: VectorStoreCapabilities = {\n    transactions: false,\n    namedVectors: false,\n    rename: true,\n    rawSql: false,\n    builtInEncoding: false,\n    // Strongly consistent: a write is visible on resolve, so the option is a no-op.\n    consistency: { configurable: false, default: 'strong', modes: ['strong'] },\n  }\n  #collections = new Map<string, Map<string, StoredRow>>()\n  #dims = new Map<string, number>()\n  #orama: any | null = null\n  #dbs = new Map<string, any>()\n\n  static isAvailable(): boolean {\n    return true\n  }\n  isAvailable(): boolean {\n    return true\n  }\n\n  async connect(): Promise<void> {\n    if (!this.#orama) {\n      try {\n        const mod = await import('@orama/orama')\n        this.#orama = mod\n      } catch {\n        throw new E_VECTOR_STORE_DRIVER_UNAVAILABLE(['@orama/orama'])\n      }\n    }\n  }\n\n  async close(): Promise<void> {\n    this.#dbs.clear()\n  }\n\n  async createCollection(spec: CollectionSpec, ifNotExists: boolean): Promise<void> {\n    if (this.#collections.has(spec.collection)) {\n      if (!ifNotExists) {\n        throw new Error(`Collection \"${spec.collection}\" already exists`)\n      }\n      return\n    }\n    this.#collections.set(spec.collection, new Map())\n    this.#dims.set(spec.collection, spec.vector.dimensions)\n    if (this.#orama) {\n      const db = this.#orama.create({\n        schema: {\n          embedding: `vector[${spec.vector.dimensions}]`,\n          metadataJson: 'string',\n        },\n      })\n      this.#dbs.set(spec.collection, db)\n    }\n  }\n\n  async dropCollection(collection: string, ifExists: boolean): Promise<void> {\n    if (!this.#collections.has(collection)) {\n      if (!ifExists) {\n        throw new Error(`Collection \"${collection}\" does not exist`)\n      }\n      return\n    }\n    this.#collections.delete(collection)\n    this.#dims.delete(collection)\n    this.#dbs.delete(collection)\n  }\n\n  async hasCollection(collection: string): Promise<boolean> {\n    return this.#collections.has(collection)\n  }\n\n  async renameCollection(from: string, to: string): Promise<void> {\n    const m = this.#collections.get(from)\n    if (!m) {\n      throw new Error(`Collection \"${from}\" does not exist`)\n    }\n    if (this.#collections.has(to)) {\n      throw new Error(`Collection \"${to}\" already exists`)\n    }\n    this.#collections.set(to, m)\n    this.#collections.delete(from)\n    const d = this.#dims.get(from)\n    if (d !== undefined) {\n      this.#dims.set(to, d)\n      this.#dims.delete(from)\n    }\n    const db = this.#dbs.get(from)\n    if (db) {\n      this.#dbs.set(to, db)\n      this.#dbs.delete(from)\n    }\n  }\n\n  #coll(collection: string): Map<string, StoredRow> {\n    let m = this.#collections.get(collection)\n    if (!m) {\n      m = new Map()\n      this.#collections.set(collection, m)\n    }\n    return m\n  }\n\n  #getDb(collection: string): any {\n    return this.#dbs.get(collection)\n  }\n\n  async executeUpsert(plan: UpsertPlan): Promise<void> {\n    if (plan.records.length === 0) return\n    validateRecords(plan.records)\n    const expected = this.#dims.get(plan.collection)\n    const m = this.#coll(plan.collection)\n    const db = this.#getDb(plan.collection)\n    for (const r of plan.records) {\n      let vector = r.vector\n      if (!vector && r.document) {\n        const [v] = await this.encode([r.document], 'document')\n        vector = v\n      }\n      if (vector && expected !== undefined && vector.length !== expected) {\n        throw new E_VECTOR_STORE_DIMENSION_MISMATCH([expected, vector.length])\n      }\n      const row: StoredRow = {\n        id: r.id,\n        vector,\n        document: r.document,\n        metadata: r.metadata as Record<string, unknown> | undefined,\n      }\n      m.set(r.id, row)\n      if (db) {\n        const doc: any = {\n          id: r.id,\n          metadataJson: JSON.stringify(row.metadata ?? {}),\n          embedding: vector,\n        }\n        await this.#orama.insert(db, doc)\n      }\n    }\n  }\n\n  async executeDelete(plan: DeletePlan): Promise<void> {\n    const m = this.#coll(plan.collection)\n    const db = this.#getDb(plan.collection)\n    if (plan.ids) {\n      for (const id of plan.ids) m.delete(id)\n      if (db) {\n        for (const id of plan.ids) await this.#orama.remove(db, id)\n      }\n      return\n    }\n    if (plan.filter) {\n      for (const [id, row] of [...m]) {\n        if (evaluateFilter(plan.filter!, (row.metadata ?? {}) as any)) {\n          m.delete(id)\n          if (db) await this.#orama.remove(db, id)\n        }\n      }\n      return\n    }\n    m.clear()\n    if (db) {\n      const allIds = this.#orama.select(db, { limit: 10000 })\n      if (allIds.hits) {\n        for (const h of allIds.hits) await this.#orama.remove(db, h.id)\n      }\n    }\n  }\n\n  async executeSearch(plan: SearchPlan): Promise<VectorMatch[]> {\n    const m = this.#coll(plan.collection)\n    const db = this.#getDb(plan.collection)\n    let rows = [...m.values()]\n    let scored: { row: StoredRow; score?: number }[]\n\n    if (plan.near && 'vector' in plan.near) {\n      if (db) {\n        const r = await this.#orama.search(db, {\n          mode: 'vector',\n          vector: { value: plan.near.vector, property: 'embedding' },\n          limit: plan.topK + (plan.offset ?? 0),\n          similarity: 0,\n        })\n        const ids = new Set(r.hits.map((h: any) => h.id))\n        const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]))\n        rows = r.hits.map((h: any) => idMap.get(h.id)).filter(Boolean) as StoredRow[]\n      } else {\n        rows = rows.filter(() => true)\n      }\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({\n        row,\n        score: this.#cosineLikeScore(plan.near as { vector: number[] }, row, plan),\n      }))\n      scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0))\n    } else if (plan.near && 'serverText' in plan.near) {\n      const [qv] = await this.encode([plan.near.serverText], 'query')\n      if (db) {\n        const r = await this.#orama.search(db, {\n          mode: 'vector',\n          vector: { value: qv, property: 'embedding' },\n          limit: plan.topK + (plan.offset ?? 0),\n          similarity: 0,\n        })\n        const ids = new Set(r.hits.map((h: any) => h.id))\n        const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]))\n        rows = r.hits.map((h: any) => idMap.get(h.id)).filter(Boolean) as StoredRow[]\n      }\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({ row, score: this.#scoreVec(qv, row.vector, plan) }))\n      scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0))\n    } else if (plan.near && 'id' in plan.near) {\n      const ref = m.get(plan.near.id)\n      const qv = ref?.vector\n      if (db && qv) {\n        const r = await this.#orama.search(db, {\n          mode: 'vector',\n          vector: { value: qv, property: 'embedding' },\n          limit: plan.topK + (plan.offset ?? 0),\n          similarity: 0,\n        })\n        const ids = new Set(r.hits.map((h: any) => h.id))\n        const idMap = new Map(rows.filter((row) => ids.has(row.id)).map((row) => [row.id, row]))\n        rows = r.hits.map((h: any) => idMap.get(h.id)).filter(Boolean) as StoredRow[]\n      }\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({\n        row,\n        score: qv ? this.#scoreVec(qv, row.vector, plan) : undefined,\n      }))\n      scored.sort((a, b) => (b.score ?? 0) - (a.score ?? 0))\n    } else {\n      if (plan.filter)\n        rows = rows.filter((row) => evaluateFilter(plan.filter!, (row.metadata ?? {}) as any))\n      scored = rows.map((row) => ({ row }))\n    }\n\n    const limited = scored.slice(plan.offset ?? 0, (plan.offset ?? 0) + plan.topK)\n    return limited.map(({ row, score }) => this.#project(row, score, plan.projection))\n  }\n\n  #scoreVec(q: number[], v: number[] | undefined, _plan: SearchPlan): number | undefined {\n    if (!v) return undefined\n    const metric = this.options.metric ?? 'cosine'\n    let raw: number\n    if (metric === 'cosine') {\n      const dot = this.#dot(q, v)\n      const na = Math.sqrt(this.#dot(q, q))\n      const nb = Math.sqrt(this.#dot(v, v))\n      raw = na && nb ? dot / (na * nb) : 0\n      return clamp01(raw)\n    }\n    if (metric === 'dot') {\n      raw = this.#dot(q, v)\n      return clamp01(raw)\n    }\n    let s = 0\n    for (let i = 0; i < Math.min(q.length, v.length); i++) {\n      const d = q[i] - v[i]\n      s += d * d\n    }\n    raw = Math.sqrt(s)\n    return clamp01(1 / (1 + raw))\n  }\n\n  #cosineLikeScore(\n    near: { vector: number[] },\n    row: StoredRow,\n    plan: SearchPlan\n  ): number | undefined {\n    return this.#scoreVec(near.vector, row.vector, plan)\n  }\n\n  #dot(a: number[], b: number[]): number {\n    let s = 0\n    for (let i = 0; i < Math.min(a.length, b.length); i++) s += a[i] * b[i]\n    return s\n  }\n\n  #project(row: StoredRow, score: number | undefined, p: Projection): VectorMatch {\n    const out: VectorMatch = {}\n    if (p.id) out.id = row.id\n    if (score !== undefined) out.score = score\n    if (p.vector && row.vector) out.vector = row.vector\n    if (p.document && row.document !== undefined) out.document = row.document\n    if (p.metadata && row.metadata !== undefined) out.metadata = row.metadata as any\n    return out\n  }\n}\n"],"mappings":";;;;;;;;;AAqBA,IAAa,mBAAb,cAAsC,gBAAgB;CACpD,eAAiD;EAC/C,cAAc;EACd,cAAc;EACd,QAAQ;EACR,QAAQ;EACR,iBAAiB;EAEjB,aAAa;GAAE,cAAc;GAAO,SAAS;GAAU,OAAO,CAAC,QAAQ;EAAE;CAC3E;CACA,+BAAe,IAAI,IAAoC;CACvD,wBAAQ,IAAI,IAAoB;CAChC,SAAqB;CACrB,uBAAO,IAAI,IAAiB;CAE5B,OAAO,cAAuB;EAC5B,OAAO;CACT;CACA,cAAuB;EACrB,OAAO;CACT;CAEA,MAAM,UAAyB;EAC7B,IAAI,CAAC,KAAKA,QACR,IAAI;GACF,MAAM,MAAM,MAAM,OAAO;GACzB,KAAKA,SAAS;EAChB,QAAQ;GACN,MAAM,IAAI,kCAAkC,CAAC,cAAc,CAAC;EAC9D;CAEJ;CAEA,MAAM,QAAuB;EAC3B,KAAKC,KAAK,MAAM;CAClB;CAEA,MAAM,iBAAiB,MAAsB,aAAqC;EAChF,IAAI,KAAKC,aAAa,IAAI,KAAK,UAAU,GAAG;GAC1C,IAAI,CAAC,aACH,MAAM,IAAI,MAAM,eAAe,KAAK,WAAW,iBAAiB;GAElE;EACF;EACA,KAAKA,aAAa,IAAI,KAAK,4BAAY,IAAI,IAAI,CAAC;EAChD,KAAKC,MAAM,IAAI,KAAK,YAAY,KAAK,OAAO,UAAU;EACtD,IAAI,KAAKH,QAAQ;GACf,MAAM,KAAK,KAAKA,OAAO,OAAO,EAC5B,QAAQ;IACN,WAAW,UAAU,KAAK,OAAO,WAAW;IAC5C,cAAc;GAChB,EACF,CAAC;GACD,KAAKC,KAAK,IAAI,KAAK,YAAY,EAAE;EACnC;CACF;CAEA,MAAM,eAAe,YAAoB,UAAkC;EACzE,IAAI,CAAC,KAAKC,aAAa,IAAI,UAAU,GAAG;GACtC,IAAI,CAAC,UACH,MAAM,IAAI,MAAM,eAAe,WAAW,iBAAiB;GAE7D;EACF;EACA,KAAKA,aAAa,OAAO,UAAU;EACnC,KAAKC,MAAM,OAAO,UAAU;EAC5B,KAAKF,KAAK,OAAO,UAAU;CAC7B;CAEA,MAAM,cAAc,YAAsC;EACxD,OAAO,KAAKC,aAAa,IAAI,UAAU;CACzC;CAEA,MAAM,iBAAiB,MAAc,IAA2B;EAC9D,MAAM,IAAI,KAAKA,aAAa,IAAI,IAAI;EACpC,IAAI,CAAC,GACH,MAAM,IAAI,MAAM,eAAe,KAAK,iBAAiB;EAEvD,IAAI,KAAKA,aAAa,IAAI,EAAE,GAC1B,MAAM,IAAI,MAAM,eAAe,GAAG,iBAAiB;EAErD,KAAKA,aAAa,IAAI,IAAI,CAAC;EAC3B,KAAKA,aAAa,OAAO,IAAI;EAC7B,MAAM,IAAI,KAAKC,MAAM,IAAI,IAAI;EAC7B,IAAI,MAAM,KAAA,GAAW;GACnB,KAAKA,MAAM,IAAI,IAAI,CAAC;GACpB,KAAKA,MAAM,OAAO,IAAI;EACxB;EACA,MAAM,KAAK,KAAKF,KAAK,IAAI,IAAI;EAC7B,IAAI,IAAI;GACN,KAAKA,KAAK,IAAI,IAAI,EAAE;GACpB,KAAKA,KAAK,OAAO,IAAI;EACvB;CACF;CAEA,MAAM,YAA4C;EAChD,IAAI,IAAI,KAAKC,aAAa,IAAI,UAAU;EACxC,IAAI,CAAC,GAAG;GACN,oBAAI,IAAI,IAAI;GACZ,KAAKA,aAAa,IAAI,YAAY,CAAC;EACrC;EACA,OAAO;CACT;CAEA,OAAO,YAAyB;EAC9B,OAAO,KAAKD,KAAK,IAAI,UAAU;CACjC;CAEA,MAAM,cAAc,MAAiC;EACnD,IAAI,KAAK,QAAQ,WAAW,GAAG;EAC/B,gBAAgB,KAAK,OAAO;EAC5B,MAAM,WAAW,KAAKE,MAAM,IAAI,KAAK,UAAU;EAC/C,MAAM,IAAI,KAAKC,MAAM,KAAK,UAAU;EACpC,MAAM,KAAK,KAAKC,OAAO,KAAK,UAAU;EACtC,KAAK,MAAM,KAAK,KAAK,SAAS;GAC5B,IAAI,SAAS,EAAE;GACf,IAAI,CAAC,UAAU,EAAE,UAAU;IACzB,MAAM,CAAC,KAAK,MAAM,KAAK,OAAO,CAAC,EAAE,QAAQ,GAAG,UAAU;IACtD,SAAS;GACX;GACA,IAAI,UAAU,aAAa,KAAA,KAAa,OAAO,WAAW,UACxD,MAAM,IAAI,kCAAkC,CAAC,UAAU,OAAO,MAAM,CAAC;GAEvE,MAAM,MAAiB;IACrB,IAAI,EAAE;IACN;IACA,UAAU,EAAE;IACZ,UAAU,EAAE;GACd;GACA,EAAE,IAAI,EAAE,IAAI,GAAG;GACf,IAAI,IAAI;IACN,MAAM,MAAW;KACf,IAAI,EAAE;KACN,cAAc,KAAK,UAAU,IAAI,YAAY,CAAC,CAAC;KAC/C,WAAW;IACb;IACA,MAAM,KAAKL,OAAO,OAAO,IAAI,GAAG;GAClC;EACF;CACF;CAEA,MAAM,cAAc,MAAiC;EACnD,MAAM,IAAI,KAAKI,MAAM,KAAK,UAAU;EACpC,MAAM,KAAK,KAAKC,OAAO,KAAK,UAAU;EACtC,IAAI,KAAK,KAAK;GACZ,KAAK,MAAM,MAAM,KAAK,KAAK,EAAE,OAAO,EAAE;GACtC,IAAI,IACF,KAAK,MAAM,MAAM,KAAK,KAAK,MAAM,KAAKL,OAAO,OAAO,IAAI,EAAE;GAE5D;EACF;EACA,IAAI,KAAK,QAAQ;GACf,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,GAAG,CAAC,GAC3B,IAAI,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,GAAG;IAC7D,EAAE,OAAO,EAAE;IACX,IAAI,IAAI,MAAM,KAAKA,OAAO,OAAO,IAAI,EAAE;GACzC;GAEF;EACF;EACA,EAAE,MAAM;EACR,IAAI,IAAI;GACN,MAAM,SAAS,KAAKA,OAAO,OAAO,IAAI,EAAE,OAAO,IAAM,CAAC;GACtD,IAAI,OAAO,MACT,KAAK,MAAM,KAAK,OAAO,MAAM,MAAM,KAAKA,OAAO,OAAO,IAAI,EAAE,EAAE;EAElE;CACF;CAEA,MAAM,cAAc,MAA0C;EAC5D,MAAM,IAAI,KAAKI,MAAM,KAAK,UAAU;EACpC,MAAM,KAAK,KAAKC,OAAO,KAAK,UAAU;EACtC,IAAI,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC;EACzB,IAAI;EAEJ,IAAI,KAAK,QAAQ,YAAY,KAAK,MAAM;GACtC,IAAI,IAAI;IACN,MAAM,IAAI,MAAM,KAAKL,OAAO,OAAO,IAAI;KACrC,MAAM;KACN,QAAQ;MAAE,OAAO,KAAK,KAAK;MAAQ,UAAU;KAAY;KACzD,OAAO,KAAK,QAAQ,KAAK,UAAU;KACnC,YAAY;IACd,CAAC;IACD,MAAM,MAAM,IAAI,IAAI,EAAE,KAAK,KAAK,MAAW,EAAE,EAAE,CAAC;IAChD,MAAM,QAAQ,IAAI,IAAI,KAAK,QAAQ,QAAQ,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACvF,OAAO,EAAE,KAAK,KAAK,MAAW,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,OAAO,OAAO;GAC/D,OACE,OAAO,KAAK,aAAa,IAAI;GAE/B,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS;IAC1B;IACA,OAAO,KAAKM,iBAAiB,KAAK,MAA8B,KAAK,IAAI;GAC3E,EAAE;GACF,OAAO,MAAM,GAAG,OAAO,EAAE,SAAS,MAAM,EAAE,SAAS,EAAE;EACvD,OAAO,IAAI,KAAK,QAAQ,gBAAgB,KAAK,MAAM;GACjD,MAAM,CAAC,MAAM,MAAM,KAAK,OAAO,CAAC,KAAK,KAAK,UAAU,GAAG,OAAO;GAC9D,IAAI,IAAI;IACN,MAAM,IAAI,MAAM,KAAKN,OAAO,OAAO,IAAI;KACrC,MAAM;KACN,QAAQ;MAAE,OAAO;MAAI,UAAU;KAAY;KAC3C,OAAO,KAAK,QAAQ,KAAK,UAAU;KACnC,YAAY;IACd,CAAC;IACD,MAAM,MAAM,IAAI,IAAI,EAAE,KAAK,KAAK,MAAW,EAAE,EAAE,CAAC;IAChD,MAAM,QAAQ,IAAI,IAAI,KAAK,QAAQ,QAAQ,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACvF,OAAO,EAAE,KAAK,KAAK,MAAW,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,OAAO,OAAO;GAC/D;GACA,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS;IAAE;IAAK,OAAO,KAAKO,UAAU,IAAI,IAAI,QAAQ,IAAI;GAAE,EAAE;GACjF,OAAO,MAAM,GAAG,OAAO,EAAE,SAAS,MAAM,EAAE,SAAS,EAAE;EACvD,OAAO,IAAI,KAAK,QAAQ,QAAQ,KAAK,MAAM;GAEzC,MAAM,KADM,EAAE,IAAI,KAAK,KAAK,EACjB,GAAK;GAChB,IAAI,MAAM,IAAI;IACZ,MAAM,IAAI,MAAM,KAAKP,OAAO,OAAO,IAAI;KACrC,MAAM;KACN,QAAQ;MAAE,OAAO;MAAI,UAAU;KAAY;KAC3C,OAAO,KAAK,QAAQ,KAAK,UAAU;KACnC,YAAY;IACd,CAAC;IACD,MAAM,MAAM,IAAI,IAAI,EAAE,KAAK,KAAK,MAAW,EAAE,EAAE,CAAC;IAChD,MAAM,QAAQ,IAAI,IAAI,KAAK,QAAQ,QAAQ,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IACvF,OAAO,EAAE,KAAK,KAAK,MAAW,MAAM,IAAI,EAAE,EAAE,CAAC,EAAE,OAAO,OAAO;GAC/D;GACA,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS;IAC1B;IACA,OAAO,KAAK,KAAKO,UAAU,IAAI,IAAI,QAAQ,IAAI,IAAI,KAAA;GACrD,EAAE;GACF,OAAO,MAAM,GAAG,OAAO,EAAE,SAAS,MAAM,EAAE,SAAS,EAAE;EACvD,OAAO;GACL,IAAI,KAAK,QACP,OAAO,KAAK,QAAQ,QAAQ,eAAe,KAAK,QAAU,IAAI,YAAY,CAAC,CAAS,CAAC;GACvF,SAAS,KAAK,KAAK,SAAS,EAAE,IAAI,EAAE;EACtC;EAGA,OADgB,OAAO,MAAM,KAAK,UAAU,IAAI,KAAK,UAAU,KAAK,KAAK,IAClE,EAAQ,KAAK,EAAE,KAAK,YAAY,KAAKC,SAAS,KAAK,OAAO,KAAK,UAAU,CAAC;CACnF;CAEA,UAAU,GAAa,GAAyB,OAAuC;EACrF,IAAI,CAAC,GAAG,OAAO,KAAA;EACf,MAAM,SAAS,KAAK,QAAQ,UAAU;EACtC,IAAI;EACJ,IAAI,WAAW,UAAU;GACvB,MAAM,MAAM,KAAKC,KAAK,GAAG,CAAC;GAC1B,MAAM,KAAK,KAAK,KAAK,KAAKA,KAAK,GAAG,CAAC,CAAC;GACpC,MAAM,KAAK,KAAK,KAAK,KAAKA,KAAK,GAAG,CAAC,CAAC;GACpC,MAAM,MAAM,KAAK,OAAO,KAAK,MAAM;GACnC,OAAO,QAAQ,GAAG;EACpB;EACA,IAAI,WAAW,OAAO;GACpB,MAAM,KAAKA,KAAK,GAAG,CAAC;GACpB,OAAO,QAAQ,GAAG;EACpB;EACA,IAAI,IAAI;EACR,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,KAAK;GACrD,MAAM,IAAI,EAAE,KAAK,EAAE;GACnB,KAAK,IAAI;EACX;EACA,MAAM,KAAK,KAAK,CAAC;EACjB,OAAO,QAAQ,KAAK,IAAI,IAAI;CAC9B;CAEA,iBACE,MACA,KACA,MACoB;EACpB,OAAO,KAAKF,UAAU,KAAK,QAAQ,IAAI,QAAQ,IAAI;CACrD;CAEA,KAAK,GAAa,GAAqB;EACrC,IAAI,IAAI;EACR,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,KAAK,KAAK,EAAE,KAAK,EAAE;EACrE,OAAO;CACT;CAEA,SAAS,KAAgB,OAA2B,GAA4B;EAC9E,MAAM,MAAmB,CAAC;EAC1B,IAAI,EAAE,IAAI,IAAI,KAAK,IAAI;EACvB,IAAI,UAAU,KAAA,GAAW,IAAI,QAAQ;EACrC,IAAI,EAAE,UAAU,IAAI,QAAQ,IAAI,SAAS,IAAI;EAC7C,IAAI,EAAE,YAAY,IAAI,aAAa,KAAA,GAAW,IAAI,WAAW,IAAI;EACjE,IAAI,EAAE,YAAY,IAAI,aAAa,KAAA,GAAW,IAAI,WAAW,IAAI;EACjE,OAAO;CACT;AACF"}

package/batteries/vector/pgvector/index.d.ts

@@ -0,0 +1,43 @@
+/**
+ * @module @nhtio/adk/batteries/vector/pgvector
+ */
+import { BaseVectorStore, CallableVectorStore } from "../contract";
+import type { VectorFilter } from "../filters";
+import type { SearchPlan, UpsertPlan, DeletePlan, CollectionSpec } from "../plan";
+import type { VectorMatch, VectorStoreCapabilities, BaseVectorStoreOptions } from "../types";
+interface PgVectorOptions extends BaseVectorStoreOptions {
+    connection?: string | {
+        connectionString?: string;
+        host?: string;
+        port?: number;
+        user?: string;
+        password?: string;
+        database?: string;
+    };
+    pool?: any;
+}
+type PoolOrClient = any;
+export declare const translatePgFilter: (filter: VectorFilter) => {
+    sql: string;
+    params: unknown[];
+};
+export declare class PgVectorStore extends BaseVectorStore {
+    #private;
+    readonly capabilities: VectorStoreCapabilities;
+    constructor(options: PgVectorOptions);
+    static isAvailable(): boolean;
+    isAvailable(): boolean;
+    connect(): Promise<void>;
+    close(): Promise<void>;
+    get clientOrPool(): PoolOrClient;
+    ensureConnected(): Promise<void>;
+    createCollection(spec: CollectionSpec, ifNotExists: boolean): Promise<void>;
+    dropCollection(collection: string, ifExists: boolean): Promise<void>;
+    hasCollection(collection: string): Promise<boolean>;
+    renameCollection(from: string, to: string): Promise<void>;
+    executeUpsert(plan: UpsertPlan): Promise<void>;
+    executeSearch(plan: SearchPlan): Promise<VectorMatch[]>;
+    executeDelete(plan: DeletePlan): Promise<void>;
+    transaction(fn: (tx: CallableVectorStore) => Promise<void>): Promise<void>;
+}
+export {};