@ax-llm/ax 9.0.61 → 10.0.2

package/util/sse.js

@@ -0,0 +1,93 @@
+import { TransformStream, TransformStreamDefaultController } from 'stream/web';
+// eslint-disable-next-line @typescript-eslint/naming-convention
+export class SSEParser extends TransformStream {
+    dataParser;
+    buffer = '';
+    currentEvent = { rawData: '' };
+    constructor(dataParser = JSON.parse) {
+        super({
+            transform: (chunk, controller) => this.handleChunk(chunk, controller),
+            flush: (controller) => this.handleFlush(controller)
+        });
+        this.dataParser = dataParser;
+    }
+    handleChunk(chunk, controller) {
+        this.buffer += chunk;
+        this.processBuffer(controller);
+    }
+    handleFlush(controller) {
+        this.processBuffer(controller);
+        if (this.currentEvent.rawData) {
+            this.emitEvent(controller);
+        }
+    }
+    processBuffer(controller) {
+        const lines = this.buffer.split(/\r\n|\r|\n/);
+        this.buffer = lines.pop() || '';
+        for (const line of lines) {
+            if (line.trim() === '') {
+                this.emitEvent(controller);
+            }
+            else {
+                this.parseLine(line);
+            }
+        }
+    }
+    parseLine(line) {
+        const colonIndex = line.indexOf(':');
+        if (colonIndex === -1) {
+            // If there's no colon, treat the whole line as data
+            this.currentEvent.rawData += this.currentEvent.rawData
+                ? '\n' + line.trim()
+                : line.trim();
+            return;
+        }
+        const field = line.slice(0, colonIndex).trim();
+        const value = line.slice(colonIndex + 1).trim();
+        switch (field) {
+            case 'event':
+                this.currentEvent.event = value;
+                break;
+            case 'data':
+                this.currentEvent.rawData += this.currentEvent.rawData
+                    ? '\n' + value
+                    : value;
+                break;
+            case 'id':
+                this.currentEvent.id = value;
+                break;
+            case 'retry': {
+                const retryValue = parseInt(value, 10);
+                if (!isNaN(retryValue)) {
+                    this.currentEvent.retry = retryValue;
+                }
+                break;
+            }
+        }
+    }
+    emitEvent(controller) {
+        if (this.currentEvent.rawData) {
+            // Check for special "[DONE]" message or other non-JSON data
+            if (this.currentEvent.rawData.trim() === '[DONE]' ||
+                this.currentEvent.rawData.trim().startsWith('[')) {
+                return;
+            }
+            else {
+                try {
+                    // Attempt to parse the data using the provided dataParser
+                    const parsedData = this.dataParser(this.currentEvent.rawData);
+                    // Emit only if we successfully parsed the data
+                    controller.enqueue(parsedData);
+                }
+                catch (e) {
+                    // If parsing fails, log the error without emitting
+                    console.warn('Failed to parse event data:', e);
+                    console.log('Raw data that failed to parse:', this.currentEvent.rawData);
+                }
+            }
+            // Reset the current event
+            this.currentEvent = { rawData: '' };
+        }
+    }
+}
+//# sourceMappingURL=sse.js.map

package/util/sse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sse.js","sourceRoot":"","sources":["../../util/sse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,gCAAgC,EAAE,MAAM,YAAY,CAAC;AAS/E,gEAAgE;AAChE,MAAM,OAAO,SAAuB,SAAQ,eAA0B;IAIvC;IAHrB,MAAM,GAAW,EAAE,CAAC;IACpB,YAAY,GAAsB,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IAE1D,YAA6B,aAAkC,IAAI,CAAC,KAAK;QACvE,KAAK,CAAC;YACJ,SAAS,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,UAAU,CAAC;YACrE,KAAK,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;SACpD,CAAC,CAAC;QAJwB,eAAU,GAAV,UAAU,CAAkC;IAKzE,CAAC;IAEO,WAAW,CACjB,KAAa,EACb,UAA+C;QAE/C,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IACjC,CAAC;IAEO,WAAW,CAAC,UAA+C;QACjE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAC/B,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,UAA+C;QACnE,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEhC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBACvB,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,IAAY;QAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;YACtB,oDAAoD;YACpD,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO;gBACpD,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE;gBACpB,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAEhD,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,IAAI,CAAC,YAAY,CAAC,KAAK,GAAG,KAAK,CAAC;gBAChC,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO;oBACpD,CAAC,CAAC,IAAI,GAAG,KAAK;oBACd,CAAC,CAAC,KAAK,CAAC;gBACV,MAAM;YACR,KAAK,IAAI;gBACP,IAAI,CAAC,YAAY,CAAC,EAAE,GAAG,KAAK,CAAC;gBAC7B,MAAM;YACR,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBACvC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;oBACvB,IAAI,CAAC,YAAY,CAAC,KAAK,GAAG,UAAU,CAAC;gBACvC,CAAC;gBACD,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,UAA+C;QAC/D,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC9B,4DAA4D;YAC5D,IACE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,QAAQ;gBAC7C,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAChD,CAAC;gBACD,OAAO;YACT,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC;oBACH,0DAA0D;oBAC1D,MAAM,UAAU,GAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;oBACjE,+CAA+C;oBAC/C,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACjC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,mDAAmD;oBACnD,OAAO,CAAC,IAAI,CAAC,6BAA6B,EAAE,CAAC,CAAC,CAAC;oBAC/C,OAAO,CAAC,GAAG,CACT,gCAAgC,EAChC,IAAI,CAAC,YAAY,CAAC,OAAO,CAC1B,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,0BAA0B;YAC1B,IAAI,CAAC,YAAY,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;CACF"}

package/util/transform.d.ts

@@ -1,7 +1,4 @@
 import { TransformStream } from 'stream/web';
-export declare class JSONStringifyStream<O> extends TransformStream<string, O> {
-    constructor();
-}
 export declare class RespTransformStream<I, O> extends TransformStream<I, O> {
     constructor(transformFn: (arg0: I) => O, doneCallback?: (args0: readonly O[]) => Promise<void>);
 }

package/util/transform.js

@@ -1,26 +1,4 @@
 import { TransformStream } from 'stream/web';
-class JSONTransformer {
-    async transform(obj, controller) {
-        (obj.split('\n') ?? [])
-            .map(extractJson)
-            .map((v) => {
-            try {
-                return v && v.length > 0 ? JSON.parse(v) : null;
-            }
-            catch (e) {
-                return null;
-            }
-        })
-            .filter((v) => v)
-            .forEach((v) => v && controller.enqueue(v));
-    }
-}
-// eslint-disable-next-line @typescript-eslint/naming-convention
-export class JSONStringifyStream extends TransformStream {
-    constructor() {
-        super(new JSONTransformer());
-    }
-}
 class TypeTransformer {
     buffer;
     doneCallback;
@@ -48,12 +26,4 @@ export class RespTransformStream extends TransformStream {
         super(new TypeTransformer(transformFn, doneCallback));
     }
 }
-const extractJson = (str) => {
-    const startIndex = str.indexOf('{');
-    const endIndex = str.lastIndexOf('}');
-    if (startIndex !== -1 && endIndex !== -1) {
-        return str.substring(startIndex, endIndex + 1);
-    }
-    return null;
-};
 //# sourceMappingURL=transform.js.map

package/util/transform.js.map

@@ -1 +1 @@
-{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../util/transform.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,eAAe,EAEhB,MAAM,YAAY,CAAC;AAEpB,MAAM,eAAe;IACnB,KAAK,CAAC,SAAS,CACb,GAAW,EACX,UAA+C;QAE/C,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;aACpB,GAAG,CAAC,WAAW,CAAC;aAChB,GAAG,CAAI,CAAC,CAAC,EAAE,EAAE;YACZ,IAAI,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5D,CAAC;YAAC,OAAO,CAAU,EAAE,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;aAChB,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;CACF;AAED,gEAAgE;AAChE,MAAM,OAAO,mBAAuB,SAAQ,eAA0B;IACpE;QACE,KAAK,CAAC,IAAI,eAAe,EAAK,CAAC,CAAC;IAClC,CAAC;CACF;AAED,MAAM,eAAe;IACX,MAAM,CAAO;IACb,YAAY,CAA0C;IACtD,WAAW,CAAiB;IAEpC,YACE,WAA2B,EAC3B,YAAqD;QAErD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAM,EAAE,UAA+C;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA+C;QACzD,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC7C,UAAU,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;CACF;AAED,gEAAgE;AAChE,MAAM,OAAO,mBAA0B,SAAQ,eAAqB;IAClE,YACE,WAA2B,EAC3B,YAAqD;QAErD,KAAK,CAAC,IAAI,eAAe,CAAO,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF;AAED,MAAM,WAAW,GAAG,CAAC,GAAW,EAAiB,EAAE;IACjD,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAEtC,IAAI,UAAU,KAAK,CAAC,CAAC,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;QACzC,OAAO,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC,CAAC"}
+{"version":3,"file":"transform.js","sourceRoot":"","sources":["../../util/transform.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,eAAe,EAEhB,MAAM,YAAY,CAAC;AAEpB,MAAM,eAAe;IACX,MAAM,CAAO;IACb,YAAY,CAA0C;IACtD,WAAW,CAAiB;IAEpC,YACE,WAA2B,EAC3B,YAAqD;QAErD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAM,EAAE,UAA+C;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA+C;QACzD,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC7C,UAAU,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;CACF;AAED,gEAAgE;AAChE,MAAM,OAAO,mBAA0B,SAAQ,eAAqB;IAClE,YACE,WAA2B,EAC3B,YAAqD;QAErD,KAAK,CAAC,IAAI,eAAe,CAAO,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF"}

package/cjs/dsp/gen2.d.ts

@@ -1,8 +0,0 @@
-import type { AxAIService } from '../ai/types.js';
-import { AxGenerate, type AxGenerateOptions, type AxGenerateResult } from './generate.js';
-import type { AxGenIn, AxGenOut, AxProgramForwardOptions } from './program.js';
-import type { AxSignature } from './sig.js';
-export declare class AxGen<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<AxGenOut> = AxGenerateResult<AxGenOut>> extends AxGenerate<IN, OUT> {
-    constructor(signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);
-    forward(ai: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
-}

package/cjs/dsp/gen2.js

@@ -1,15 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AxGen = void 0;
-const generate_js_1 = require("./generate.js");
-class AxGen extends generate_js_1.AxGenerate {
-    constructor(signature, options) {
-        super(null, signature, options);
-    }
-    // @ts-expect-error changing the over ridden function type
-    async forward(ai, values, options) {
-        return await super.forward(values, { ...options, ai });
-    }
-}
-exports.AxGen = AxGen;
-//# sourceMappingURL=gen2.js.map

package/cjs/dsp/gen2.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"gen2.js","sourceRoot":"","sources":["../../../dsp/gen2.ts"],"names":[],"mappings":";;;AAEA,+CAIuB;AAIvB,MAAa,KAGX,SAAQ,wBAAmB;IAC3B,YACE,SAAyC,EACzC,OAAqC;QAErC,KAAK,CAAC,IAA8B,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,0DAA0D;IAC1D,KAAK,CAAC,OAAO,CACX,EAAyB,EACzB,MAAU,EACV,OAA2C;QAE3C,OAAO,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,CAAC;CACF;AAnBD,sBAmBC"}

package/dsp/gen2.d.ts

@@ -1,8 +0,0 @@
-import type { AxAIService } from '../ai/types.js';
-import { AxGenerate, type AxGenerateOptions, type AxGenerateResult } from './generate.js';
-import type { AxGenIn, AxGenOut, AxProgramForwardOptions } from './program.js';
-import type { AxSignature } from './sig.js';
-export declare class AxGen<IN extends AxGenIn = AxGenIn, OUT extends AxGenerateResult<AxGenOut> = AxGenerateResult<AxGenOut>> extends AxGenerate<IN, OUT> {
-    constructor(signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);
-    forward(ai: Readonly<AxAIService>, values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
-}

package/dsp/gen2.js

@@ -1,11 +0,0 @@
-import { AxGenerate } from './generate.js';
-export class AxGen extends AxGenerate {
-    constructor(signature, options) {
-        super(null, signature, options);
-    }
-    // @ts-expect-error changing the over ridden function type
-    async forward(ai, values, options) {
-        return await super.forward(values, { ...options, ai });
-    }
-}
-//# sourceMappingURL=gen2.js.map

package/dsp/gen2.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"gen2.js","sourceRoot":"","sources":["../../dsp/gen2.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,UAAU,EAGX,MAAM,eAAe,CAAC;AAIvB,MAAM,OAAO,KAGX,SAAQ,UAAmB;IAC3B,YACE,SAAyC,EACzC,OAAqC;QAErC,KAAK,CAAC,IAA8B,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,0DAA0D;IAC1D,KAAK,CAAC,OAAO,CACX,EAAyB,EACzB,MAAU,EACV,OAA2C;QAE3C,OAAO,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,CAAC;CACF"}