@ax-llm/ax 10.0.24 → 10.0.26

package/dsp/generate.js

@@ -1,251 +0,0 @@
-import { ReadableStream } from 'stream/web';
-import { mergeFunctionCalls } from '../ai/util.js';
-import { AxMemory } from '../mem/index.js';
-import { AxSpanKind } from '../trace/index.js';
-import { assertAssertions, assertRequiredFields, assertStreamingAssertions, AxAssertionError } from './asserts.js';
-import { extractValues, streamingExtractFinalValue, streamingExtractValues } from './extract.js';
-import { parseFunctionCalls, parseFunctions, processFunctions } from './functions.js';
-import { AxProgramWithSignature } from './program.js';
-import { AxPromptTemplate } from './prompt.js';
-import { AxSignature } from './sig.js';
-import { AxValidationError } from './validate.js';
-export class AxGen extends AxProgramWithSignature {
-    promptTemplate;
-    asserts;
-    streamingAsserts;
-    options;
-    functions;
-    functionsExecuted = new Set();
-    constructor(signature, options) {
-        super(signature, { description: options?.description });
-        this.options = options;
-        this.promptTemplate = new (options?.promptTemplate ?? AxPromptTemplate)(this.signature, options?.functions);
-        this.asserts = this.options?.asserts ?? [];
-        this.streamingAsserts = this.options?.streamingAsserts ?? [];
-        this.usage = [];
-        if (options?.functions) {
-            this.functions = parseFunctions(options.functions);
-        }
-    }
-    addAssert = (fn, message, optional) => {
-        this.asserts.push({ fn, message, optional });
-    };
-    addStreamingAssert = (fieldName, fn, message, optional) => {
-        this.streamingAsserts.push({ fieldName, fn, message, optional });
-    };
-    async forwardSendRequest({ ai, mem, options }) {
-        const { sessionId, traceId, modelConfig, model, rateLimiter, stream, functions, functionCall: _functionCall } = options ?? {};
-        const chatPrompt = mem?.history(sessionId) ?? [];
-        if (chatPrompt.length === 0) {
-            throw new Error('No chat prompt found');
-        }
-        const functionCall = _functionCall ?? this.options?.functionCall;
-        const res = await ai.chat({
-            chatPrompt,
-            functions,
-            functionCall,
-            modelConfig,
-            model
-        }, {
-            sessionId,
-            traceId,
-            rateLimiter,
-            stream
-        });
-        return res;
-    }
-    async forwardCore({ ai, mem, options }) {
-        const { sessionId, traceId, model, functions } = options ?? {};
-        const usageInfo = {
-            ai: ai.getName(),
-            model: ai.getModelInfo().name
-        };
-        const res = await this.forwardSendRequest({
-            ai,
-            mem,
-            options
-        });
-        if (res instanceof ReadableStream) {
-            return (await this.processSteamingResponse({
-                ai,
-                model,
-                res,
-                usageInfo,
-                mem,
-                traceId,
-                sessionId,
-                functions
-            }));
-        }
-        return (await this.processResponse({
-            ai,
-            model,
-            res,
-            usageInfo,
-            mem,
-            traceId,
-            sessionId,
-            functions
-        }));
-    }
-    async processSteamingResponse({ ai, model, res, usageInfo, mem, sessionId, traceId, functions }) {
-        const functionCalls = [];
-        const values = {};
-        const xstate = { s: -1 };
-        let content = '';
-        for await (const v of res) {
-            for (const result of v.results ?? []) {
-                if (v.modelUsage) {
-                    this.usage.push({ ...usageInfo, ...v.modelUsage });
-                }
-                if (result.content) {
-                    content += result.content;
-                    mem.updateResult({ name: result.name, content }, sessionId);
-                    assertStreamingAssertions(this.streamingAsserts, values, xstate, content, false);
-                    streamingExtractValues(this.signature, values, xstate, content);
-                    assertAssertions(this.asserts, values);
-                }
-                if (result.functionCalls) {
-                    mergeFunctionCalls(functionCalls, result.functionCalls);
-                    mem.updateResult({ name: result.name, content, functionCalls }, sessionId);
-                }
-                if (result.finishReason === 'length') {
-                    throw new Error('Max tokens reached before completion');
-                }
-            }
-        }
-        const funcs = parseFunctionCalls(ai, functionCalls, values, model);
-        if (funcs) {
-            if (!functions) {
-                throw new Error('Functions are not defined');
-            }
-            const fx = await processFunctions(ai, functions, funcs, mem, sessionId, traceId);
-            this.functionsExecuted = new Set([...this.functionsExecuted, ...fx]);
-        }
-        streamingExtractFinalValue(values, xstate, content);
-        assertStreamingAssertions(this.streamingAsserts, values, xstate, content, true);
-        assertAssertions(this.asserts, values);
-        return { ...values };
-    }
-    async processResponse({ ai, res, usageInfo, mem, sessionId, traceId, functions }) {
-        const values = {};
-        for (const result of res.results ?? []) {
-            if (res.modelUsage) {
-                this.usage.push({ ...usageInfo, ...res.modelUsage });
-            }
-            mem.addResult(result, sessionId);
-            if (result.content) {
-                extractValues(this.signature, values, result.content);
-                assertAssertions(this.asserts, values);
-            }
-            if (result.functionCalls) {
-                const funcs = parseFunctionCalls(ai, result.functionCalls, values);
-                if (funcs) {
-                    if (!functions) {
-                        throw new Error('Functions are not defined');
-                    }
-                    const fx = await processFunctions(ai, functions, funcs, mem, sessionId, traceId);
-                    this.functionsExecuted = new Set([...this.functionsExecuted, ...fx]);
-                }
-            }
-            if (result.finishReason === 'length') {
-                throw new Error('Max tokens reached before completion');
-            }
-        }
-        return { ...values };
-    }
-    async _forward(ai, values, options, span) {
-        const stopFunction = (options?.stopFunction ?? this.options?.stopFunction)?.toLowerCase();
-        const maxRetries = options?.maxRetries ?? this.options?.maxRetries ?? 15;
-        const maxSteps = options?.maxSteps ?? this.options?.maxSteps ?? 10;
-        const mem = options?.mem ?? this.options?.mem ?? new AxMemory();
-        let err;
-        if (options?.functions && options?.functions.length > 0) {
-            const promptTemplate = this.options?.promptTemplate ?? AxPromptTemplate;
-            this.promptTemplate = new promptTemplate(this.signature, options.functions);
-        }
-        const prompt = this.promptTemplate.render(values, {
-            examples: this.examples,
-            demos: this.demos
-        });
-        mem.add(prompt, options?.sessionId);
-        multiStepLoop: for (let n = 0; n < maxSteps; n++) {
-            for (let i = 0; i < maxRetries; i++) {
-                try {
-                    const output = await this.forwardCore({
-                        options,
-                        ai,
-                        mem
-                    });
-                    const lastMemItem = mem.getLast(options?.sessionId);
-                    const stopFunctionExecuted = stopFunction && this.functionsExecuted.has(stopFunction);
-                    if (lastMemItem?.role === 'function') {
-                        if (!stopFunction || !stopFunctionExecuted) {
-                            continue multiStepLoop;
-                        }
-                    }
-                    if (!stopFunctionExecuted) {
-                        assertRequiredFields(this.signature, output);
-                    }
-                    this.trace = { ...values, ...output };
-                    return output;
-                }
-                catch (e) {
-                    let extraFields;
-                    span?.recordAxSpanException(e);
-                    if (e instanceof AxValidationError) {
-                        extraFields = e.getFixingInstructions();
-                        err = e;
-                    }
-                    else if (e instanceof AxAssertionError) {
-                        const e1 = e;
-                        extraFields = e1.getFixingInstructions(this.signature);
-                        err = e;
-                    }
-                    else {
-                        throw e;
-                    }
-                    if (extraFields) {
-                        const content = this.promptTemplate.renderExtraFields(extraFields);
-                        mem.add({ role: 'user', content }, options?.sessionId);
-                        if (options?.debug) {
-                            console.log('Error Correction:', content);
-                        }
-                    }
-                }
-            }
-            if (err instanceof AxAssertionError && err.getOptional()) {
-                return err.getValue();
-            }
-            throw new Error(`Unable to fix validation error: ${err?.message}`);
-        }
-        throw new Error(`Max steps reached: ${maxSteps}`);
-    }
-    async forward(ai, values, options) {
-        const tracer = this.options?.tracer ?? options?.tracer;
-        let functions = this.functions;
-        if (options?.functions) {
-            functions = parseFunctions(options.functions, this.functions);
-        }
-        if (!tracer) {
-            return await this._forward(ai, values, {
-                ...options,
-                functions
-            });
-        }
-        const funcNames = functions?.map((f) => f.name).join(',');
-        const attributes = {
-            ['generate.signature']: this.signature.toString(),
-            ['generate.functions']: funcNames ?? ''
-        };
-        return await tracer.startActiveSpan('Generate', {
-            kind: AxSpanKind.SERVER,
-            attributes
-        }, async (span) => {
-            const res = this._forward(ai, values, options, span);
-            span.end();
-            return res;
-        });
-    }
-}
-//# sourceMappingURL=generate.js.map

package/dsp/generate.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../dsp/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAU5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAmB,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAAe,UAAU,EAAiB,MAAM,mBAAmB,CAAC;AAE3E,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,yBAAyB,EAEzB,gBAAgB,EAEjB,MAAM,cAAc,CAAC;AACtB,OAAO,EAEL,aAAa,EACb,0BAA0B,EAC1B,sBAAsB,EACvB,MAAM,cAAc,CAAC;AACtB,OAAO,EAGL,kBAAkB,EAClB,cAAc,EACd,gBAAgB,EACjB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAIL,sBAAsB,EACvB,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AACvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAoClD,MAAM,OAAO,KAGX,SAAQ,sBAA+B;IAC/B,cAAc,CAAmB;IACjC,OAAO,CAAgB;IACvB,gBAAgB,CAAyB;IACzC,OAAO,CAAmC;IAC1C,SAAS,CAAgB;IACzB,iBAAiB,GAAgB,IAAI,GAAG,EAAU,CAAC;IAE3D,YACE,SAAyC,EACzC,OAAgC;QAEhC,KAAK,CAAC,SAAS,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAExD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,OAAO,EAAE,cAAc,IAAI,gBAAgB,CAAC,CACrE,IAAI,CAAC,SAAS,EACd,OAAO,EAAE,SAAS,CACnB,CAAC;QACF,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,OAAO,EAAE,gBAAgB,IAAI,EAAE,CAAC;QAC7D,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAEhB,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,SAAS,GAAG,cAAc,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAEM,SAAS,GAAG,CACjB,EAAqB,EACrB,OAAgB,EAChB,QAAkB,EAClB,EAAE;QACF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC/C,CAAC,CAAC;IAEK,kBAAkB,GAAG,CAC1B,SAAiB,EACjB,EAA8B,EAC9B,OAAgB,EAChB,QAAkB,EAClB,EAAE;QACF,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IACnE,CAAC,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,EAC/B,EAAE,EACF,GAAG,EACH,OAAO,EAKP;QACA,MAAM,EACJ,SAAS,EACT,OAAO,EACP,WAAW,EACX,KAAK,EACL,WAAW,EACX,MAAM,EACN,SAAS,EACT,YAAY,EAAE,aAAa,EAC5B,GAAG,OAAO,IAAI,EAAE,CAAC;QAElB,MAAM,UAAU,GAAG,GAAG,EAAE,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QAEjD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,YAAY,GAAG,aAAa,IAAI,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC;QAEjE,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,IAAI,CACvB;YACE,UAAU;YACV,SAAS;YACT,YAAY;YACZ,WAAW;YACX,KAAK;SACN,EACD;YACE,SAAS;YACT,OAAO;YACP,WAAW;YACX,MAAM;SACP,CACF,CAAC;QAEF,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,EACxB,EAAE,EACF,GAAG,EACH,OAAO,EAKP;QACA,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;QAE/D,MAAM,SAAS,GAAG;YAChB,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE;YAChB,KAAK,EAAE,EAAE,CAAC,YAAY,EAAE,CAAC,IAAI;SAC9B,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC;YACxC,EAAE;YACF,GAAG;YACH,OAAO;SACR,CAAC,CAAC;QAEH,IAAI,GAAG,YAAY,cAAc,EAAE,CAAC;YAClC,OAAO,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC;gBACzC,EAAE;gBACF,KAAK;gBACL,GAAG;gBACH,SAAS;gBACT,GAAG;gBACH,OAAO;gBACP,SAAS;gBACT,SAAS;aACV,CAAC,CAAmB,CAAC;QACxB,CAAC;QAED,OAAO,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC;YACjC,EAAE;YACF,KAAK;YACL,GAAG;YACH,SAAS;YACT,GAAG;YACH,OAAO;YACP,SAAS;YACT,SAAS;SACV,CAAC,CAAmB,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,EACpC,EAAE,EACF,KAAK,EACL,GAAG,EACH,SAAS,EACT,GAAG,EACH,SAAS,EACT,OAAO,EACP,SAAS,EAGV;QACC,MAAM,aAAa,GACjB,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,EAAE,CAAC;QAClB,MAAM,MAAM,GAAoB,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAE1C,IAAI,OAAO,GAAG,EAAE,CAAC;QAEjB,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;YAC1B,KAAK,MAAM,MAAM,IAAI,CAAC,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;gBACrC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;oBACjB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,SAAS,EAAE,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC;gBACrD,CAAC;gBAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC;oBAE1B,GAAG,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,SAAS,CAAC,CAAC;oBAE5D,yBAAyB,CACvB,IAAI,CAAC,gBAAgB,EACrB,MAAM,EACN,MAAM,EACN,OAAO,EACP,KAAK,CACN,CAAC;oBACF,sBAAsB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;oBAChE,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACzC,CAAC;gBAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBACzB,kBAAkB,CAAC,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;oBAExD,GAAG,CAAC,YAAY,CACd,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,aAAa,EAAE,EAC7C,SAAS,CACV,CAAC;gBACJ,CAAC;gBAED,IAAI,MAAM,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;oBACrC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,kBAAkB,CAAC,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QACnE,IAAI,KAAK,EAAE,CAAC;YACV,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YACD,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAC/B,EAAE,EACF,SAAS,EACT,KAAK,EACL,GAAG,EACH,SAAS,EACT,OAAO,CACR,CAAC;YACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QACvE,CAAC;QAED,0BAA0B,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QACpD,yBAAyB,CACvB,IAAI,CAAC,gBAAgB,EACrB,MAAM,EACN,MAAM,EACN,OAAO,EACP,IAAI,CACL,CAAC;QACF,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAEvC,OAAO,EAAE,GAAG,MAAM,EAAoB,CAAC;IACzC,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,EAC5B,EAAE,EACF,GAAG,EACH,SAAS,EACT,GAAG,EACH,SAAS,EACT,OAAO,EACP,SAAS,EACuC;QAChD,MAAM,MAAM,GAAG,EAAE,CAAC;QAElB,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YACvC,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBACnB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,SAAS,EAAE,GAAG,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAEjC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;gBACtD,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACzC,CAAC;YAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,kBAAkB,CAAC,EAAE,EAAE,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;gBAEnE,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,CAAC,SAAS,EAAE,CAAC;wBACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;oBAC/C,CAAC;oBACD,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAC/B,EAAE,EACF,SAAS,EACT,KAAK,EACL,GAAG,EACH,SAAS,EACT,OAAO,CACR,CAAC;oBACF,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;gBACvE,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QAED,OAAO,EAAE,GAAG,MAAM,EAAoB,CAAC;IACzC,CAAC;IAEO,KAAK,CAAC,QAAQ,CACpB,EAAyB,EACzB,MAAU,EACV,OAA2C,EAC3C,IAAa;QAEb,MAAM,YAAY,GAAG,CACnB,OAAO,EAAE,YAAY,IAAI,IAAI,CAAC,OAAO,EAAE,YAAY,CACpD,EAAE,WAAW,EAAE,CAAC;QAEjB,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,IAAI,CAAC,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;QACzE,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC;QACnE,MAAM,GAAG,GAAG,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,QAAQ,EAAE,CAAC;QAEhE,IAAI,GAAqD,CAAC;QAE1D,IAAI,OAAO,EAAE,SAAS,IAAI,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,EAAE,cAAc,IAAI,gBAAgB,CAAC;YACxE,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CACtC,IAAI,CAAC,SAAS,EACd,OAAO,CAAC,SAAS,CAClB,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAK,MAAM,EAAE;YACpD,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;QAEH,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAEpC,aAAa,EAAE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC;wBACpC,OAAO;wBACP,EAAE;wBACF,GAAG;qBACJ,CAAC,CAAC;oBAEH,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;oBAEpD,MAAM,oBAAoB,GACxB,YAAY,IAAI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAE3D,IAAI,WAAW,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;wBACrC,IAAI,CAAC,YAAY,IAAI,CAAC,oBAAoB,EAAE,CAAC;4BAC3C,SAAS,aAAa,CAAC;wBACzB,CAAC;oBACH,CAAC;oBAED,IAAI,CAAC,oBAAoB,EAAE,CAAC;wBAC1B,oBAAoB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;oBAC/C,CAAC;oBAED,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;oBACtC,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,IAAI,WAAW,CAAC;oBAChB,IAAI,EAAE,qBAAqB,CAAC,CAAU,CAAC,CAAC;oBAExC,IAAI,CAAC,YAAY,iBAAiB,EAAE,CAAC;wBACnC,WAAW,GAAG,CAAC,CAAC,qBAAqB,EAAE,CAAC;wBACxC,GAAG,GAAG,CAAC,CAAC;oBACV,CAAC;yBAAM,IAAI,CAAC,YAAY,gBAAgB,EAAE,CAAC;wBACzC,MAAM,EAAE,GAAG,CAAqB,CAAC;wBACjC,WAAW,GAAG,EAAE,CAAC,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBACvD,GAAG,GAAG,CAAC,CAAC;oBACV,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,CAAC;oBACV,CAAC;oBAED,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;wBACnE,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;wBAEhE,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;4BACnB,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;wBAC5C,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,GAAG,YAAY,gBAAgB,IAAI,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC;gBACzD,OAAO,GAAG,CAAC,QAAQ,EAAS,CAAC;YAC/B,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,mCAAmC,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,sBAAsB,QAAQ,EAAE,CAAC,CAAC;IACpD,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,EAAyB,EACzB,MAAU,EACV,OAA2C;QAE3C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,OAAO,EAAE,MAAM,CAAC;QAEvD,IAAI,SAAS,GAA6B,IAAI,CAAC,SAAS,CAAC;QAEzD,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YACvB,SAAS,GAAG,cAAc,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,MAAM,EAAE;gBACrC,GAAG,OAAO;gBACV,SAAS;aACV,CAAC,CAAC;QACL,CAAC;QAED,MAAM,SAAS,GAAG,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE1D,MAAM,UAAU,GAAG;YACjB,CAAC,oBAAoB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE;YACjD,CAAC,oBAAoB,CAAC,EAAE,SAAS,IAAI,EAAE;SACxC,CAAC;QAEF,OAAO,MAAM,MAAM,CAAC,eAAe,CACjC,UAAU,EACV;YACE,IAAI,EAAE,UAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}

package/dsp/index.d.ts

@@ -1,15 +0,0 @@
-export * from './generate.js';
-export * from './sig.js';
-export * from './optimize.js';
-export * from './evaluate.js';
-export * from './program.js';
-export * from './eval.js';
-export * from './loader.js';
-export * from './strutil.js';
-export * from './router.js';
-export * from './loader.js';
-export type { AxAssertion, AxStreamingAssertion } from './asserts.js';
-export type { AxPromptTemplate, AxFieldTemplateFn } from './prompt.js';
-export type { AxInstanceRegistry } from './registry.js';
-export type { AxResponseHandlerArgs } from './generate.js';
-export type { ParsedSignature, InputParsedFieldList, OutputParsedFieldList, ParsedIdentifier } from './parser.js';

package/dsp/index.js

@@ -1,11 +0,0 @@
-export * from './generate.js';
-export * from './sig.js';
-export * from './optimize.js';
-export * from './evaluate.js';
-export * from './program.js';
-export * from './eval.js';
-export * from './loader.js';
-export * from './strutil.js';
-export * from './router.js';
-export * from './loader.js';
-//# sourceMappingURL=index.js.map

package/dsp/index.js.map

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

package/dsp/jsonschema.d.ts

@@ -1,2 +0,0 @@
-import type { AxFunctionJSONSchema } from '../ai/index.js';
-export declare const validateJSONSchema: (schema: Readonly<AxFunctionJSONSchema>) => void;

package/dsp/jsonschema.js

@@ -1,97 +0,0 @@
-// eslint-disable-next-line @typescript-eslint/naming-convention
-export const validateJSONSchema = (schema) => {
-    const errors = [];
-    const validateSchemaObject = (schema, path = '') => {
-        const validTypes = [
-            'array',
-            'integer',
-            'number',
-            'string',
-            'boolean',
-            'null',
-            'object'
-        ];
-        if (!validTypes.includes(schema.type)) {
-            errors.push(`Invalid type '${schema.type}' at ${path || 'root'}`);
-            return;
-        }
-        if (schema.type === 'object' && schema.properties) {
-            if (typeof schema.properties !== 'object' ||
-                Array.isArray(schema.properties)) {
-                errors.push(`Invalid properties object at ${path || 'root'}`);
-            }
-            else {
-                for (const key in schema.properties) {
-                    const value = schema.properties[key];
-                    if (typeof value !== 'object') {
-                        errors.push(`Invalid schema object at ${path}${key}`);
-                        continue;
-                    }
-                    validateSchemaObject(value, `${path}${key}.`);
-                }
-            }
-            if (schema.required && !Array.isArray(schema.required)) {
-                errors.push(`'required' should be an array at ${path || 'root'}`);
-            }
-        }
-        if (schema.type === 'array' && schema.items) {
-            if (typeof schema.items !== 'object') {
-                errors.push(`Invalid items schema at ${path || 'root'}`);
-            }
-            else {
-                validateSchemaObject(schema.items, `${path}items.`);
-            }
-        }
-    };
-    validateSchemaObject(schema);
-    if (errors.length > 0) {
-        throw new Error(errors.join('; '));
-    }
-};
-// Example Usage:
-/*
-const validSchema: AxFunctionJSONSchema = {
-  type: 'object',
-  properties: {
-    id: { type: 'integer' },
-    name: { type: 'string' },
-    email: { type: 'string' },
-    isActive: { type: 'boolean' },
-    tags: {
-      type: 'array',
-      items: { type: 'string' }
-    }
-  },
-  required: ['id', 'name', 'email']
-};
-
-const invalidSchema: any = {
-  type: 'object',
-  properties: {
-    id: { type: 'integer' },
-    name: { type: 'string' },
-    email: { type: 'unknownType' }, // Invalid type
-    isActive: { type: 'boolean' },
-    tags: {
-      type: 'array',
-      items: { type: 'string' }
-    }
-  },
-  required: 'id,name,email' // Invalid 'required' field
-};
-
-try {
-  validateSchemaStructure(validSchema);
-  console.log('Schema is valid!');
-} catch (error) {
-  console.error('Schema validation failed:', error.message);
-}
-
-try {
-  validateSchemaStructure(invalidSchema);
-  console.log('Schema is valid!');
-} catch (error) {
-  console.error('Schema validation failed:', error.message);
-}
-*/
-//# sourceMappingURL=jsonschema.js.map

package/dsp/jsonschema.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"jsonschema.js","sourceRoot":"","sources":["../../dsp/jsonschema.ts"],"names":[],"mappings":"AAEA,gEAAgE;AAChE,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,MAAsC,EAChC,EAAE;IACR,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,MAAM,oBAAoB,GAAG,CAC3B,MAAsC,EACtC,OAAe,EAAE,EACX,EAAE;QACR,MAAM,UAAU,GAAG;YACjB,OAAO;YACP,SAAS;YACT,QAAQ;YACR,QAAQ;YACR,SAAS;YACT,MAAM;YACN,QAAQ;SACT,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,iBAAiB,MAAM,CAAC,IAAI,QAAQ,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAClD,IACE,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;gBACrC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAChC,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,gCAAgC,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YAChE,CAAC;iBAAM,CAAC;gBACN,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;oBACpC,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;wBAC9B,MAAM,CAAC,IAAI,CAAC,4BAA4B,IAAI,GAAG,GAAG,EAAE,CAAC,CAAC;wBACtD,SAAS;oBACX,CAAC;oBACD,oBAAoB,CAAC,KAAK,EAAE,GAAG,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvD,MAAM,CAAC,IAAI,CAAC,oCAAoC,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YACpE,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAC5C,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACrC,MAAM,CAAC,IAAI,CAAC,2BAA2B,IAAI,IAAI,MAAM,EAAE,CAAC,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACN,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,IAAI,QAAQ,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAE7B,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACrC,CAAC;AACH,CAAC,CAAC;AAEF,iBAAiB;AAEjB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4CE"}

package/dsp/loader.d.ts

@@ -1,30 +0,0 @@
-import type { AxFieldValue } from './program.js';
-export type AxDataRow = {
-    row: Record<string, AxFieldValue>;
-};
-export declare class AxHFDataLoader {
-    private rows;
-    private baseUrl;
-    private dataset;
-    private split;
-    private config;
-    private options?;
-    constructor({ dataset, split, config, options }: Readonly<{
-        dataset: string;
-        split: string;
-        config: string;
-        options?: Readonly<{
-            offset?: number;
-            length?: number;
-        }>;
-    }>);
-    private fetchDataFromAPI;
-    loadData(): Promise<AxDataRow[]>;
-    setData(rows: AxDataRow[]): void;
-    getData(): AxDataRow[];
-    getRows<T>({ count, fields, renameMap }: Readonly<{
-        count: number;
-        fields: readonly string[];
-        renameMap?: Record<string, string>;
-    }>): Promise<T[]>;
-}

package/dsp/loader.js

@@ -1,81 +0,0 @@
-export class AxHFDataLoader {
-    rows = [];
-    baseUrl;
-    dataset;
-    split;
-    config;
-    options;
-    constructor({ dataset, split, config, options }) {
-        this.baseUrl = 'https://datasets-server.huggingface.co/rows';
-        this.dataset = dataset;
-        this.split = split;
-        this.config = config;
-        this.options = options;
-    }
-    async fetchDataFromAPI(url) {
-        try {
-            const response = await fetch(url);
-            if (!response.ok) {
-                throw new Error(`Error fetching data: ${response.statusText}`);
-            }
-            const data = (await response.json());
-            if (!data?.rows) {
-                throw new Error('Invalid data format');
-            }
-            return data.rows;
-        }
-        catch (error) {
-            console.error('Error fetching data from API:', error);
-            throw error;
-        }
-    }
-    // https://datasets-server.huggingface.co/rows?dataset=hotpot_qa&config=distractor&split=train&offset=0&length=100
-    async loadData() {
-        const offset = this.options?.offset ?? 0;
-        const length = this.options?.length ?? 100;
-        const ds = encodeURIComponent(this.dataset);
-        const url = `${this.baseUrl}?dataset=${ds}&config=${this.config}&split=${this.split}&offset=${offset}&length=${length}`;
-        console.log('Downloading data from API.');
-        this.rows = (await this.fetchDataFromAPI(url));
-        return this.rows;
-    }
-    // eslint-disable-next-line functional/prefer-immutable-types
-    setData(rows) {
-        this.rows = rows;
-    }
-    getData() {
-        return this.rows;
-    }
-    async getRows({ count, fields, renameMap }) {
-        if (this.rows.length === 0) {
-            throw new Error('No data loaded, call loadData or setData first.');
-        }
-        const dataRows = this.rows.slice(0, count);
-        return dataRows
-            .map((item) => {
-            const result = {};
-            fields.forEach((field) => {
-                const keys = field.split('.');
-                // Initial value should match the type of the rows, and be indexable by string
-                let value = item.row;
-                for (const key of keys) {
-                    // Use type assertion to tell TypeScript that value will always be an object that can be indexed with string keys
-                    if (Object.prototype.hasOwnProperty.call(value, key)) {
-                        value = value[key];
-                    }
-                }
-                if (!value) {
-                    return;
-                }
-                const resultFieldName = renameMap && field in renameMap ? renameMap[field] : field;
-                if (!resultFieldName) {
-                    throw new Error(`Invalid field name: ${field}`);
-                }
-                result[resultFieldName] = value;
-            });
-            return result;
-        })
-            .filter((v) => Object.keys(v).length !== 0);
-    }
-}
-//# sourceMappingURL=loader.js.map

package/dsp/loader.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"loader.js","sourceRoot":"","sources":["../../dsp/loader.ts"],"names":[],"mappings":"AAIA,MAAM,OAAO,cAAc;IACjB,IAAI,GAAgB,EAAE,CAAC;IACvB,OAAO,CAAS;IAEhB,OAAO,CAAS;IAChB,KAAK,CAAS;IACd,MAAM,CAAS;IACf,OAAO,CAAkD;IAEjE,YAAY,EACV,OAAO,EACP,KAAK,EACL,MAAM,EACN,OAAO,EAMP;QACA,IAAI,CAAC,OAAO,GAAG,6CAA6C,CAAC;QAC7D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,GAAW;QACxC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;YAClC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACjE,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA0B,CAAC;YAC9D,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YACD,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,kHAAkH;IAE3G,KAAK,CAAC,QAAQ;QACnB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,GAAG,CAAC;QAC3C,MAAM,EAAE,GAAG,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE5C,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,YAAY,EAAE,WAAW,IAAI,CAAC,MAAM,UAAU,IAAI,CAAC,KAAK,WAAW,MAAM,WAAW,MAAM,EAAE,CAAC;QAExH,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAgB,CAAC;QAC9D,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,6DAA6D;IACtD,OAAO,CAAC,IAAiB;QAC9B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAEM,OAAO;QACZ,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO,CAAI,EACtB,KAAK,EACL,MAAM,EACN,SAAS,EAKT;QACA,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAE3C,OAAO,QAAQ;aACZ,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACZ,MAAM,MAAM,GAAiC,EAAE,CAAC;YAEhD,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC9B,8EAA8E;gBAC9E,IAAI,KAAK,GAA2B,IAAI,CAAC,GAAG,CAAC;gBAC7C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;oBACvB,iHAAiH;oBACjH,IACE,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAClC,KAAgC,EAChC,GAAG,CACJ,EACD,CAAC;wBACD,KAAK,GAAI,KAAiC,CAAC,GAAG,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,OAAO;gBACT,CAAC;gBACD,MAAM,eAAe,GACnB,SAAS,IAAI,KAAK,IAAI,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBAC7D,IAAI,CAAC,eAAe,EAAE,CAAC;oBACrB,MAAM,IAAI,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;gBAClD,CAAC;gBACD,MAAM,CAAC,eAAe,CAAC,GAAG,KAAqB,CAAC;YAClD,CAAC,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,CAAQ,CAAC;IACvD,CAAC;CACF"}

package/dsp/modelinfo.d.ts

@@ -1,8 +0,0 @@
-import type { AxModelInfo } from '../ai/types.js';
-interface GetModelInfoParams {
-    model: string;
-    modelInfo: readonly AxModelInfo[];
-    modelMap?: Record<string, string>;
-}
-export declare function axGetModelInfo({ model, modelInfo, modelMap }: Readonly<GetModelInfoParams>): Readonly<AxModelInfo>;
-export {};

package/dsp/modelinfo.js

@@ -1,32 +0,0 @@
-export function axGetModelInfo({ model, modelInfo, modelMap = {} }) {
-    // First check if there's a mapping for this model
-    const mappedModel = modelMap?.[model] ?? model;
-    // Try exact match first
-    const exactMatch = modelInfo.find((v) => v.name === model);
-    if (exactMatch)
-        return exactMatch;
-    // Handle normalization if no exact match
-    const normalizedName = mappedModel
-        // Remove vendor prefixes
-        .replace(/^(anthropic\.|openai\.)/, '')
-        // Remove various postfixes one by one, stopping after first match
-        .replace(/-latest$/, '')
-        .replace(/-\d{8}$/, '') // YYYYMMDD
-        .replace(/-v\d+:\d+$/, '') // v2:0
-        .replace(/@\d{8}$/, '') // @YYYYMMDD
-        .replace(/-\d{2,}(-[a-zA-Z0-9-]+)?$/, '') // XX or XXXXX-something
-        .replace(/-v\d+@\d{8}$/, '') // vX@YYYYMMDD
-        .replace(/-v\d+$/, ''); // Remove standalone version number
-    // Try to find a match with the normalized name
-    const normalizedMatch = modelInfo.find((v) => v.name === normalizedName);
-    if (normalizedMatch)
-        return normalizedMatch;
-    // Return default if no match found
-    return {
-        name: model,
-        currency: 'usd',
-        promptTokenCostPer1M: 0,
-        completionTokenCostPer1M: 0
-    };
-}
-//# sourceMappingURL=modelinfo.js.map

package/dsp/modelinfo.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"modelinfo.js","sourceRoot":"","sources":["../../dsp/modelinfo.ts"],"names":[],"mappings":"AAQA,MAAM,UAAU,cAAc,CAAC,EAC7B,KAAK,EACL,SAAS,EACT,QAAQ,GAAG,EAAE,EACgB;IAC7B,kDAAkD;IAClD,MAAM,WAAW,GAAG,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC;IAE/C,wBAAwB;IACxB,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC;IAC3D,IAAI,UAAU;QAAE,OAAO,UAAU,CAAC;IAElC,yCAAyC;IACzC,MAAM,cAAc,GAAG,WAAW;QAChC,yBAAyB;SACxB,OAAO,CAAC,yBAAyB,EAAE,EAAE,CAAC;QACvC,kEAAkE;SACjE,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,WAAW;SAClC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,OAAO;SACjC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,YAAY;SACnC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC,wBAAwB;SACjE,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,cAAc;SAC1C,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,mCAAmC;IAE7D,+CAA+C;IAC/C,MAAM,eAAe,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;IACzE,IAAI,eAAe;QAAE,OAAO,eAAe,CAAC;IAE5C,mCAAmC;IACnC,OAAO;QACL,IAAI,EAAE,KAAK;QACX,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,CAAC;QACvB,wBAAwB,EAAE,CAAC;KAC5B,CAAC;AACJ,CAAC"}

package/dsp/optimize.d.ts

@@ -1,30 +0,0 @@
-import type { AxAIService } from '../ai/types.js';
-import type { AxFieldValue, AxGenIn, AxGenOut, AxProgram, AxProgramDemos } from './program.js';
-export type AxExample = Record<string, AxFieldValue>;
-export type AxMetricFn = <T extends AxGenOut = AxGenOut>(arg0: Readonly<{
-    prediction: T;
-    example: AxExample;
-}>) => boolean;
-export type AxMetricFnArgs = Parameters<AxMetricFn>[0];
-export type AxOptimizerArgs<IN extends AxGenIn, OUT extends AxGenOut> = {
-    ai: AxAIService;
-    program: Readonly<AxProgram<IN, OUT>>;
-    examples: Readonly<AxExample[]>;
-    options?: {
-        maxRounds?: number;
-        maxExamples?: number;
-        maxDemos?: number;
-    };
-};
-export declare class AxBootstrapFewShot<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> {
-    private ai;
-    private program;
-    private examples;
-    private maxRounds;
-    private maxDemos;
-    private maxExamples;
-    private traces;
-    constructor({ ai, program, examples, options }: Readonly<AxOptimizerArgs<IN, OUT>>);
-    private compileRound;
-    compile(metricFn: AxMetricFn, options?: Readonly<AxOptimizerArgs<IN, OUT>['options']>): Promise<AxProgramDemos[]>;
-}