@ax-llm/ax 9.0.20 → 9.0.22

package/mem/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../mem/memory.ts"],"names":[],"mappings":"AAMA,MAAM,OAAO,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"}

package/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/mem/types.js

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

package/mem/types.js.map

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ax-llm/ax",
-  "version": "9.0.20",
+  "version": "9.0.22",
   "description": "The best library to work with LLMs",
   "repository": {
     "type": "git",
@@ -43,12 +43,9 @@
     ]
   },
   "files": [
-    "build",
-    "!**/*.spec.*",
-    "!**/*.json",
-    "CHANGELOG.md",
-    "LICENSE",
-    "README.md"
+    "**/*",
+    "!**/*.test.*",
+    "!**/*.json"
   ],
   "bugs": {
     "url": "https://github.com/@ax-llm/ax/issues"

package/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/prompts/agent.js

@@ -0,0 +1,108 @@
+import { AxSignature } from '../dsp/index.js';
+import { AxProgram } from '../dsp/program.js';
+import { AxSpanKind } from '../trace/index.js';
+import { AxChainOfThought } from './cot.js';
+import { AxReAct } from './react.js';
+export 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 AxReAct(ai, signature, opt)
+                : new 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: AxSpanKind.SERVER,
+            attributes
+        }, async (span) => {
+            const res = await this.program.forward(values, options);
+            span.end();
+            return res;
+        });
+    }
+}
+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/prompts/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../prompts/agent.ts"],"names":[],"mappings":"AACA,OAAO,EAA0B,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACtE,OAAO,EAGL,SAAS,EAMV,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAWrC,MAAM,OAAO,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,OAAO,CAAU,EAAE,EAAE,SAAS,EAAE,GAAG,CAAC;gBAC1C,CAAC,CAAC,IAAI,gBAAgB,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,UAAU,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;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/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/prompts/cot.js

@@ -0,0 +1,17 @@
+import { AxGenerate } from '../dsp/generate.js';
+import { AxSignature } from '../dsp/sig.js';
+export class AxChainOfThought extends AxGenerate {
+    constructor(ai, signature, options) {
+        const sig = new 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);
+    }
+}
+//# sourceMappingURL=cot.js.map

package/prompts/cot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../prompts/cot.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAA0B,MAAM,oBAAoB,CAAC;AAExE,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,MAAM,OAAO,gBAGX,SAAQ,UAAwC;IAChD,YACE,EAAe,EACf,SAAyC,EACzC,OAAqC;QAErC,MAAM,GAAG,GAAG,IAAI,WAAW,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"}

package/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/prompts/index.js

@@ -0,0 +1,5 @@
+export * from './cot.js';
+export * from './react.js';
+export * from './rag.js';
+export * from './agent.js';
+//# sourceMappingURL=index.js.map

package/prompts/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../prompts/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC;AAC3B,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC"}

package/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/prompts/rag.js

@@ -0,0 +1,30 @@
+import { AxGenerate, AxSignature, axStringUtil } from '../dsp/index.js';
+import {} from '../dsp/program.js';
+import { AxChainOfThought } from './cot.js';
+export class AxRAG extends 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 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 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 = axStringUtil.dedup([...context, val]);
+        }
+        return super.forward({ context, question }, options);
+    }
+}
+//# sourceMappingURL=rag.js.map

package/prompts/rag.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../prompts/rag.ts"],"names":[],"mappings":"AACA,OAAO,EACL,UAAU,EAEV,WAAW,EACX,YAAY,EACb,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAgC,MAAM,mBAAmB,CAAC;AAEjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,KAAM,SAAQ,gBAG1B;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,WAAW,CAC1B,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,UAAU,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,YAAY,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"}

package/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/prompts/react.js

@@ -0,0 +1,22 @@
+import { AxSignature } from '../dsp/sig.js';
+import { AxChainOfThought } from './cot.js';
+export class AxReAct extends 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 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);
+    }
+}
+//# sourceMappingURL=react.js.map

package/prompts/react.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"react.js","sourceRoot":"","sources":["../../prompts/react.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,OAAO,OAGX,SAAQ,gBAAyB;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,WAAW,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"}

package/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;
+}

package/trace/index.js

@@ -0,0 +1,67 @@
+export const axSpanAttributes = {
+    // LLM
+    LLM_SYSTEM: 'gen_ai.system',
+    LLM_REQUEST_MODEL: 'gen_ai.request.model',
+    LLM_REQUEST_MAX_TOKENS: 'gen_ai.request.max_tokens',
+    LLM_REQUEST_TEMPERATURE: 'gen_ai.request.temperature',
+    LLM_REQUEST_TOP_K: 'gen_ai.request.top_k',
+    LLM_REQUEST_FREQUENCY_PENALTY: 'gen_ai.request.frequency_penalty',
+    LLM_REQUEST_PRESENCE_PENALTY: 'gen_ai.request.presence_penalty',
+    LLM_REQUEST_STOP_SEQUENCES: 'gen_ai.request.stop_sequences',
+    LLM_REQUEST_LLM_IS_STREAMING: 'gen_ai.request.llm_is_streaming',
+    LLM_REQUEST_TOP_P: 'gen_ai.request.top_p',
+    LLM_USAGE_PROMPT_TOKENS: 'gen_ai.usage.prompt_tokens',
+    LLM_USAGE_COMPLETION_TOKENS: 'gen_ai.usage.completion_tokens',
+    // Vector DB
+    DB_SYSTEM: 'db.system',
+    DB_TABLE: 'db.table',
+    DB_NAMESPACE: 'db.namespace',
+    DB_ID: 'db.id',
+    DB_QUERY_TEXT: 'db.query.text',
+    DB_VECTOR: 'db.vector',
+    DB_OPERATION_NAME: 'db.operation.name',
+    DB_VECTOR_QUERY_TOP_K: 'db.vector.query.top_k',
+    DB_QUERY_EMBEDDINGS: 'db.query.embeddings',
+    DB_QUERY_RESULT: 'db.query.result',
+    // Query Embeddings
+    DB_QUERY_EMBEDDINGS_VECTOR: 'db.query.embeddings.vector',
+    // Query Result (canonical format)
+    DB_QUERY_RESULT_ID: 'db.query.result.id',
+    DB_QUERY_RESULT_SCORE: 'db.query.result.score',
+    DB_QUERY_RESULT_DISTANCE: 'db.query.result.distance',
+    DB_QUERY_RESULT_METADATA: 'db.query.result.metadata',
+    DB_QUERY_RESULT_VECTOR: 'db.query.result.vector',
+    DB_QUERY_RESULT_DOCUMENT: 'db.query.result.document'
+};
+export const axSpanEvents = {
+    LLM_PROMPT: 'gen_ai.prompt'
+};
+export var AxLLMRequestTypeValues;
+(function (AxLLMRequestTypeValues) {
+    AxLLMRequestTypeValues["COMPLETION"] = "completion";
+    AxLLMRequestTypeValues["CHAT"] = "chat";
+    AxLLMRequestTypeValues["RERANK"] = "rerank";
+    AxLLMRequestTypeValues["UNKNOWN"] = "unknown";
+})(AxLLMRequestTypeValues || (AxLLMRequestTypeValues = {}));
+export var AxSpanKindValues;
+(function (AxSpanKindValues) {
+    AxSpanKindValues["WORKFLOW"] = "workflow";
+    AxSpanKindValues["TASK"] = "task";
+    AxSpanKindValues["AGENT"] = "agent";
+    AxSpanKindValues["TOOL"] = "tool";
+    AxSpanKindValues["UNKNOWN"] = "unknown";
+})(AxSpanKindValues || (AxSpanKindValues = {}));
+export var AxSpanStatusCode;
+(function (AxSpanStatusCode) {
+    AxSpanStatusCode["OK"] = "OK";
+    AxSpanStatusCode["ERROR"] = "ERROR";
+})(AxSpanStatusCode || (AxSpanStatusCode = {}));
+export var AxSpanKind;
+(function (AxSpanKind) {
+    AxSpanKind["INTERNAL"] = "INTERNAL";
+    AxSpanKind["SERVER"] = "SERVER";
+    AxSpanKind["CLIENT"] = "CLIENT";
+    AxSpanKind["PRODUCER"] = "PRODUCER";
+    AxSpanKind["CONSUMER"] = "CONSUMER";
+})(AxSpanKind || (AxSpanKind = {}));
+//# sourceMappingURL=index.js.map

package/trace/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../trace/index.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,MAAM;IACN,UAAU,EAAE,eAAe;IAC3B,iBAAiB,EAAE,sBAAsB;IACzC,sBAAsB,EAAE,2BAA2B;IACnD,uBAAuB,EAAE,4BAA4B;IACrD,iBAAiB,EAAE,sBAAsB;IACzC,6BAA6B,EAAE,kCAAkC;IACjE,4BAA4B,EAAE,iCAAiC;IAC/D,0BAA0B,EAAE,+BAA+B;IAC3D,4BAA4B,EAAE,iCAAiC;IAC/D,iBAAiB,EAAE,sBAAsB;IAEzC,uBAAuB,EAAE,4BAA4B;IACrD,2BAA2B,EAAE,gCAAgC;IAE7D,YAAY;IACZ,SAAS,EAAE,WAAW;IACtB,QAAQ,EAAE,UAAU;IACpB,YAAY,EAAE,cAAc;IAC5B,KAAK,EAAE,OAAO;IACd,aAAa,EAAE,eAAe;IAC9B,SAAS,EAAE,WAAW;IACtB,iBAAiB,EAAE,mBAAmB;IACtC,qBAAqB,EAAE,uBAAuB;IAE9C,mBAAmB,EAAE,qBAAqB;IAC1C,eAAe,EAAE,iBAAiB;IAElC,mBAAmB;IACnB,0BAA0B,EAAE,4BAA4B;IAExD,kCAAkC;IAClC,kBAAkB,EAAE,oBAAoB;IACxC,qBAAqB,EAAE,uBAAuB;IAC9C,wBAAwB,EAAE,0BAA0B;IACpD,wBAAwB,EAAE,0BAA0B;IACpD,sBAAsB,EAAE,wBAAwB;IAChD,wBAAwB,EAAE,0BAA0B;CACrD,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B,UAAU,EAAE,eAAe;CAC5B,CAAC;AAEF,MAAM,CAAN,IAAY,sBAKX;AALD,WAAY,sBAAsB;IAChC,mDAAyB,CAAA;IACzB,uCAAa,CAAA;IACb,2CAAiB,CAAA;IACjB,6CAAmB,CAAA;AACrB,CAAC,EALW,sBAAsB,KAAtB,sBAAsB,QAKjC;AAED,MAAM,CAAN,IAAY,gBAMX;AAND,WAAY,gBAAgB;IAC1B,yCAAqB,CAAA;IACrB,iCAAa,CAAA;IACb,mCAAe,CAAA;IACf,iCAAa,CAAA;IACb,uCAAmB,CAAA;AACrB,CAAC,EANW,gBAAgB,KAAhB,gBAAgB,QAM3B;AAuBD,MAAM,CAAN,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IAC1B,6BAAS,CAAA;IACT,mCAAe,CAAA;AACjB,CAAC,EAHW,gBAAgB,KAAhB,gBAAgB,QAG3B;AA2DD,MAAM,CAAN,IAAY,UAMX;AAND,WAAY,UAAU;IACpB,mCAAqB,CAAA;IACrB,+BAAiB,CAAA;IACjB,+BAAiB,CAAA;IACjB,mCAAqB,CAAA;IACrB,mCAAqB,CAAA;AACvB,CAAC,EANW,UAAU,KAAV,UAAU,QAMrB"}

package/util/apicall.d.ts

@@ -0,0 +1,18 @@
+import { type ReadableStream } from 'stream/web';
+import type { AxSpan } from '../trace/index.js';
+/**
+ * Util: API details
+ * @export
+ */
+export type API = {
+    name?: string;
+    headers?: Record<string, string>;
+    put?: boolean;
+};
+export declare const apiCall: <TRequest = unknown, TResponse = unknown>(api: Readonly<API & {
+    url: string | URL;
+    stream?: boolean;
+    debug?: boolean;
+    fetch?: typeof fetch;
+    span?: AxSpan;
+}>, json: TRequest) => Promise<TResponse | ReadableStream<TResponse>>;