@ax-llm/ax 9.0.21 → 9.0.22

package/cjs/mem/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./memory.js"), exports);
+__exportStar(require("./types.js"), exports);
+//# sourceMappingURL=index.js.map

package/cjs/mem/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../mem/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA4B;AAC5B,6CAA2B"}

package/cjs/mem/memory.d.ts

@@ -0,0 +1,16 @@
+import type { AxChatRequest, AxChatResponseResult } from '../ai/types.js';
+import type { AxAIMemory } from './types.js';
+export declare class AxMemory implements AxAIMemory {
+    private data;
+    private sdata;
+    private limit;
+    constructor(limit?: number);
+    add(value: Readonly<AxChatRequest['chatPrompt'][0] | AxChatRequest['chatPrompt']>, sessionId?: string): void;
+    addResult({ content, name, functionCalls }: Readonly<AxChatResponseResult>, sessionId?: string): void;
+    updateResult({ content, name, functionCalls }: Readonly<AxChatResponseResult>, sessionId?: string): void;
+    history(sessionId?: string): AxChatRequest['chatPrompt'];
+    peek(sessionId?: string): AxChatRequest['chatPrompt'];
+    getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;
+    reset(sessionId?: string): void;
+    private get;
+}

package/cjs/mem/memory.js

@@ -0,0 +1,81 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AxMemory = void 0;
+class AxMemory {
+    data = [];
+    sdata = new Map();
+    limit;
+    constructor(limit = 50) {
+        if (limit <= 0) {
+            throw Error("argument 'last' must be greater than 0");
+        }
+        this.limit = limit;
+    }
+    add(value, sessionId) {
+        const d = this.get(sessionId);
+        let n = 0;
+        if (Array.isArray(value)) {
+            n = d.push(...structuredClone(value));
+        }
+        else {
+            n = d.push({
+                ...structuredClone(value)
+            });
+        }
+        if (d.length > this.limit) {
+            d.splice(0, this.limit + n - this.limit);
+        }
+    }
+    addResult({ content, name, functionCalls }, sessionId) {
+        if (!content && (!functionCalls || functionCalls.length === 0)) {
+            return;
+        }
+        this.add({ content, name, role: 'assistant', functionCalls }, sessionId);
+    }
+    updateResult({ content, name, functionCalls }, sessionId) {
+        const items = this.get(sessionId);
+        const lastItem = items.at(-1);
+        if (!lastItem || lastItem.role !== 'assistant') {
+            this.addResult({ content, name, functionCalls }, sessionId);
+            return;
+        }
+        if ('content' in lastItem && content) {
+            lastItem.content = content;
+        }
+        if ('name' in lastItem && name) {
+            lastItem.name = name;
+        }
+        if ('functionCalls' in lastItem && functionCalls) {
+            lastItem.functionCalls = functionCalls;
+        }
+    }
+    history(sessionId) {
+        return this.get(sessionId);
+    }
+    peek(sessionId) {
+        return this.get(sessionId);
+    }
+    getLast(sessionId) {
+        const d = this.get(sessionId);
+        return d.at(-1);
+    }
+    reset(sessionId) {
+        if (!sessionId) {
+            this.data = [];
+        }
+        else {
+            this.sdata.set(sessionId, []);
+        }
+    }
+    get(sessionId) {
+        if (!sessionId) {
+            return this.data;
+        }
+        if (!this.sdata.has(sessionId)) {
+            this.sdata.set(sessionId, []);
+        }
+        return this.sdata.get(sessionId) || [];
+    }
+}
+exports.AxMemory = AxMemory;
+//# sourceMappingURL=memory.js.map

package/cjs/mem/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../mem/memory.ts"],"names":[],"mappings":";;;AAMA,MAAa,QAAQ;IACX,IAAI,GAAgC,EAAE,CAAC;IACvC,KAAK,GAAG,IAAI,GAAG,EAAuC,CAAC;IACvD,KAAK,CAAS;IAEtB,YAAY,KAAK,GAAG,EAAE;QACpB,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,GAAG,CACD,KAEC,EACD,SAAkB;QAElB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,GAAG,CAAC,CAAC;QAEV,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACT,GAAG,eAAe,CAAC,KAAK,CAAC;aACQ,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;YAC1B,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,SAAS,CACP,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAkC,EAChE,SAAkB;QAElB,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC,aAAa,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CACV,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAkC,EAChE,SAAkB;QAElB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAE3B,CAAC;QAEF,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/C,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,SAAS,CAAC,CAAC;YAC5D,OAAO;QACT,CAAC;QAED,IAAI,SAAS,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;YACrC,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;QAC7B,CAAC;QACD,IAAI,MAAM,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;YAC/B,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC;QACvB,CAAC;QACD,IAAI,eAAe,IAAI,QAAQ,IAAI,aAAa,EAAE,CAAC;YACjD,QAAQ,CAAC,aAAa,GAAG,aAAa,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,SAAkB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,CAAC,SAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,SAAkB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,GAAG,CAAC,SAAkB;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;CACF;AArGD,4BAqGC"}

package/cjs/mem/types.d.ts

@@ -0,0 +1,10 @@
+import type { AxChatRequest, AxChatResponseResult } from '../ai/index.js';
+export interface AxAIMemory {
+    add(result: Readonly<AxChatRequest['chatPrompt'] | AxChatRequest['chatPrompt'][0]>, sessionId?: string): void;
+    addResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
+    updateResult(result: Readonly<AxChatResponseResult>, sessionId?: string): void;
+    history(sessionId?: string): AxChatRequest['chatPrompt'];
+    peek(sessionId?: string): AxChatRequest['chatPrompt'];
+    reset(sessionId?: string): void;
+    getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;
+}

package/cjs/mem/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/cjs/mem/types.js.map

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

package/cjs/prompts/agent.d.ts

@@ -0,0 +1,34 @@
+import type { AxAIService, AxFunction } from '../ai/types.js';
+import { type AxGenerateOptions, AxSignature } from '../dsp/index.js';
+import { type AxGenIn, type AxGenOut, type AxProgramDemos, type AxProgramExamples, type AxProgramForwardOptions, type AxTunable, type AxUsable } from '../dsp/program.js';
+export interface AxAgentic extends AxTunable, AxUsable {
+    getFunction(): AxFunction;
+}
+export type AxAgentOptions = Omit<AxGenerateOptions, 'functions' | 'functionCall'>;
+export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> implements AxAgentic {
+    private program;
+    private name;
+    private description;
+    private subAgentList?;
+    private func;
+    constructor(ai: AxAIService, { name, description, signature, agents, functions }: Readonly<{
+        name: string;
+        description: string;
+        signature: AxSignature | string;
+        agents?: AxAgentic[];
+        functions?: AxFunction[];
+    }>, options?: Readonly<AxAgentOptions>);
+    getSignature(): AxSignature;
+    setExamples(examples: Readonly<AxProgramExamples>): void;
+    setId(id: string): void;
+    setParentId(parentId: string): void;
+    getTraces(): import("../dsp/program.js").AxProgramTrace[];
+    setDemos(demos: readonly AxProgramDemos[]): void;
+    getUsage(): (import("../ai/types.js").AxTokenUsage & {
+        ai: string;
+        model: string;
+    })[];
+    resetUsage(): void;
+    getFunction(): AxFunction;
+    forward(values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
+}

package/cjs/prompts/agent.js

@@ -0,0 +1,110 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AxAgent = void 0;
+const index_js_1 = require("../trace/index.js");
+const cot_js_1 = require("./cot.js");
+const react_js_1 = require("./react.js");
+class AxAgent {
+    program;
+    name;
+    description;
+    subAgentList;
+    func;
+    constructor(ai, { name, description, signature, agents, functions }, options) {
+        const funcs = [
+            ...(functions ?? []),
+            ...(agents?.map((a) => a.getFunction()) ?? [])
+        ];
+        const opt = {
+            ...options,
+            functions: funcs
+        };
+        this.program =
+            funcs.length > 0
+                ? new react_js_1.AxReAct(ai, signature, opt)
+                : new cot_js_1.AxChainOfThought(ai, signature, opt);
+        if (!name || name.length < 5) {
+            throw new Error('Agent name must be at least 10 characters (more descriptive): ' + name);
+        }
+        if (!description || description.length < 20) {
+            throw new Error('Agent description must be at least 20 characters (explain in detail what the agent does): ' +
+                description);
+        }
+        this.name = name;
+        this.description = description;
+        this.subAgentList = agents?.map((a) => a.getFunction().name).join(', ');
+        this.func = {
+            name: toCamelCase(this.name),
+            description: this.description,
+            parameters: this.program.getSignature().toJSONSchema(),
+            func: () => this.forward
+        };
+        for (const agent of agents ?? []) {
+            this.program.register(agent);
+        }
+    }
+    getSignature() {
+        return this.program.getSignature();
+    }
+    setExamples(examples) {
+        this.program.setExamples(examples);
+    }
+    setId(id) {
+        this.program.setId(id);
+    }
+    setParentId(parentId) {
+        this.program.setParentId(parentId);
+    }
+    getTraces() {
+        return this.program.getTraces();
+    }
+    setDemos(demos) {
+        this.program.setDemos(demos);
+    }
+    getUsage() {
+        return this.program.getUsage();
+    }
+    resetUsage() {
+        this.program.resetUsage();
+    }
+    getFunction() {
+        return this.func;
+    }
+    async forward(values, options) {
+        if (!options?.tracer) {
+            return await this.program.forward(values, options);
+        }
+        const attributes = {
+            ['agent.name']: this.name,
+            ['agent.description']: this.description,
+            ['agent.subAgents']: this.subAgentList ?? 'none'
+        };
+        return await options?.tracer.startActiveSpan('Agent', {
+            kind: index_js_1.AxSpanKind.SERVER,
+            attributes
+        }, async (span) => {
+            const res = await this.program.forward(values, options);
+            span.end();
+            return res;
+        });
+    }
+}
+exports.AxAgent = AxAgent;
+function toCamelCase(inputString) {
+    // Split the string by any non-alphanumeric character (including underscores, spaces, hyphens)
+    const words = inputString.split(/[^a-zA-Z0-9]/);
+    // Map through each word, capitalize the first letter of each word except the first word
+    const camelCaseString = words
+        .map((word, index) => {
+        // Lowercase the word to handle cases like uppercase letters in input
+        const lowerWord = word.toLowerCase();
+        // Capitalize the first letter of each word except the first one
+        if (index > 0 && lowerWord && lowerWord[0]) {
+            return lowerWord[0].toUpperCase() + lowerWord.slice(1);
+        }
+        return lowerWord;
+    })
+        .join('');
+    return camelCaseString;
+}
+//# sourceMappingURL=agent.js.map

package/cjs/prompts/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../prompts/agent.ts"],"names":[],"mappings":";;;AAYA,gDAA+C;AAE/C,qCAA4C;AAC5C,yCAAqC;AAWrC,MAAa,OAAO;IAGV,OAAO,CAAqB;IAE5B,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAa;IAEzB,YACE,EAAe,EACf,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAkC;QAElC,MAAM,KAAK,GAAiB;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC;YACpB,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;SAC/C,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,GAAG,OAAO;YACV,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,OAAO;YACV,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,kBAAO,CAAU,EAAE,EAAE,SAAS,EAAE,GAAG,CAAC;gBAC1C,CAAC,CAAC,IAAI,yBAAgB,CAAU,EAAE,EAAE,SAAS,EAAE,GAAG,CAAC,CAAC;QAExD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CACb,gEAAgE,GAAG,IAAI,CACxE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,4FAA4F;gBAC1F,WAAW,CACd,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,YAAY,EAAE;YACtD,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEM,YAAY;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;IACrC,CAAC;IAEM,WAAW,CAAC,QAAqC;QACtD,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,KAAK,CAAC,EAAU;QACrB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IAEM,WAAW,CAAC,QAAgB;QACjC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAEM,SAAS;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAEM,QAAQ,CAAC,KAAgC;QAC9C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAEM,QAAQ;QACb,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IACjC,CAAC;IAEM,UAAU;QACf,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC5B,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO,CAClB,MAAU,EACV,OAA2C;QAE3C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,IAAI;YACzB,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,WAAW;YACvC,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,YAAY,IAAI,MAAM;SACjD,CAAC;QAEF,OAAO,MAAM,OAAO,EAAE,MAAM,CAAC,eAAe,CAC1C,OAAO,EACP;YACE,IAAI,EAAE,qBAAU,CAAC,MAAM;YACvB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AAtID,0BAsIC;AAED,SAAS,WAAW,CAAC,WAAmB;IACtC,8FAA8F;IAC9F,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAEhD,wFAAwF;IACxF,MAAM,eAAe,GAAG,KAAK;SAC1B,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACnB,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAErC,gEAAgE;QAChE,IAAI,KAAK,GAAG,CAAC,IAAI,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3C,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO,eAAe,CAAC;AACzB,CAAC"}

package/cjs/prompts/cot.d.ts

@@ -0,0 +1,9 @@
+import type { AxAIService } from '../ai/types.js';
+import { AxGenerate, type AxGenerateOptions } from '../dsp/generate.js';
+import type { AxGenIn, AxGenOut } from '../dsp/program.js';
+import { AxSignature } from '../dsp/sig.js';
+export declare class AxChainOfThought<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxGenerate<IN, OUT & {
+    reason: string;
+}> {
+    constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);
+}

package/cjs/prompts/cot.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AxChainOfThought = void 0;
+const generate_js_1 = require("../dsp/generate.js");
+const sig_js_1 = require("../dsp/sig.js");
+class AxChainOfThought extends generate_js_1.AxGenerate {
+    constructor(ai, signature, options) {
+        const sig = new sig_js_1.AxSignature(signature);
+        const description = `Let's work this out in a step by step way in order to ensure we have the right answer.`;
+        sig.setOutputFields([
+            {
+                name: 'reason',
+                description
+            },
+            ...sig.getOutputFields()
+        ]);
+        super(ai, sig, options);
+    }
+}
+exports.AxChainOfThought = AxChainOfThought;
+//# sourceMappingURL=cot.js.map

package/cjs/prompts/cot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../prompts/cot.ts"],"names":[],"mappings":";;;AACA,oDAAwE;AAExE,0CAA4C;AAE5C,MAAa,gBAGX,SAAQ,wBAAwC;IAChD,YACE,EAAe,EACf,SAAyC,EACzC,OAAqC;QAErC,MAAM,GAAG,GAAG,IAAI,oBAAW,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,wFAAwF,CAAC;QAE7G,GAAG,CAAC,eAAe,CAAC;YAClB;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW;aACZ;YACD,GAAG,GAAG,CAAC,eAAe,EAAE;SACzB,CAAC,CAAC;QAEH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF;AAtBD,4CAsBC"}

package/cjs/prompts/index.d.ts

@@ -0,0 +1,4 @@
+export * from './cot.js';
+export * from './react.js';
+export * from './rag.js';
+export * from './agent.js';

package/cjs/prompts/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./cot.js"), exports);
+__exportStar(require("./react.js"), exports);
+__exportStar(require("./rag.js"), exports);
+__exportStar(require("./agent.js"), exports);
+//# sourceMappingURL=index.js.map

package/cjs/prompts/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../prompts/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,6CAA2B;AAC3B,2CAAyB;AACzB,6CAA2B"}

package/cjs/prompts/rag.d.ts

@@ -0,0 +1,23 @@
+import type { AxAIService } from '../ai/types.js';
+import { type AxGenerateOptions } from '../dsp/index.js';
+import { type AxProgramForwardOptions } from '../dsp/program.js';
+import { AxChainOfThought } from './cot.js';
+export declare class AxRAG extends AxChainOfThought<{
+    context: string[];
+    question: string;
+}, {
+    answer: string;
+}> {
+    private genQuery;
+    private queryFn;
+    private maxHops;
+    constructor(ai: AxAIService, queryFn: (query: string) => Promise<string>, options: Readonly<AxGenerateOptions & {
+        maxHops?: number;
+    }>);
+    forward({ question }: Readonly<{
+        question: string;
+    }>, options?: Readonly<AxProgramForwardOptions>): Promise<{
+        answer: string;
+        reason: string;
+    }>;
+}

package/cjs/prompts/rag.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AxRAG = void 0;
+const index_js_1 = require("../dsp/index.js");
+const cot_js_1 = require("./cot.js");
+class AxRAG extends cot_js_1.AxChainOfThought {
+    genQuery;
+    queryFn;
+    maxHops;
+    constructor(ai, queryFn, options) {
+        const sig = '"Answer questions with short factoid answers." context:string[] "may contain relevant facts", question -> answer';
+        super(ai, sig, options);
+        this.maxHops = options?.maxHops ?? 3;
+        const qsig = new index_js_1.AxSignature('"Write a simple search query that will help answer a complex question." context?:string[] "may contain relevant facts", question -> query "question to further our understanding"');
+        this.genQuery = new index_js_1.AxGenerate(ai, qsig);
+        this.queryFn = queryFn;
+        this.register(this.genQuery);
+    }
+    async forward({ question }, options) {
+        let context = [];
+        for (let i = 0; i < this.maxHops; i++) {
+            const { query } = await this.genQuery.forward({
+                context,
+                question
+            }, options);
+            const val = await this.queryFn(query);
+            context = index_js_1.axStringUtil.dedup([...context, val]);
+        }
+        return super.forward({ context, question }, options);
+    }
+}
+exports.AxRAG = AxRAG;
+//# sourceMappingURL=rag.js.map

package/cjs/prompts/rag.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../prompts/rag.ts"],"names":[],"mappings":";;;AACA,8CAKyB;AAGzB,qCAA4C;AAE5C,MAAa,KAAM,SAAQ,yBAG1B;IACS,QAAQ,CAGd;IACM,OAAO,CAAqC;IAC5C,OAAO,CAAS;IAExB,YACE,EAAe,EACf,OAA2C,EAC3C,OAA2D;QAE3D,MAAM,GAAG,GACP,kHAAkH,CAAC;QACrH,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAExB,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,CAAC,CAAC;QAErC,MAAM,IAAI,GAAG,IAAI,sBAAW,CAC1B,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,qBAAU,CAG5B,EAAE,EAAE,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,EAAE,QAAQ,EAAkC,EAC5C,OAA2C;QAE3C,IAAI,OAAO,GAAa,EAAE,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAC3C;gBACE,OAAO;gBACP,QAAQ;aACT,EACD,OAAO,CACR,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtC,OAAO,GAAG,uBAAY,CAAC,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;CACF;AArDD,sBAqDC"}

package/cjs/prompts/react.d.ts

@@ -0,0 +1,8 @@
+import type { AxAIService } from '../ai/index.js';
+import type { AxGenerateOptions } from '../dsp/generate.js';
+import type { AxGenIn, AxGenOut } from '../dsp/program.js';
+import { AxSignature } from '../dsp/sig.js';
+import { AxChainOfThought } from './cot.js';
+export declare class AxReAct<IN extends AxGenIn = AxGenIn, OUT extends AxGenOut = AxGenOut> extends AxChainOfThought<IN, OUT> {
+    constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options: Readonly<AxGenerateOptions>);
+}

package/cjs/prompts/react.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AxReAct = void 0;
+const sig_js_1 = require("../dsp/sig.js");
+const cot_js_1 = require("./cot.js");
+class AxReAct extends cot_js_1.AxChainOfThought {
+    constructor(ai, signature, options) {
+        if (!options?.functions || options.functions.length === 0) {
+            throw new Error('No functions provided');
+        }
+        const functions = [...options.functions];
+        const funcList = functions.map((f) => `'${f.name}'`).join(', ');
+        const sig = new sig_js_1.AxSignature(signature);
+        sig.setDescription(`Use the provided functions ${funcList} to complete the task and return the result if any.`);
+        // sig.addInputField({
+        //   name: 'observation',
+        //   description: 'Result value from executed function',
+        //   isOptional: true,
+        //   type: { name: 'string', isArray: true }
+        // });
+        sig.setOutputFields(sig.getOutputFields().map((v) => ({ ...v, isOptional: true })));
+        super(ai, sig, options);
+    }
+}
+exports.AxReAct = AxReAct;
+//# sourceMappingURL=react.js.map

package/cjs/prompts/react.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"react.js","sourceRoot":"","sources":["../../../prompts/react.ts"],"names":[],"mappings":";;;AAGA,0CAA4C;AAE5C,qCAA4C;AAE5C,MAAa,OAGX,SAAQ,yBAAyB;IACjC,YACE,EAAe,EACf,SAAyC,EACzC,OAAoC;QAEpC,IAAI,CAAC,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1D,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,GAAG,GAAG,IAAI,oBAAW,CAAC,SAAS,CAAC,CAAC;QACvC,GAAG,CAAC,cAAc,CAChB,8BAA8B,QAAQ,qDAAqD,CAC5F,CAAC;QAEF,sBAAsB;QACtB,yBAAyB;QACzB,wDAAwD;QACxD,sBAAsB;QACtB,4CAA4C;QAC5C,MAAM;QAEN,GAAG,CAAC,eAAe,CACjB,GAAG,CAAC,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAC/D,CAAC;QAEF,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF;AAlCD,0BAkCC"}

package/cjs/trace/index.d.ts

@@ -0,0 +1,171 @@
+export declare const axSpanAttributes: {
+    LLM_SYSTEM: string;
+    LLM_REQUEST_MODEL: string;
+    LLM_REQUEST_MAX_TOKENS: string;
+    LLM_REQUEST_TEMPERATURE: string;
+    LLM_REQUEST_TOP_K: string;
+    LLM_REQUEST_FREQUENCY_PENALTY: string;
+    LLM_REQUEST_PRESENCE_PENALTY: string;
+    LLM_REQUEST_STOP_SEQUENCES: string;
+    LLM_REQUEST_LLM_IS_STREAMING: string;
+    LLM_REQUEST_TOP_P: string;
+    LLM_USAGE_PROMPT_TOKENS: string;
+    LLM_USAGE_COMPLETION_TOKENS: string;
+    DB_SYSTEM: string;
+    DB_TABLE: string;
+    DB_NAMESPACE: string;
+    DB_ID: string;
+    DB_QUERY_TEXT: string;
+    DB_VECTOR: string;
+    DB_OPERATION_NAME: string;
+    DB_VECTOR_QUERY_TOP_K: string;
+    DB_QUERY_EMBEDDINGS: string;
+    DB_QUERY_RESULT: string;
+    DB_QUERY_EMBEDDINGS_VECTOR: string;
+    DB_QUERY_RESULT_ID: string;
+    DB_QUERY_RESULT_SCORE: string;
+    DB_QUERY_RESULT_DISTANCE: string;
+    DB_QUERY_RESULT_METADATA: string;
+    DB_QUERY_RESULT_VECTOR: string;
+    DB_QUERY_RESULT_DOCUMENT: string;
+};
+export declare const axSpanEvents: {
+    LLM_PROMPT: string;
+};
+export declare enum AxLLMRequestTypeValues {
+    COMPLETION = "completion",
+    CHAT = "chat",
+    RERANK = "rerank",
+    UNKNOWN = "unknown"
+}
+export declare enum AxSpanKindValues {
+    WORKFLOW = "workflow",
+    TASK = "task",
+    AGENT = "agent",
+    TOOL = "tool",
+    UNKNOWN = "unknown"
+}
+export type AxSpanAttributeValue = string | number | boolean | undefined | null;
+export interface AxSpanAttributes {
+    [key: string]: AxSpanAttributeValue;
+}
+export interface AxSpanTimeInput {
+    timestamp: number;
+}
+export interface AxSpanException {
+    message: string;
+    name: string;
+    stack?: string;
+}
+export interface AxSpanContext {
+    traceId: string;
+    spanId: string;
+}
+export declare enum AxSpanStatusCode {
+    OK = "OK",
+    ERROR = "ERROR"
+}
+export interface AxSpanStatus {
+    code: AxSpanStatusCode;
+    message?: string;
+}
+export interface AxSpanStatus {
+    code: AxSpanStatusCode;
+    description?: string;
+}
+export interface AxSpan {
+    addEvent(name: string, attributesOrStartTime?: Readonly<AxSpanAttributes | AxSpanTimeInput>, startTime?: Readonly<AxSpanTimeInput>): AxSpan;
+    end(endTime?: Readonly<AxSpanTimeInput>): void;
+    isRecording(): boolean;
+    recordAxSpanException(exception: Readonly<AxSpanException>, time?: Readonly<AxSpanTimeInput>): void;
+    setAttribute(key: string, value: AxSpanAttributeValue): AxSpan;
+    setAttributes(attributes: AxSpanAttributes): AxSpan;
+    setStatus(status: Readonly<AxSpanStatus>): AxSpan;
+    spanContext(): AxSpanContext;
+    updateName(name: string): AxSpan;
+}
+export interface AxContext {
+    /**
+     * Deletes a value associated with a key from the context.
+     * Returns a new context that inherits from the current context but does not contain the value for the specified key.
+     * @param key The symbol key for which to clear the value.
+     * @returns A new Context instance without the specified key.
+     */
+    deleteValue(key: symbol): AxContext;
+    /**
+     * Retrieves a value from the context using a symbol as the key.
+     * @param key The symbol key which identifies a context value.
+     * @returns The value associated with the key, if any; otherwise, undefined.
+     */
+    getValue(key: symbol): unknown;
+    /**
+     * Sets a value in the context for the specified key.
+     * Returns a new context that inherits from the current context with the new key-value pair added.
+     * @param key The symbol key for which to set the value.
+     * @param value The value to set for the given key.
+     * @returns A new Context instance with the updated key-value pair.
+     */
+    setValue(key: symbol, value: unknown): AxContext;
+}
+export declare enum AxSpanKind {
+    INTERNAL = "INTERNAL",
+    SERVER = "SERVER",
+    CLIENT = "CLIENT",
+    PRODUCER = "PRODUCER",
+    CONSUMER = "CONSUMER"
+}
+export interface AxSpanLink {
+    context: AxSpanContext;
+    attributes?: AxSpanAttributes;
+}
+export interface AxSpanOptions {
+    /**
+     * Optional attributes that can be attached to the AxSpan.
+     * @optional
+     */
+    attributes?: AxSpanAttributes;
+    /**
+     * The kind of span, defaults to AxSpanKind.INTERNAL if not specified.
+     * @optional
+     */
+    kind?: AxSpanKind;
+    /**
+     *  AxSpanLinks that associate this new AxSpan with other AxSpans.
+     * @optional
+     */
+    links?: AxSpanLink[];
+    /**
+     * Indicates whether the span should be a root span, ignoring any parent span from the context.
+     * @optional
+     */
+    root?: boolean;
+    /**
+     * A manually specified start time for the span, if required.
+     * @optional
+     */
+    startTime?: AxSpanTimeInput;
+}
+export interface AxSpan {
+    setStatus(status: Readonly<AxSpanStatus>): void;
+    end(): void;
+}
+export interface AxTracer {
+    /**
+     * Starts a new AxSpan with the specified options and executes the provided function. The span is automatically closed after the function executes.
+     * @param name The name of the span.
+     * @param options AxSpan options to apply to the span.
+     * @param fn The function to execute within the span's context.
+     * @returns The return value of the function.
+     */
+    startActiveSpan<F extends (span: AxSpan) => unknown>(name: string, options: Readonly<AxSpanOptions>, fn: F): ReturnType<F>;
+    /**
+     * Starts a new AxSpan with the specified options and context, then executes the provided function. The span is automatically closed after the function executes.
+     * @param name The name of the span.
+     * @param options AxSpan options to apply to the span.
+     * @param context Context to be used for the span.
+     * @param fn The function to execute within the span's context.
+     * @returns The return value of the function.
+     */
+    startActiveSpan<F extends (span: AxSpan) => unknown>(name: string, options: Readonly<AxSpanOptions>, context: AxContext, fn: F): ReturnType<F>;
+    startSpan(name: string, options?: Readonly<AxSpanOptions>, context?: AxContext): AxSpan;
+}