@ax-llm/ax 9.0.21 → 9.0.22

package/db/cloudflare.js

@@ -0,0 +1,98 @@
+import { apiCall } from '../util/apicall.js';
+import { AxDBBase } from './base.js';
+const baseURL = 'https://api.cloudflare.com/client/v4/accounts/';
+/**
+ * Cloudflare: DB Service
+ * @export
+ */
+export class AxDBCloudflare extends AxDBBase {
+    apiKey;
+    accountId;
+    constructor({ apiKey, accountId, fetch, tracer }) {
+        if (!apiKey || !accountId) {
+            throw new Error('Cloudflare credentials not set');
+        }
+        super({ name: 'Cloudflare', fetch, tracer });
+        this.apiKey = apiKey;
+        this.accountId = accountId;
+    }
+    _upsert = async (req, update, options) => {
+        const res = (await apiCall({
+            url: new URL(`${this.accountId}/vectorize/indexes/${req.table}/upsert`, baseURL),
+            headers: {
+                'X-Auth-Key': this.apiKey
+            },
+            fetch: this.fetch,
+            span: options?.span
+        }, {
+            id: req.id,
+            values: req.values,
+            namespace: req.namespace,
+            metadata: req.metadata
+        }));
+        if (res.errors) {
+            throw new Error(`Cloudflare upsert failed: ${res.errors.map(({ message }) => message).join(', ')}`);
+        }
+        return {
+            ids: res.result.ids
+        };
+    };
+    batchUpsert = async (batchReq, update, options) => {
+        if (update) {
+            throw new Error('Weaviate does not support batch update');
+        }
+        if (batchReq.length < 1) {
+            throw new Error('Batch request is empty');
+        }
+        if (!batchReq[0] || !batchReq[0].table) {
+            throw new Error('Table name is empty');
+        }
+        const table = batchReq[0].table;
+        const res = (await apiCall({
+            url: new URL(`${this.accountId}/vectorize/indexes/${table}/upsert`, baseURL),
+            headers: {
+                'X-Auth-Key': this.apiKey
+            },
+            fetch: this.fetch,
+            span: options?.span
+        }, batchReq.map((req) => ({
+            id: req.id,
+            values: req.values,
+            namespace: req.namespace,
+            metadata: req.metadata
+        }))));
+        if (res.errors) {
+            throw new Error(`Cloudflare batch upsert failed: ${res.errors
+                .map(({ message }) => message)
+                .join(', ')}`);
+        }
+        return {
+            ids: res.result.ids
+        };
+    };
+    query = async (req, options) => {
+        const res = (await apiCall({
+            url: new URL(`${this.accountId}/vectorize/indexes/${req.table}/query`, baseURL),
+            headers: {
+                'X-Auth-Key': this.apiKey
+            },
+            fetch: this.fetch,
+            span: options?.span
+        }, {
+            vector: req.values,
+            topK: req.limit || 10,
+            returnValues: true
+        }));
+        if (res.errors) {
+            throw new Error(`Cloudflare query failed: ${res.errors.map(({ message }) => message).join(', ')}`);
+        }
+        const matches = res.result.matches.map(({ id, score, values, metadata }) => ({
+            id,
+            score,
+            values,
+            metadata
+        }));
+        return { matches };
+    };
+}
+//# sourceMappingURL=cloudflare.js.map

package/db/cloudflare.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare.js","sourceRoot":"","sources":["../../db/cloudflare.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAE7C,OAAO,EAAE,QAAQ,EAA6C,MAAM,WAAW,CAAC;AAQhF,MAAM,OAAO,GAAG,gDAAgD,CAAC;AA8BjE;;;GAGG;AACH,MAAM,OAAO,cAAe,SAAQ,QAAQ;IAClC,MAAM,CAAS;IACf,SAAS,CAAS;IAE1B,YAAY,EACV,MAAM,EACN,SAAS,EACT,KAAK,EACL,MAAM,EACqC;QAC3C,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QACD,KAAK,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAEQ,OAAO,GAAG,KAAK,EACtB,GAAgC,EAChC,MAAgB,EAChB,OAA2C,EACd,EAAE;QAC/B,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,GAAG,CACV,GAAG,IAAI,CAAC,SAAS,sBAAsB,GAAG,CAAC,KAAK,SAAS,EACzD,OAAO,CACR;YACD,OAAO,EAAE;gBACP,YAAY,EAAE,IAAI,CAAC,MAAM;aAC1B;YACD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD;YACE,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CACF,CAA+B,CAAC;QAEjC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,6BAA6B,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACnF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG;SACpB,CAAC;IACJ,CAAC,CAAC;IAEO,WAAW,GAAG,KAAK,EAC1B,QAAuC,EACvC,MAAgB,EAChB,OAA2C,EACd,EAAE;QAC/B,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAEhC,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,GAAG,CACV,GAAG,IAAI,CAAC,SAAS,sBAAsB,KAAK,SAAS,EACrD,OAAO,CACR;YACD,OAAO,EAAE;gBACP,YAAY,EAAE,IAAI,CAAC,MAAM;aAC1B;YACD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACrB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC,CAAC,CACJ,CAA+B,CAAC;QAEjC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,mCAAmC,GAAG,CAAC,MAAM;iBAC1C,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC;iBAC7B,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG;SACpB,CAAC;IACJ,CAAC,CAAC;IAEO,KAAK,GAAG,KAAK,EACpB,GAA+B,EAC/B,OAA2C,EACf,EAAE;QAC9B,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,GAAG,CACV,GAAG,IAAI,CAAC,SAAS,sBAAsB,GAAG,CAAC,KAAK,QAAQ,EACxD,OAAO,CACR;YACD,OAAO,EAAE;gBACP,YAAY,EAAE,IAAI,CAAC,MAAM;aAC1B;YACD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD;YACE,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,IAAI,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE;YACrB,YAAY,EAAE,IAAI;SACnB,CACF,CAA8B,CAAC;QAEhC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,4BAA4B,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAClF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CACpC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACpC,EAAE;YACF,KAAK;YACL,MAAM;YACN,QAAQ;SACT,CAAC,CACH,CAAC;QACF,OAAO,EAAE,OAAO,EAAuB,CAAC;IAC1C,CAAC,CAAC;CACH"}

package/db/index.d.ts

@@ -0,0 +1,6 @@
+export * from './types.js';
+export * from './weaviate.js';
+export * from './pinecone.js';
+export * from './cloudflare.js';
+export * from './memory.js';
+export * from './wrap.js';

package/db/index.js

@@ -0,0 +1,7 @@
+export * from './types.js';
+export * from './weaviate.js';
+export * from './pinecone.js';
+export * from './cloudflare.js';
+export * from './memory.js';
+export * from './wrap.js';
+//# sourceMappingURL=index.js.map

package/db/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../db/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAC;AAC3B,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,cAAc,iBAAiB,CAAC;AAChC,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC"}

package/db/memory.d.ts

@@ -0,0 +1,21 @@
+import { AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions } from './base.js';
+import type { AxDBQueryRequest, AxDBQueryResponse, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
+export type AxDBMemoryOpOptions = AxDBBaseOpOptions;
+export interface AxDBMemoryArgs extends AxDBBaseArgs {
+    name: 'memory';
+}
+export type AxDBState = Record<string, Record<string, AxDBUpsertRequest>>;
+/**
+ * MemoryDB: DB Service
+ * @export
+ */
+export declare class AxDBMemory extends AxDBBase {
+    private state;
+    constructor({ tracer }?: Readonly<Omit<AxDBMemoryArgs, 'name'>>);
+    _upsert: (req: Readonly<AxDBUpsertRequest>, _update?: boolean, _options?: Readonly<AxDBMemoryOpOptions>) => Promise<AxDBUpsertResponse>;
+    _batchUpsert: (batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean, _options?: Readonly<AxDBMemoryOpOptions>) => Promise<AxDBUpsertResponse>;
+    _query: (req: Readonly<AxDBQueryRequest>, _options?: Readonly<AxDBMemoryOpOptions>) => Promise<AxDBQueryResponse>;
+    getDB: () => AxDBState;
+    setDB: (state: AxDBState) => void;
+    clearDB: () => void;
+}

package/db/memory.js

@@ -0,0 +1,99 @@
+import { AxDBBase } from './base.js';
+/**
+ * MemoryDB: DB Service
+ * @export
+ */
+export class AxDBMemory extends AxDBBase {
+    state;
+    constructor({ tracer } = {}) {
+        super({ name: 'Memory', tracer });
+        this.state = {};
+    }
+    _upsert = async (req, 
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    _update, 
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    _options) => {
+        if (!this.state[req.table]) {
+            this.state[req.table] = {
+                [req.id]: req
+            };
+        }
+        else {
+            const obj = this.state[req.table];
+            if (!obj) {
+                throw new Error('Table not found: ' + req.table);
+            }
+            obj[req.id] = req;
+        }
+        return { ids: [req.id] };
+    };
+    _batchUpsert = async (batchReq, update, 
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    _options) => {
+        const ids = [];
+        for (const req of batchReq) {
+            const res = await this.upsert(req, update);
+            ids.push(...res.ids);
+        }
+        return { ids };
+    };
+    _query = async (req, 
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    _options) => {
+        const table = this.state[req.table];
+        if (!table) {
+            return { matches: [] };
+        }
+        const matches = [];
+        Object.entries(table).forEach(([id, data]) => {
+            if (req.values && data.values) {
+                const score = distance(req.values, data.values);
+                matches.push({ id: id, score: score, metadata: data.metadata });
+            }
+        });
+        matches.sort((a, b) => a.score - b.score);
+        if (req.limit) {
+            matches.length = req.limit;
+        }
+        return { matches };
+    };
+    getDB = () => {
+        return structuredClone(this.state);
+    };
+    setDB = (state) => {
+        this.state = structuredClone(state);
+    };
+    clearDB = () => {
+        this.state = {};
+    };
+}
+const distance = (a, b) => {
+    if (a.length !== b.length) {
+        throw new Error('Vectors must be of the same length.');
+    }
+    let dotProduct = 0;
+    let normA = 0;
+    let normB = 0;
+    let zeroVectorA = true;
+    let zeroVectorB = true;
+    const vectorA = new Float64Array(a);
+    const vectorB = new Float64Array(b);
+    for (let i = 0; i < vectorA.length; i++) {
+        dotProduct += vectorA[i] * vectorB[i];
+        normA += vectorA[i] * vectorA[i];
+        normB += vectorB[i] * vectorB[i];
+        if (vectorA[i] !== 0)
+            zeroVectorA = false;
+        if (vectorB[i] !== 0)
+            zeroVectorB = false;
+    }
+    if (zeroVectorA || zeroVectorB) {
+        return 1; // Return maximum distance if one vector is zero
+    }
+    const sqrtNormA = Math.sqrt(normA);
+    const sqrtNormB = Math.sqrt(normB);
+    const similarity = dotProduct / (sqrtNormA * sqrtNormB);
+    return 1 - similarity; // Returning distance as 1 - cosine similarity.
+};
+//# sourceMappingURL=memory.js.map

package/db/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../db/memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAA6C,MAAM,WAAW,CAAC;AAgBhF;;;GAGG;AACH,MAAM,OAAO,UAAW,SAAQ,QAAQ;IAC9B,KAAK,CAAY;IAEzB,YAAY,EAAE,MAAM,KAA6C,EAAE;QACjE,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;IAEQ,OAAO,GAAG,KAAK,EACtB,GAAgC;IAChC,6DAA6D;IAC7D,OAAiB;IACjB,6DAA6D;IAC7D,QAAwC,EACX,EAAE;QAC/B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG;gBACtB,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,GAAG;aACd,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC;YACnD,CAAC;YACD,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC;QACpB,CAAC;QAED,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IAC3B,CAAC,CAAC;IAEO,YAAY,GAAG,KAAK,EAC3B,QAAuC,EACvC,MAAgB;IAChB,6DAA6D;IAC7D,QAAwC,EACX,EAAE;QAC/B,MAAM,GAAG,GAAa,EAAE,CAAC;QACzB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAC3C,GAAG,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;QAED,OAAO,EAAE,GAAG,EAAE,CAAC;IACjB,CAAC,CAAC;IAEO,MAAM,GAAG,KAAK,EACrB,GAA+B;IAC/B,6DAA6D;IAC7D,QAAwC,EACZ,EAAE;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,GAAiC,EAAE,CAAC;QAEjD,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE;YAC3C,IAAI,GAAG,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC9B,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAChD,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAClE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC;QAC7B,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC,CAAC;IAEK,KAAK,GAAG,GAAG,EAAE;QAClB,OAAO,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC,CAAC;IAEK,KAAK,GAAG,CAAC,KAAgB,EAAE,EAAE;QAClC,IAAI,CAAC,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC;IAEK,OAAO,GAAG,GAAG,EAAE;QACpB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC,CAAC;CACH;AAED,MAAM,QAAQ,GAAG,CAAC,CAAoB,EAAE,CAAoB,EAAU,EAAE;IACtE,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IAED,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,WAAW,GAAG,IAAI,CAAC;IACvB,IAAI,WAAW,GAAG,IAAI,CAAC;IAEvB,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;IAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,UAAU,IAAI,OAAO,CAAC,CAAC,CAAE,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;QACxC,KAAK,IAAI,OAAO,CAAC,CAAC,CAAE,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;QACnC,KAAK,IAAI,OAAO,CAAC,CAAC,CAAE,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;QACnC,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;YAAE,WAAW,GAAG,KAAK,CAAC;QAC1C,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;YAAE,WAAW,GAAG,KAAK,CAAC;IAC5C,CAAC;IAED,IAAI,WAAW,IAAI,WAAW,EAAE,CAAC;QAC/B,OAAO,CAAC,CAAC,CAAC,gDAAgD;IAC5D,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,UAAU,GAAG,UAAU,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,UAAU,CAAC,CAAC,+CAA+C;AACxE,CAAC,CAAC"}

package/db/pinecone.d.ts

@@ -0,0 +1,21 @@
+import { AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions } from './base.js';
+import type { AxDBQueryRequest, AxDBQueryResponse, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
+export type AxDBPineconeOpOptions = AxDBBaseOpOptions;
+export interface AxDBPineconeArgs extends AxDBBaseArgs {
+    name: 'pinecone';
+    apiKey: string;
+    host: string;
+    fetch?: typeof fetch;
+}
+/**
+ * Pinecone: DB Service
+ * @export
+ */
+export declare class AxDBPinecone extends AxDBBase {
+    private apiKey;
+    private apiURL;
+    constructor({ apiKey, host, fetch, tracer }: Readonly<Omit<AxDBPineconeArgs, 'name'>>);
+    _upsert: (req: Readonly<AxDBUpsertRequest>, update?: boolean, options?: Readonly<AxDBPineconeOpOptions>) => Promise<AxDBUpsertResponse>;
+    _batchUpsert: (batchReq: Readonly<AxDBUpsertRequest[]>, _update?: boolean, options?: Readonly<AxDBPineconeOpOptions>) => Promise<AxDBUpsertResponse>;
+    query: (req: Readonly<AxDBQueryRequest>, options?: Readonly<AxDBPineconeOpOptions>) => Promise<AxDBQueryResponse>;
+}

package/db/pinecone.js

@@ -0,0 +1,71 @@
+import { apiCall } from '../util/apicall.js';
+import { AxDBBase } from './base.js';
+const createPineconeQueryRequest = (req) => {
+    const pineconeQueryRequest = {
+        namespace: req.namespace,
+        topK: req.limit || 10,
+        filter: {},
+        includeValues: true,
+        includeMetadata: true,
+        vector: req.values ?? [],
+        id: req.id
+    };
+    return pineconeQueryRequest;
+};
+/**
+ * Pinecone: DB Service
+ * @export
+ */
+export class AxDBPinecone extends AxDBBase {
+    apiKey;
+    apiURL;
+    constructor({ apiKey, host, fetch, tracer }) {
+        if (!apiKey || apiKey === '') {
+            throw new Error('Pinecone API key not set');
+        }
+        super({ name: 'Pinecone', fetch, tracer });
+        this.apiKey = apiKey;
+        this.apiURL = host;
+    }
+    _upsert = async (req, update, options) => {
+        await this._batchUpsert([req], update, options);
+        return { ids: [req.id] };
+    };
+    _batchUpsert = async (batchReq, _update, options) => {
+        if (batchReq.length === 0) {
+            throw new Error('Batch request is empty');
+        }
+        await apiCall({
+            url: this.apiURL,
+            headers: { Authorization: `Bearer ${this.apiKey}` },
+            name: '/vectors/upsert',
+            fetch: this.fetch,
+            span: options?.span
+        }, batchReq.map(({ id, values = [], metadata }) => ({
+            id,
+            values,
+            metadata
+        })));
+        return { ids: batchReq.map(({ id }) => id) };
+    };
+    query = async (req, options) => {
+        if (req.text) {
+            throw new Error('Pinecone does not support text');
+        }
+        const res = (await apiCall({
+            url: this.apiURL,
+            headers: { Authorization: `Bearer ${this.apiKey}` },
+            name: '/query',
+            fetch: this.fetch,
+            span: options?.span
+        }, createPineconeQueryRequest(req)));
+        const matches = res.matches.map(({ id, score, values, metadata }) => ({
+            id,
+            score,
+            metadata,
+            values
+        }));
+        return { matches };
+    };
+}
+//# sourceMappingURL=pinecone.js.map

package/db/pinecone.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pinecone.js","sourceRoot":"","sources":["../../db/pinecone.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAE7C,OAAO,EAAE,QAAQ,EAA6C,MAAM,WAAW,CAAC;AA6BhF,MAAM,0BAA0B,GAAG,CACjC,GAA+B,EACP,EAAE;IAC1B,MAAM,oBAAoB,GAA2B;QACnD,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,IAAI,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE;QACrB,MAAM,EAAE,EAAE;QACV,aAAa,EAAE,IAAI;QACnB,eAAe,EAAE,IAAI;QACrB,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,EAAE;QACxB,EAAE,EAAE,GAAG,CAAC,EAAE;KACX,CAAC;IAEF,OAAO,oBAAoB,CAAC;AAC9B,CAAC,CAAC;AASF;;;GAGG;AACH,MAAM,OAAO,YAAa,SAAQ,QAAQ;IAChC,MAAM,CAAS;IACf,MAAM,CAAS;IAEvB,YAAY,EACV,MAAM,EACN,IAAI,EACJ,KAAK,EACL,MAAM,EACmC;QACzC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,KAAK,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAEQ,OAAO,GAAG,KAAK,EACtB,GAAgC,EAChC,MAAgB,EAChB,OAAyC,EACZ,EAAE;QAC/B,MAAM,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAChD,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IAC3B,CAAC,CAAC;IAEO,YAAY,GAAG,KAAK,EAC3B,QAAuC,EACvC,OAAiB,EACjB,OAAyC,EACZ,EAAE;QAC/B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,OAAO,CACX;YACE,GAAG,EAAE,IAAI,CAAC,MAAM;YAChB,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;YACnD,IAAI,EAAE,iBAAiB;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,GAAG,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAC/C,EAAE;YACF,MAAM;YACN,QAAQ;SACT,CAAC,CAAC,CACJ,CAAC;QAEF,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;IAC/C,CAAC,CAAC;IAEO,KAAK,GAAG,KAAK,EACpB,GAA+B,EAC/B,OAAyC,EACb,EAAE;QAC9B,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,CAAC,MAAM;YAChB,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;YACnD,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD,0BAA0B,CAAC,GAAG,CAAC,CAChC,CAA4B,CAAC;QAE9B,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACpE,EAAE;YACF,KAAK;YACL,QAAQ;YACR,MAAM;SACP,CAAC,CAAC,CAAC;QAEJ,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC,CAAC;CACH"}

package/db/types.d.ts

@@ -0,0 +1,35 @@
+export type AxDBUpsertRequest = {
+    id: string;
+    text?: string;
+    values?: readonly number[];
+    metadata?: Record<string, string>;
+    table: string;
+    namespace?: string;
+};
+export type AxDBUpsertResponse = {
+    ids: string[];
+};
+export type AxDBQueryRequest = {
+    id?: string;
+    text?: string;
+    values?: readonly number[];
+    table: string;
+    columns?: string[];
+    limit?: number;
+    namespace?: string;
+};
+export type AxDBQueryResponse = {
+    matches: {
+        id: string;
+        score: number;
+        metadata?: Record<string, string>;
+        table?: string;
+    }[];
+};
+export interface AxDBService extends AxDBQueryService {
+    upsert(req: Readonly<AxDBUpsertRequest>, update?: boolean): Promise<AxDBUpsertResponse>;
+    batchUpsert(batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean): Promise<AxDBUpsertResponse>;
+}
+export interface AxDBQueryService {
+    query(req: Readonly<AxDBQueryRequest>): Promise<AxDBQueryResponse>;
+}

package/db/types.js

@@ -0,0 +1,3 @@
+// For upsert
+export {};
+//# sourceMappingURL=types.js.map

package/db/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../db/types.ts"],"names":[],"mappings":"AAAA,aAAa"}

package/db/weaviate.d.ts

@@ -0,0 +1,21 @@
+import { AxDBBase, type AxDBBaseArgs, type AxDBBaseOpOptions } from './base.js';
+import type { AxDBQueryRequest, AxDBQueryResponse, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
+export type AxDBWeaviateOpOptions = AxDBBaseOpOptions;
+export interface AxDBWeaviateArgs extends AxDBBaseArgs {
+    name: 'weaviate';
+    apiKey: string;
+    host: string;
+    fetch?: typeof fetch;
+}
+/**
+ * Weaviate: DB Service
+ * @export
+ */
+export declare class AxDBWeaviate extends AxDBBase {
+    private apiKey;
+    private apiURL;
+    constructor({ apiKey, host, fetch, tracer }: Readonly<Omit<AxDBWeaviateArgs, 'name'>>);
+    _upsert: (req: Readonly<AxDBUpsertRequest>, update?: boolean, options?: Readonly<AxDBWeaviateOpOptions>) => Promise<AxDBUpsertResponse>;
+    _batchUpsert: (batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean, options?: Readonly<AxDBWeaviateOpOptions>) => Promise<AxDBUpsertResponse>;
+    _query: (req: Readonly<AxDBQueryRequest>, options?: Readonly<AxDBWeaviateOpOptions>) => Promise<AxDBQueryResponse>;
+}

package/db/weaviate.js

@@ -0,0 +1,127 @@
+import { apiCall } from '../util/apicall.js';
+import { AxDBBase } from './base.js';
+/**
+ * Weaviate: DB Service
+ * @export
+ */
+export class AxDBWeaviate extends AxDBBase {
+    apiKey;
+    apiURL;
+    constructor({ apiKey, host, fetch, tracer }) {
+        if (!apiKey || apiKey === '') {
+            throw new Error('Weaviate API key not set');
+        }
+        super({ name: 'Weaviate', fetch, tracer });
+        this.apiKey = apiKey;
+        this.apiURL = host;
+    }
+    _upsert = async (req, update, options) => {
+        const res = (await apiCall({
+            url: this.apiURL,
+            headers: { Authorization: `Bearer ${this.apiKey}` },
+            name: `/v1/objects/${req.table}/${req.id}`,
+            put: update ? true : false,
+            fetch: this.fetch,
+            span: options?.span
+        }, {
+            id: req.id,
+            class: req.table,
+            tenant: req.namespace,
+            vector: req.values,
+            properties: req.metadata ?? {}
+        }));
+        if (res?.result?.errors) {
+            throw new Error(`Weaviate upsert failed: ${res.result.errors.error
+                .map(({ message }) => message)
+                .join(', ')}`);
+        }
+        return {
+            ids: [res.id]
+        };
+    };
+    _batchUpsert = async (batchReq, update, options) => {
+        if (update) {
+            throw new Error('Weaviate does not support batch update');
+        }
+        if (batchReq.length === 0) {
+            throw new Error('Batch request is empty');
+        }
+        const objects = batchReq.map((req) => ({
+            id: req.id,
+            class: req.table,
+            tenant: req.namespace,
+            vector: req.values,
+            properties: req.metadata ?? {}
+        }));
+        const res = (await apiCall({
+            url: this.apiURL,
+            headers: { Authorization: `Bearer ${this.apiKey}` },
+            name: '/v1/batch/objects',
+            fetch: this.fetch,
+            span: options?.span
+        }, { objects }));
+        if (res?.some(({ result }) => result?.errors)) {
+            throw new Error(`Weaviate batch upsert failed: ${res
+                .map(({ result }) => result?.errors?.error.map(({ message }) => message).join(', '))
+                .join(', ')}`);
+        }
+        return {
+            ids: res.map(({ id }) => id)
+        };
+    };
+    _query = async (req, options) => {
+        let filter = '';
+        if (req.columns && req.columns.length === 0) {
+            throw new Error('Weaviate requires at least one column');
+        }
+        if (req.values) {
+            filter = `nearVector: {
+            vector: [${req.values.join(',')}],
+        }`;
+        }
+        else if (req.text) {
+            filter = `nearText: {
+            concepts: ['${req.text}'],
+        }`;
+        }
+        else {
+            throw new Error('Weaviate requires either text or values');
+        }
+        const res = (await apiCall({
+            url: this.apiURL,
+            headers: { Authorization: `Bearer ${this.apiKey}` },
+            name: '/v1/graphql',
+            fetch: this.fetch,
+            span: options?.span
+        }, {
+            query: `{
+          Get {
+            ${req.table} (
+              limit: ${req.limit || 10},
+              ${filter}
+            ) {
+                ${req.columns?.join('\n')}
+            }
+          }
+        }`
+        }));
+        if (res.errors) {
+            throw new Error(`Weaviate query failed: ${res.errors
+                .map(({ message }) => message)
+                .join(', ')}`);
+        }
+        const resMatches = res.data.Get[req.table];
+        if (!resMatches) {
+            return { matches: [] };
+        }
+        const matches = resMatches.map((match) => {
+            return {
+                id: match.id,
+                score: 1,
+                metadata: match
+            };
+        });
+        return { matches };
+    };
+}
+//# sourceMappingURL=weaviate.js.map

package/db/weaviate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"weaviate.js","sourceRoot":"","sources":["../../db/weaviate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAE7C,OAAO,EAAE,QAAQ,EAA6C,MAAM,WAAW,CAAC;AAiChF;;;GAGG;AACH,MAAM,OAAO,YAAa,SAAQ,QAAQ;IAChC,MAAM,CAAS;IACf,MAAM,CAAS;IAEvB,YAAY,EACV,MAAM,EACN,IAAI,EACJ,KAAK,EACL,MAAM,EACmC;QACzC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,KAAK,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAEQ,OAAO,GAAG,KAAK,EACtB,GAAgC,EAChC,MAAgB,EAChB,OAAyC,EACZ,EAAE;QAC/B,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,CAAC,MAAM;YAChB,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;YACnD,IAAI,EAAE,eAAe,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,EAAE,EAAE;YAC1C,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK;YAC1B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD;YACE,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,GAAG,CAAC,SAAS;YACrB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,UAAU,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;SAC/B,CACF,CAA6B,CAAC;QAE/B,IAAI,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CACb,2BAA2B,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK;iBAC/C,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC;iBAC7B,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC;SACd,CAAC;IACJ,CAAC,CAAC;IAEO,YAAY,GAAG,KAAK,EAC3B,QAAuC,EACvC,MAAgB,EAChB,OAAyC,EACZ,EAAE;QAC/B,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACrC,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,MAAM,EAAE,GAAG,CAAC,SAAS;YACrB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,UAAU,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;SAC/B,CAAC,CAAC,CAAC;QAEJ,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,CAAC,MAAM;YAChB,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;YACnD,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD,EAAE,OAAO,EAAE,CACZ,CAA+B,CAAC;QAEjC,IAAI,GAAG,EAAE,IAAI,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CACb,iCAAiC,GAAG;iBACjC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAClB,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAC/D;iBACA,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;SAC7B,CAAC;IACJ,CAAC,CAAC;IAEO,MAAM,GAAG,KAAK,EACrB,GAA+B,EAC/B,OAAyC,EACb,EAAE;QAC9B,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,GAAG;uBACQ,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;UACjC,CAAC;QACP,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;YACpB,MAAM,GAAG;0BACW,GAAG,CAAC,IAAI;UACxB,CAAC;QACP,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,GAAG,GAAG,CAAC,MAAM,OAAO,CACxB;YACE,GAAG,EAAE,IAAI,CAAC,MAAM;YAChB,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;YACnD,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE,OAAO,EAAE,IAAI;SACpB,EACD;YACE,KAAK,EAAE;;cAED,GAAG,CAAC,KAAK;uBACA,GAAG,CAAC,KAAK,IAAI,EAAE;gBACtB,MAAM;;kBAEJ,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC;;;UAG/B;SACH,CACF,CAA4B,CAAC;QAE9B,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,0BAA0B,GAAG,CAAC,MAAM;iBACjC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC;iBAC7B,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAE3C,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACvC,OAAO;gBACL,EAAE,EAAE,KAAK,CAAC,EAAY;gBACtB,KAAK,EAAE,CAAC;gBACR,QAAQ,EAAE,KAAK;aAChB,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAuB,CAAC;IAC1C,CAAC,CAAC;CACH"}

package/db/wrap.d.ts

@@ -0,0 +1,13 @@
+import { type AxDBCloudflareArgs } from './cloudflare.js';
+import { type AxDBMemoryArgs } from './memory.js';
+import { type AxDBPineconeArgs } from './pinecone.js';
+import type { AxDBQueryRequest, AxDBQueryResponse, AxDBService, AxDBUpsertRequest, AxDBUpsertResponse } from './types.js';
+import { type AxDBWeaviateArgs } from './weaviate.js';
+export type AxDBArgs = AxDBCloudflareArgs | AxDBPineconeArgs | AxDBWeaviateArgs | AxDBMemoryArgs;
+export declare class AxDB implements AxDBService {
+    private db;
+    constructor(args: Readonly<AxDBArgs>);
+    upsert(req: Readonly<AxDBUpsertRequest>, update?: boolean): Promise<AxDBUpsertResponse>;
+    batchUpsert(batchReq: Readonly<AxDBUpsertRequest[]>, update?: boolean): Promise<AxDBUpsertResponse>;
+    query(req: Readonly<AxDBQueryRequest>): Promise<AxDBQueryResponse>;
+}

package/db/wrap.js

@@ -0,0 +1,35 @@
+import { AxDBCloudflare } from './cloudflare.js';
+import { AxDBMemory } from './memory.js';
+import { AxDBPinecone } from './pinecone.js';
+import { AxDBWeaviate } from './weaviate.js';
+export class AxDB {
+    db;
+    constructor(args) {
+        switch (args.name) {
+            case 'weaviate':
+                this.db = new AxDBWeaviate(args);
+                break;
+            case 'pinecone':
+                this.db = new AxDBPinecone(args);
+                break;
+            case 'cloudflare':
+                this.db = new AxDBCloudflare(args);
+                break;
+            case 'memory':
+                this.db = new AxDBMemory(args);
+                break;
+            default:
+                throw new Error(`Unknown DB`);
+        }
+    }
+    async upsert(req, update) {
+        return await this.db.upsert(req, update);
+    }
+    async batchUpsert(batchReq, update) {
+        return await this.db.batchUpsert(batchReq, update);
+    }
+    async query(req) {
+        return await this.db.query(req);
+    }
+}
+//# sourceMappingURL=wrap.js.map

package/db/wrap.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrap.js","sourceRoot":"","sources":["../../db/wrap.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAA2B,MAAM,iBAAiB,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAuB,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAyB,MAAM,eAAe,CAAC;AAQpE,OAAO,EAAE,YAAY,EAAyB,MAAM,eAAe,CAAC;AAQpE,MAAM,OAAO,IAAI;IACP,EAAE,CAAc;IACxB,YAAY,IAAwB;QAClC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM;YACR,KAAK,YAAY;gBACf,IAAI,CAAC,EAAE,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;gBAC/B,MAAM;YACR;gBACE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IACD,KAAK,CAAC,MAAM,CACV,GAAgC,EAChC,MAAgB;QAEhB,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,WAAW,CACf,QAAuC,EACvC,MAAgB;QAEhB,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAA+B;QACzC,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC;CACF"}

package/docs/index.d.ts

@@ -0,0 +1,4 @@
+export * from './tika.js';
+export * from './manager.js';
+export * from './reranker.js';
+export * from './rewriter.js';

package/docs/index.js

@@ -0,0 +1,5 @@
+export * from './tika.js';
+export * from './manager.js';
+export * from './reranker.js';
+export * from './rewriter.js';
+//# sourceMappingURL=index.js.map

package/docs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../docs/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,cAAc,CAAC;AAC7B,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC"}