@ax-llm/ax 9.0.0 → 9.0.2

package/build/module/src/funcs/functions.js

@@ -0,0 +1,55 @@
+import JSON5 from 'json5';
+import { validateJSONSchema } from './jsonschema.js';
+export class AxFunctionProcessor {
+    funcList;
+    constructor(funcList) {
+        funcList
+            .filter((v) => v.parameters)
+            .forEach((v) => validateJSONSchema(v.parameters));
+        this.funcList = funcList;
+    }
+    executeFunction = async (fnSpec, func, options) => {
+        if (!fnSpec.func) {
+            throw new Error(`AxFunction handler for ${fnSpec.name} not implemented`);
+        }
+        let argOBj;
+        if (func.args && func.args.length > 0) {
+            argOBj = JSON5.parse(func.args);
+        }
+        if (!fnSpec.parameters) {
+            const res = fnSpec.func.length === 1
+                ? await fnSpec.func(options)
+                : await fnSpec.func();
+            return {
+                name: fnSpec.name,
+                result: JSON.stringify(res, null, 2)
+            };
+        }
+        const res = fnSpec.func.length === 2
+            ? await fnSpec.func(argOBj, options)
+            : await fnSpec.func(argOBj);
+        return {
+            id: func.id,
+            name: func.name,
+            args: func.args,
+            result: JSON.stringify(res, null, 2)
+        };
+    };
+    execute = async (func, options) => {
+        const fnSpec = this.funcList.find((v) => v.name.localeCompare(func.name) === 0);
+        if (!fnSpec) {
+            throw new Error(`AxFunction not found: ` + func.name);
+        }
+        if (!fnSpec.func) {
+            throw new Error('No handler for function: ' + func.name);
+        }
+        // execute value function calls
+        const funcExec = await this.executeFunction(fnSpec, func, options);
+        // // signal error if no data returned
+        // if (!funcExec.result || funcExec.result.length === 0) {
+        //   funcExec.result = `No data returned by function`;
+        // }
+        return funcExec;
+    };
+}
+//# sourceMappingURL=functions.js.map

package/build/module/src/funcs/functions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"functions.js","sourceRoot":"","sources":["../../../../src/funcs/functions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAI1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAgBrD,MAAM,OAAO,mBAAmB;IACtB,QAAQ,CAAwB;IAExC,YAAY,QAA+B;QACzC,QAAQ;aACL,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;aAC3B,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC,UAAW,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAEO,eAAe,GAAG,KAAK,EAC7B,MAA4B,EAC5B,IAA0C,EAC1C,OAA4C,EACnB,EAAE;QAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,MAAM,CAAC,IAAI,kBAAkB,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,MAAM,CAAC;QACX,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC5B,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE1B,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GACP,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;YACtB,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC;YACpC,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;SACrC,CAAC;IACJ,CAAC,CAAC;IAEK,OAAO,GAAG,KAAK,EACpB,IAA0C,EAC1C,OAA4C,EACnB,EAAE;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAC/B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAEnE,sCAAsC;QACtC,0DAA0D;QAC1D,sDAAsD;QACtD,IAAI;QACJ,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;CACH"}

package/build/module/src/funcs/index.d.ts

@@ -1,2 +1,3 @@
 export * from './code.js';
 export * from './embed.js';
+export * from './functions.js';

package/build/module/src/funcs/index.js

@@ -1,3 +1,4 @@
 export * from './code.js';
 export * from './embed.js';
+export * from './functions.js';
 //# sourceMappingURL=index.js.map

package/build/module/src/funcs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/funcs/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,YAAY,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/funcs/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,YAAY,CAAC;AAC3B,cAAc,gBAAgB,CAAC"}

package/build/module/src/funcs/jsonschema.d.ts

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

package/build/module/src/funcs/jsonschema.js

@@ -0,0 +1,97 @@
+// 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/build/module/src/funcs/jsonschema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"jsonschema.js","sourceRoot":"","sources":["../../../../src/funcs/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/build/module/src/index.d.ts

@@ -1,8 +1,7 @@
 export * from './ai/index.js';
-export * from './text/index.js';
+export * from './mem/index.js';
 export * from './prompts/index.js';
 export * from './funcs/index.js';
-export * from './types/index.js';
 export * from './db/index.js';
 export * from './dsp/index.js';
 export * from './docs/index.js';

package/build/module/src/index.js

@@ -1,8 +1,7 @@
 export * from './ai/index.js';
-export * from './text/index.js';
+export * from './mem/index.js';
 export * from './prompts/index.js';
 export * from './funcs/index.js';
-export * from './types/index.js';
 export * from './db/index.js';
 export * from './dsp/index.js';
 export * from './docs/index.js';

package/build/module/src/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,eAAe,CAAC;AAC9B,cAAc,iBAAiB,CAAC;AAChC,cAAc,oBAAoB,CAAC;AACnC,cAAc,kBAAkB,CAAC;AACjC,cAAc,kBAAkB,CAAC;AACjC,cAAc,eAAe,CAAC;AAC9B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AAEjC;;;;EAIE"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,eAAe,CAAC;AAC9B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,oBAAoB,CAAC;AACnC,cAAc,kBAAkB,CAAC;AACjC,cAAc,eAAe,CAAC;AAC9B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AAEjC;;;;EAIE"}

package/build/module/src/mem/index.d.ts

@@ -0,0 +1,2 @@
+export * from './memory.js';
+export * from './types.js';

package/build/module/src/mem/index.js

@@ -0,0 +1,3 @@
+export * from './memory.js';
+export * from './types.js';
+//# sourceMappingURL=index.js.map

package/build/module/src/mem/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/mem/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAC;AAC5B,cAAc,YAAY,CAAC"}

package/build/module/src/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/build/module/src/mem/memory.js

@@ -0,0 +1,69 @@
+export 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) {
+        this.add({ content, name, role: 'assistant', functionCalls }, sessionId);
+    }
+    updateResult({ content, name, functionCalls }, sessionId) {
+        const item = this.get(sessionId);
+        if ('content' in item && content) {
+            item.content = content;
+        }
+        if ('name' in item && name) {
+            item.name = name;
+        }
+        if ('functionCalls' in item && functionCalls) {
+            item.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) || [];
+    }
+}
+//# sourceMappingURL=memory.js.map

package/build/module/src/mem/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../../../src/mem/memory.ts"],"names":[],"mappings":"AAIA,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;QACV,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,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,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,SAAS,IAAI,IAAI,IAAI,OAAO,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QACD,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACnB,CAAC;QACD,IAAI,eAAe,IAAI,IAAI,IAAI,aAAa,EAAE,CAAC;YAC7C,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACrC,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/build/module/src/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/build/module/src/mem/types.js

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

package/build/module/src/mem/types.js.map

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

package/build/module/src/prompts/agent.d.ts

@@ -1,24 +1,23 @@
-import { type GenerateOptions, Signature } from '../dsp/index.js';
-import { type GenIn, type GenOut, type ITunable, type IUsable, Program, type ProgramForwardOptions } from '../dsp/program.js';
-import { type AITextFunction } from '../text/index.js';
-import type { AIService } from '../text/types.js';
-export interface AgentI extends ITunable, IUsable {
-    getFunction(): AITextFunction;
+import type { AxAIService, AxFunction } from '../ai/types.js';
+import { type AxGenerateOptions, AxSignature } from '../dsp/index.js';
+import { type AxGenIn, type AxGenOut, AxProgram, type AxProgramForwardOptions, type AxTunable, type AxUsable } from '../dsp/program.js';
+export interface AxAgentic extends AxTunable, AxUsable {
+    getFunction(): AxFunction;
 }
-export type AgentOptions = Omit<GenerateOptions, 'functions' | 'functionCall'>;
-export declare class Agent<IN extends GenIn, OUT extends GenOut> extends Program<IN, OUT> implements AgentI {
+export type AxAgentOptions = Omit<AxGenerateOptions, 'functions' | 'functionCall'>;
+export declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut> extends AxProgram<IN, OUT> implements AxAgentic {
     private gen;
     private name;
     private description;
     private subAgentList?;
     private func;
-    constructor(ai: AIService, { name, description, signature, agents, functions }: Readonly<{
+    constructor(ai: AxAIService, { name, description, signature, agents, functions }: Readonly<{
         name: string;
         description: string;
-        signature: Signature | string;
-        agents?: AgentI[];
-        functions?: AITextFunction[];
-    }>, options?: Readonly<AgentOptions>);
-    getFunction(): AITextFunction;
-    forward(values: IN, options?: Readonly<ProgramForwardOptions>): Promise<OUT>;
+        signature: AxSignature | string;
+        agents?: AxAgentic[];
+        functions?: AxFunction[];
+    }>, options?: Readonly<AxAgentOptions>);
+    getFunction(): AxFunction;
+    forward(values: IN, options?: Readonly<AxProgramForwardOptions>): Promise<OUT>;
 }

package/build/module/src/prompts/agent.js

@@ -1,10 +1,9 @@
-import { Signature } from '../dsp/index.js';
-import { Program } from '../dsp/program.js';
-import {} from '../text/index.js';
-import { SpanKind } from '../trace/index.js';
-import { ChainOfThought } from './cot.js';
-import { ReAct } from './react.js';
-export class Agent extends Program {
+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 extends AxProgram {
     gen;
     name;
     description;
@@ -12,7 +11,7 @@ export class Agent extends Program {
     func;
     constructor(ai, { name, description, signature, agents, functions }, options) {
         super();
-        const sig = new Signature(signature);
+        const sig = new AxSignature(signature);
         const funcs = [
             ...(functions ?? []),
             ...(agents?.map((a) => a.getFunction()) ?? [])
@@ -24,8 +23,8 @@ export class Agent extends Program {
         };
         this.gen =
             funcs.length > 0
-                ? new ReAct(ai, sig, opt)
-                : new ChainOfThought(ai, sig, opt);
+                ? new AxReAct(ai, sig, opt)
+                : new AxChainOfThought(ai, sig, opt);
         this.name = name;
         this.description = description;
         this.subAgentList = agents?.map((a) => a.getFunction().name).join(', ');
@@ -53,7 +52,7 @@ export class Agent extends Program {
             ['agent.subAgents']: this.subAgentList ?? 'none'
         };
         return await options?.tracer.startActiveSpan('Agent', {
-            kind: SpanKind.SERVER,
+            kind: AxSpanKind.SERVER,
             attributes
         }, async (span) => {
             const res = await this.gen.forward(values, options);

package/build/module/src/prompts/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../../src/prompts/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAwB,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAClE,OAAO,EAKL,OAAO,EAER,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAuB,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAQnC,MAAM,OAAO,KACX,SAAQ,OAAgB;IAGhB,GAAG,CAAmB;IAEtB,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,YAAY,CAAU;IACtB,IAAI,CAAiB;IAE7B,YACE,EAAa,EACb,EACE,IAAI,EACJ,WAAW,EACX,SAAS,EACT,MAAM,EACN,SAAS,EAOT,EACF,OAAgC;QAEhC,KAAK,EAAE,CAAC;QAER,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QAErC,MAAM,KAAK,GAAqB;YAC9B,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,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,OAAO,EAAE,OAAO,EAAE,OAAO;YACzB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,GAAG;YACN,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,KAAK,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC;gBAClC,CAAC,CAAC,IAAI,cAAc,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAEhD,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;QACxE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,GAAG,CAAC,YAAY,EAAE;YAC9B,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,MAAU,EACV,OAAyC;QAEzC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,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,QAAQ,CAAC,MAAM;YACrB,UAAU;SACX,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../../../src/prompts/agent.ts"],"names":[],"mappings":"AACA,OAAO,EAA0B,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACtE,OAAO,EAGL,SAAS,EAIV,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,OACX,SAAQ,SAAkB;IAGlB,GAAG,CAAqB;IAExB,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,KAAK,EAAE,CAAC;QAER,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;QAEvC,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,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,OAAO,EAAE,OAAO,EAAE,OAAO;YACzB,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,GAAG;YACN,KAAK,CAAC,MAAM,GAAG,CAAC;gBACd,CAAC,CAAC,IAAI,OAAO,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,gBAAgB,CAAU,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAElD,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;QACxE,IAAI,CAAC,IAAI,GAAG;YACV,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,GAAG,CAAC,YAAY,EAAE;YAC9B,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO;SACzB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEM,WAAW;QAChB,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAEe,KAAK,CAAC,OAAO,CAC3B,MAAU,EACV,OAA2C;QAE3C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACjD,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,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,GAAG,CAAC;QACb,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}

package/build/module/src/prompts/cot.d.ts

@@ -1,9 +1,9 @@
-import { Generate, type GenerateOptions } from '../dsp/generate.js';
-import type { GenIn, GenOut } from '../dsp/program.js';
-import { Signature } from '../dsp/sig.js';
-import type { AIService } from '../text/types.js';
-export declare class ChainOfThought<IN extends GenIn = GenIn, OUT extends GenOut = GenOut> extends Generate<IN, OUT & {
+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: AIService, signature: Readonly<Signature | string>, options?: Readonly<GenerateOptions>);
+    constructor(ai: AxAIService, signature: Readonly<AxSignature | string>, options?: Readonly<AxGenerateOptions>);
 }

package/build/module/src/prompts/cot.js

@@ -1,8 +1,8 @@
-import { Generate } from '../dsp/generate.js';
-import { Signature } from '../dsp/sig.js';
-export class ChainOfThought extends Generate {
+import { AxGenerate } from '../dsp/generate.js';
+import { AxSignature } from '../dsp/sig.js';
+export class AxChainOfThought extends AxGenerate {
     constructor(ai, signature, options) {
-        const sig = new Signature(signature);
+        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([
             {

package/build/module/src/prompts/cot.js.map

@@ -1 +1 @@
-{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../../src/prompts/cot.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAwB,MAAM,oBAAoB,CAAC;AAEpE,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG1C,MAAM,OAAO,cAGX,SAAQ,QAAsC;IAC9C,YACE,EAAa,EACb,SAAuC,EACvC,OAAmC;QAEnC,MAAM,GAAG,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,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"}
+{"version":3,"file":"cot.js","sourceRoot":"","sources":["../../../../src/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/build/module/src/prompts/prompts.test.js

@@ -1,7 +1,8 @@
 import { Readable } from 'stream';
 import test from 'ava';
-import { AI } from '../ai/index.js';
-import { ChainOfThought } from './cot.js';
+import { axAI } from '../ai/index.js';
+import { AxSignature } from '../dsp/sig.js';
+import { AxChainOfThought } from './cot.js';
 const text = `The technological singularity—or simply the singularity[1]—is a hypothetical future point in time at which technological growth becomes uncontrollable and irreversible.`;
 const examples = [
     {
@@ -35,9 +36,9 @@ const mockFetch = async () => {
 };
 test('generate prompt', async (t) => {
     const options = { fetch: mockFetch };
-    const ai = AI('openai', { apiKey: 'no-key', options });
+    const ai = axAI('openai', { apiKey: 'no-key', options });
     // const ai = AI('ollama', { model: 'nous-hermes2' });
-    const gen = new ChainOfThought(ai, `text -> shortSummary "summarize in 5 to 10 words"`);
+    const gen = new AxChainOfThought(ai, `text -> shortSummary "summarize in 5 to 10 words"`);
     gen.setExamples(examples);
     const res = await gen.forward({ text });
     t.deepEqual(res, {
@@ -45,4 +46,7 @@ test('generate prompt', async (t) => {
         shortSummary: 'More blah blah blah'
     });
 });
+test('generate prompt: invalid signature', async (t) => {
+    t.throws(() => new AxSignature(`text -> output:image`));
+});
 //# sourceMappingURL=prompts.test.js.map

package/build/module/src/prompts/prompts.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompts.test.js","sourceRoot":"","sources":["../../../../src/prompts/prompts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAElC,OAAO,IAAI,MAAM,KAAK,CAAC;AAEvB,OAAO,EAAE,EAAE,EAAmB,MAAM,gBAAgB,CAAC;AAErD,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,MAAM,IAAI,GAAG,0KAA0K,CAAC;AAExL,MAAM,QAAQ,GAAG;IACf;QACE,IAAI,EAAE,0TAA0T;QAChU,YAAY,EACV,+GAA+G;KAClH;IACD;QACE,IAAI,EAAE,gZAAgZ;QACtZ,YAAY,EACV,+HAA+H;KAClI;CACF,CAAC;AAEF,MAAM,SAAS,GAAG,KAAK,IAAuB,EAAE;IAC9C,MAAM,OAAO,GAAG;QACd,OAAO,EAAE;YACP;gBACE,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,4DAA4D;iBACtE;aACF;SACF;KACF,CAAC;IAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC;YACN,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IAClC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IACrC,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAgB,CAAC,CAAC;IAErE,sDAAsD;IAEtD,MAAM,GAAG,GAAG,IAAI,cAAc,CAC5B,EAAE,EACF,mDAAmD,CACpD,CAAC;IACF,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE1B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAExC,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE;QACf,MAAM,EAAE,gBAAgB;QACxB,YAAY,EAAE,qBAAqB;KACpC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"prompts.test.js","sourceRoot":"","sources":["../../../../src/prompts/prompts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAElC,OAAO,IAAI,MAAM,KAAK,CAAC;AAEvB,OAAO,EAAE,IAAI,EAAqB,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAE5C,MAAM,IAAI,GAAG,0KAA0K,CAAC;AAExL,MAAM,QAAQ,GAAG;IACf;QACE,IAAI,EAAE,0TAA0T;QAChU,YAAY,EACV,+GAA+G;KAClH;IACD;QACE,IAAI,EAAE,gZAAgZ;QACtZ,YAAY,EACV,+HAA+H;KAClI;CACF,CAAC;AAEF,MAAM,SAAS,GAAG,KAAK,IAAuB,EAAE;IAC9C,MAAM,OAAO,GAAG;QACd,OAAO,EAAE;YACP;gBACE,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,4DAA4D;iBACtE;aACF;SACF;KACF,CAAC;IAEF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,OAAO,CAAC;YACN,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IAClC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;IACrC,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAkB,CAAC,CAAC;IAEzE,sDAAsD;IAEtD,MAAM,GAAG,GAAG,IAAI,gBAAgB,CAC9B,EAAE,EACF,mDAAmD,CACpD,CAAC;IACF,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAE1B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;IAExC,CAAC,CAAC,SAAS,CAAC,GAAG,EAAE;QACf,MAAM,EAAE,gBAAgB;QACxB,YAAY,EAAE,qBAAqB;KACpC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,oCAAoC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;IACrD,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC"}

package/build/module/src/prompts/rag.d.ts

@@ -1,8 +1,8 @@
-import { type GenerateOptions } from '../dsp/index.js';
-import { type ProgramForwardOptions } from '../dsp/program.js';
-import type { AIService } from '../text/types.js';
-import { ChainOfThought } from './cot.js';
-export declare class RAG extends ChainOfThought<{
+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;
 }, {
@@ -11,12 +11,12 @@ export declare class RAG extends ChainOfThought<{
     private genQuery;
     private queryFn;
     private maxHops;
-    constructor(ai: AIService, queryFn: (query: string) => Promise<string>, options: Readonly<GenerateOptions & {
+    constructor(ai: AxAIService, queryFn: (query: string) => Promise<string>, options: Readonly<AxGenerateOptions & {
         maxHops?: number;
     }>);
     forward({ question }: Readonly<{
         question: string;
-    }>, options?: Readonly<ProgramForwardOptions>): Promise<{
+    }>, options?: Readonly<AxProgramForwardOptions>): Promise<{
         answer: string;
         reason: string;
     }>;

package/build/module/src/prompts/rag.js

@@ -1,7 +1,7 @@
-import { dedup, Generate, Signature } from '../dsp/index.js';
+import { AxGenerate, AxSignature, axStringUtil } from '../dsp/index.js';
 import {} from '../dsp/program.js';
-import { ChainOfThought } from './cot.js';
-export class RAG extends ChainOfThought {
+import { AxChainOfThought } from './cot.js';
+export class AxRAG extends AxChainOfThought {
     genQuery;
     queryFn;
     maxHops;
@@ -9,8 +9,8 @@ export class RAG extends ChainOfThought {
         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 Signature('"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 Generate(ai, qsig);
+        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);
     }
@@ -22,7 +22,7 @@ export class RAG extends ChainOfThought {
                 question
             }, options);
             const val = await this.queryFn(query);
-            context = dedup([...context, val]);
+            context = axStringUtil.dedup([...context, val]);
         }
         return super.forward({ context, question }, options);
     }

package/build/module/src/prompts/rag.js.map

@@ -1 +1 @@
-{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../../src/prompts/rag.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,QAAQ,EAER,SAAS,EACV,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAA8B,MAAM,mBAAmB,CAAC;AAG/D,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,MAAM,OAAO,GAAI,SAAQ,cAGxB;IACS,QAAQ,CAGd;IACM,OAAO,CAAqC;IAC5C,OAAO,CAAS;IAExB,YACE,EAAa,EACb,OAA2C,EAC3C,OAAyD;QAEzD,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,SAAS,CACxB,mLAAmL,CACpL,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAG1B,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,OAAyC;QAEzC,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,KAAK,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;CACF"}
+{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../../../src/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/build/module/src/prompts/react.d.ts

@@ -1,8 +1,8 @@
-import type { GenerateOptions } from '../dsp/generate.js';
-import type { GenIn, GenOut } from '../dsp/program.js';
-import { Signature } from '../dsp/sig.js';
-import type { AIService } from '../text/types.js';
-import { ChainOfThought } from './cot.js';
-export declare class ReAct<IN extends GenIn = GenIn, OUT extends GenOut = GenOut> extends ChainOfThought<IN, OUT> {
-    constructor(ai: AIService, signature: Readonly<Signature | string>, options: Readonly<GenerateOptions>);
+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>);
 }