@ax-llm/ax 10.0.25 → 10.0.27

package/db/memory.js

@@ -1,99 +0,0 @@
-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

@@ -1 +0,0 @@
-{"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

@@ -1,21 +0,0 @@
-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

@@ -1,71 +0,0 @@
-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

@@ -1 +0,0 @@
-{"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

@@ -1,35 +0,0 @@
-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

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

package/db/types.js.map

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

package/db/weaviate.d.ts

@@ -1,21 +0,0 @@
-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

@@ -1,127 +0,0 @@
-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

@@ -1 +0,0 @@
-{"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,IAAI,CAAW;gBACzB,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

@@ -1,13 +0,0 @@
-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

@@ -1,35 +0,0 @@
-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

@@ -1 +0,0 @@
-{"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

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

package/docs/index.js

@@ -1,5 +0,0 @@
-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

@@ -1 +0,0 @@
-{"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"}

package/docs/manager.d.ts

@@ -1,45 +0,0 @@
-import { type AxAIService, type AxDBService, type AxProgram } from '../index.js';
-export type AxRewriteIn = {
-    query: string;
-};
-export type AxRewriteOut = {
-    rewrittenQuery: string;
-};
-export type AxRerankerIn = {
-    query: string;
-    items: string[];
-};
-export type AxRerankerOut = {
-    rankedItems: string[];
-};
-export interface AxDBLoaderOptions {
-    chunker?: (text: string) => string[];
-    rewriter?: AxProgram<AxRewriteIn, AxRewriteOut>;
-    reranker?: AxProgram<AxRerankerIn, AxRerankerOut>;
-}
-export interface AxDBManagerArgs {
-    ai: AxAIService;
-    db: AxDBService;
-    config?: AxDBLoaderOptions;
-}
-export interface AxDBMatch {
-    score: number;
-    text: string;
-}
-export declare class AxDBManager {
-    private ai;
-    private db;
-    private chunker;
-    private rewriter?;
-    private reranker?;
-    constructor({ ai, db, config }: Readonly<AxDBManagerArgs>);
-    private defaultChunker;
-    insert: (text: Readonly<string | string[]>, options?: Readonly<{
-        batchSize?: number;
-        maxWordsPerChunk?: number;
-        minWordsPerChunk?: number;
-    }>) => Promise<void>;
-    query: (query: Readonly<string | string[] | number | number[]>, { topPercent }?: Readonly<{
-        topPercent?: number;
-    }> | undefined) => Promise<AxDBMatch[][]>;
-}