@rudderjs/ai 0.0.1

package/dist/middleware.js

@@ -0,0 +1,72 @@
+/** Run onConfig hooks — piped (each transforms config, next sees result) */
+export function runOnConfig(middlewares, ctx, config, phase) {
+    let result = config;
+    for (const mw of middlewares) {
+        if (mw.onConfig) {
+            const transformed = mw.onConfig(ctx, result, phase);
+            if (transformed)
+                result = transformed;
+        }
+    }
+    return result;
+}
+/** Run onChunk hooks — piped (each can transform or drop by returning null) */
+export function runOnChunk(middlewares, ctx, chunk) {
+    let result = chunk;
+    for (const mw of middlewares) {
+        if (result === null)
+            break;
+        if (mw.onChunk) {
+            result = mw.onChunk(ctx, result);
+        }
+    }
+    return result;
+}
+/** Run onBeforeToolCall hooks — first non-void result wins */
+export async function runOnBeforeToolCall(middlewares, ctx, toolName, args) {
+    for (const mw of middlewares) {
+        if (mw.onBeforeToolCall) {
+            const result = await mw.onBeforeToolCall(ctx, toolName, args);
+            if (result !== undefined && result !== null)
+                return result;
+        }
+    }
+    return undefined;
+}
+/** Run a sequential hook (onStart, onIteration, onAfterToolCall, etc.) */
+export async function runSequential(middlewares, hook, ctx) {
+    for (const mw of middlewares) {
+        const fn = mw[hook];
+        if (fn)
+            await fn.call(mw, ctx);
+    }
+}
+/** Run onAfterToolCall hooks sequentially */
+export async function runOnAfterToolCall(middlewares, ctx, toolName, args, result) {
+    for (const mw of middlewares) {
+        if (mw.onAfterToolCall)
+            await mw.onAfterToolCall(ctx, toolName, args, result);
+    }
+}
+/** Run onUsage hooks sequentially */
+export async function runOnUsage(middlewares, ctx, usage) {
+    for (const mw of middlewares) {
+        if (mw.onUsage)
+            await mw.onUsage(ctx, usage);
+    }
+}
+/** Run onAbort hooks sequentially */
+export async function runOnAbort(middlewares, ctx, reason) {
+    for (const mw of middlewares) {
+        if (mw.onAbort)
+            await mw.onAbort(ctx, reason);
+    }
+}
+/** Run onError hooks sequentially */
+export async function runOnError(middlewares, ctx, error) {
+    for (const mw of middlewares) {
+        if (mw.onError)
+            await mw.onError(ctx, error);
+    }
+}
+//# sourceMappingURL=middleware.js.map

package/dist/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AASA,4EAA4E;AAC5E,MAAM,UAAU,WAAW,CACzB,WAA2B,EAC3B,GAAsB,EACtB,MAA8B,EAC9B,KAA6B;IAE7B,IAAI,MAAM,GAAG,MAAM,CAAA;IACnB,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC;YAChB,MAAM,WAAW,GAAG,EAAE,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,CAAA;YACnD,IAAI,WAAW;gBAAE,MAAM,GAAG,WAAW,CAAA;QACvC,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,+EAA+E;AAC/E,MAAM,UAAU,UAAU,CACxB,WAA2B,EAC3B,GAAsB,EACtB,KAAkB;IAElB,IAAI,MAAM,GAAuB,KAAK,CAAA;IACtC,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,MAAM,KAAK,IAAI;YAAE,MAAK;QAC1B,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;YACf,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;QAClC,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,8DAA8D;AAC9D,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,WAA2B,EAC3B,GAAsB,EACtB,QAAgB,EAChB,IAA6B;IAE7B,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,gBAAgB,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,gBAAgB,CAAC,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAA;YAC7D,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,IAAI;gBAAE,OAAO,MAAM,CAAA;QAC5D,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAA;AAClB,CAAC;AAED,0EAA0E;AAC1E,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,WAA2B,EAC3B,IAAoE,EACpE,GAAsB;IAEtB,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,CAAA;QACnB,IAAI,EAAE;YAAE,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAA;IAChC,CAAC;AACH,CAAC;AAED,6CAA6C;AAC7C,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,WAA2B,EAC3B,GAAsB,EACtB,QAAgB,EAChB,IAA6B,EAC7B,MAAe;IAEf,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,eAAe;YAAE,MAAM,EAAE,CAAC,eAAe,CAAC,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAA;IAC/E,CAAC;AACH,CAAC;AAED,qCAAqC;AACrC,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,WAA2B,EAC3B,GAAsB,EACtB,KAAiB;IAEjB,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,OAAO;YAAE,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;IAC9C,CAAC;AACH,CAAC;AAED,qCAAqC;AACrC,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,WAA2B,EAC3B,GAAsB,EACtB,MAAc;IAEd,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,OAAO;YAAE,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;IAC/C,CAAC;AACH,CAAC;AAED,qCAAqC;AACrC,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,WAA2B,EAC3B,GAAsB,EACtB,KAAc;IAEd,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,OAAO;YAAE,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;IAC9C,CAAC;AACH,CAAC"}

package/dist/output.d.ts

@@ -0,0 +1,22 @@
+import { z } from 'zod';
+export interface OutputWrapper<T = unknown> {
+    type: 'object' | 'array' | 'choice';
+    schema: z.ZodType<T>;
+    parse(text: string): T;
+    toSystemPrompt(): string;
+}
+export declare class Output {
+    /** Structured object output with Zod schema */
+    static object<T extends z.ZodRawShape>(opts: {
+        schema: z.ZodObject<T>;
+    }): OutputWrapper<z.infer<z.ZodObject<T>>>;
+    /** Structured array output — each element validated by schema */
+    static array<T extends z.ZodType>(opts: {
+        element: T;
+    }): OutputWrapper<z.infer<T>[]>;
+    /** Classification — pick one of the provided options */
+    static choice<T extends string>(opts: {
+        options: readonly [T, ...T[]];
+    }): OutputWrapper<T>;
+}
+//# sourceMappingURL=output.d.ts.map

package/dist/output.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"output.d.ts","sourceRoot":"","sources":["../src/output.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,MAAM,WAAW,aAAa,CAAC,CAAC,GAAG,OAAO;IACxC,IAAI,EAAE,QAAQ,GAAG,OAAO,GAAG,QAAQ,CAAA;IACnC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;IACpB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,CAAC,CAAA;IACtB,cAAc,IAAI,MAAM,CAAA;CACzB;AAED,qBAAa,MAAM;IACjB,+CAA+C;IAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,IAAI,EAAE;QAAE,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;KAAE,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAchH,iEAAiE;IACjE,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC,CAAA;KAAE,GAAG,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IAepF,wDAAwD;IACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,MAAM,EAAE,IAAI,EAAE;QAAE,OAAO,EAAE,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAA;KAAE,GAAG,aAAa,CAAC,CAAC,CAAC;CAa3F"}

package/dist/output.js

@@ -0,0 +1,55 @@
+import { z } from 'zod';
+import { zodToJsonSchema } from './zod-to-json-schema.js';
+export class Output {
+    /** Structured object output with Zod schema */
+    static object(opts) {
+        return {
+            type: 'object',
+            schema: opts.schema,
+            parse(text) {
+                const json = extractJson(text);
+                return opts.schema.parse(json);
+            },
+            toSystemPrompt() {
+                return `Respond with a JSON object matching this schema. Output ONLY valid JSON, no markdown fences.\n${JSON.stringify(zodToJsonSchema(opts.schema), null, 2)}`;
+            },
+        };
+    }
+    /** Structured array output — each element validated by schema */
+    static array(opts) {
+        const arraySchema = z.array(opts.element);
+        return {
+            type: 'array',
+            schema: arraySchema,
+            parse(text) {
+                const json = extractJson(text);
+                return arraySchema.parse(json);
+            },
+            toSystemPrompt() {
+                return `Respond with a JSON array where each element matches this schema. Output ONLY valid JSON.\n${JSON.stringify(zodToJsonSchema(opts.element), null, 2)}`;
+            },
+        };
+    }
+    /** Classification — pick one of the provided options */
+    static choice(opts) {
+        const enumSchema = z.enum(opts.options);
+        return {
+            type: 'choice',
+            schema: enumSchema,
+            parse(text) {
+                return enumSchema.parse(text.trim());
+            },
+            toSystemPrompt() {
+                return `Respond with exactly one of these options: ${opts.options.join(', ')}. Output ONLY the option, nothing else.`;
+            },
+        };
+    }
+}
+function extractJson(text) {
+    const stripped = text
+        .replace(/^```(?:json)?\s*\n?/m, '')
+        .replace(/\n?```\s*$/m, '')
+        .trim();
+    return JSON.parse(stripped);
+}
+//# sourceMappingURL=output.js.map

package/dist/output.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"output.js","sourceRoot":"","sources":["../src/output.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AASzD,MAAM,OAAO,MAAM;IACjB,+CAA+C;IAC/C,MAAM,CAAC,MAAM,CAA0B,IAAgC;QACrE,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,CAAC,IAAY;gBAChB,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAA;gBAC9B,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAChC,CAAC;YACD,cAAc;gBACZ,OAAO,iGAAiG,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAA;YACjK,CAAC;SACF,CAAA;IACH,CAAC;IAED,iEAAiE;IACjE,MAAM,CAAC,KAAK,CAAsB,IAAoB;QACpD,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACzC,OAAO;YACL,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,WAAsC;YAC9C,KAAK,CAAC,IAAY;gBAChB,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAA;gBAC9B,OAAO,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAChC,CAAC;YACD,cAAc;gBACZ,OAAO,8FAA8F,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAA;YAC/J,CAAC;SACF,CAAA;IACH,CAAC;IAED,wDAAwD;IACxD,MAAM,CAAC,MAAM,CAAmB,IAAuC;QACrE,MAAM,UAAU,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACvC,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,UAA0B;YAClC,KAAK,CAAC,IAAY;gBAChB,OAAO,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAM,CAAA;YAC3C,CAAC;YACD,cAAc;gBACZ,OAAO,8CAA8C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,yCAAyC,CAAA;YACvH,CAAC;SACF,CAAA;IACH,CAAC;CACF;AAED,SAAS,WAAW,CAAC,IAAY;IAC/B,MAAM,QAAQ,GAAG,IAAI;SAClB,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC;SACnC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;SAC1B,IAAI,EAAE,CAAA;IACT,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;AAC7B,CAAC"}

package/dist/provider-tools.d.ts

@@ -0,0 +1,60 @@
+/**
+ * Web search tool — uses provider-native web search when available.
+ * Falls back to a server-side DuckDuckGo fetch for providers without native support.
+ */
+export declare class WebSearch {
+    private _domains;
+    private _maxResults;
+    private constructor();
+    static make(): WebSearch;
+    domains(domains: string[]): this;
+    maxResults(n: number): this;
+    /** Convert to a tool definition that can be added to an agent's tools array */
+    toTool(): import("./tool.js").ServerToolBuilder<{
+        query: string;
+    }, {
+        results: string;
+        error?: never;
+    } | {
+        error: string;
+        results?: never;
+    }>;
+}
+/**
+ * Web fetch tool — fetches a web page and returns its text content.
+ */
+export declare class WebFetch {
+    private _maxLength;
+    private constructor();
+    static make(): WebFetch;
+    maxLength(n: number): this;
+    toTool(): import("./tool.js").ServerToolBuilder<{
+        url: string;
+    }, {
+        content: string;
+        url: string;
+        status: number;
+        error?: never;
+    } | {
+        error: string;
+        url: string;
+        content?: never;
+        status?: never;
+    }>;
+}
+/**
+ * Code execution tool — uses provider-native code interpreter when available.
+ * No server-side fallback for security — returns an error if the provider
+ * doesn't support native code execution.
+ */
+export declare class CodeExecution {
+    private constructor();
+    static make(): CodeExecution;
+    toTool(): import("./tool.js").ServerToolBuilder<{
+        code: string;
+        language: string;
+    }, {
+        error: string;
+    }>;
+}
+//# sourceMappingURL=provider-tools.d.ts.map

package/dist/provider-tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-tools.d.ts","sourceRoot":"","sources":["../src/provider-tools.ts"],"names":[],"mappings":"AAGA;;;GAGG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,QAAQ,CAAsB;IACtC,OAAO,CAAC,WAAW,CAAoB;IAEvC,OAAO;IAEP,MAAM,CAAC,IAAI,IAAI,SAAS;IAIxB,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI;IAKhC,UAAU,CAAC,CAAC,EAAE,MAAM,GAAG,IAAI;IAK3B,+EAA+E;IAC/E,MAAM;;;;;;;;;CA+BP;AAED;;GAEG;AACH,qBAAa,QAAQ;IACnB,OAAO,CAAC,UAAU,CAAoB;IAEtC,OAAO;IAEP,MAAM,CAAC,IAAI,IAAI,QAAQ;IAIvB,SAAS,CAAC,CAAC,EAAE,MAAM,GAAG,IAAI;IAK1B,MAAM;;;;;;;;;;;;;CAkCP;AAED;;;;GAIG;AACH,qBAAa,aAAa;IACxB,OAAO;IAEP,MAAM,CAAC,IAAI,IAAI,aAAa;IAI5B,MAAM;;;;;;CAmBP"}

package/dist/provider-tools.js

@@ -0,0 +1,133 @@
+import { z } from 'zod';
+import { toolDefinition } from './tool.js';
+/**
+ * Web search tool — uses provider-native web search when available.
+ * Falls back to a server-side DuckDuckGo fetch for providers without native support.
+ */
+export class WebSearch {
+    _domains;
+    _maxResults;
+    constructor() { }
+    static make() {
+        return new WebSearch();
+    }
+    domains(domains) {
+        this._domains = domains;
+        return this;
+    }
+    maxResults(n) {
+        this._maxResults = n;
+        return this;
+    }
+    /** Convert to a tool definition that can be added to an agent's tools array */
+    toTool() {
+        const domains = this._domains;
+        const maxResults = this._maxResults;
+        return toolDefinition({
+            name: 'web_search',
+            description: 'Search the web for current information.',
+            inputSchema: z.object({
+                query: z.string().describe('The search query'),
+            }),
+            meta: {
+                providerNative: true,
+                type: 'web_search',
+                domains,
+                maxResults,
+            },
+        }).server(async ({ query }) => {
+            // Fallback: simple server-side search via fetch
+            // Runs only if the provider doesn't handle it natively
+            try {
+                const url = new URL('https://html.duckduckgo.com/html/');
+                url.searchParams.set('q', query + (domains ? ` site:${domains.join(' OR site:')}` : ''));
+                const res = await fetch(url.toString(), { headers: { 'User-Agent': 'RudderJS/1.0' } });
+                const html = await res.text();
+                const text = html.replace(/<[^>]*>/g, ' ').replace(/\s+/g, ' ').trim().slice(0, 2000);
+                return { results: text };
+            }
+            catch {
+                return { error: 'Web search unavailable' };
+            }
+        });
+    }
+}
+/**
+ * Web fetch tool — fetches a web page and returns its text content.
+ */
+export class WebFetch {
+    _maxLength;
+    constructor() { }
+    static make() {
+        return new WebFetch();
+    }
+    maxLength(n) {
+        this._maxLength = n;
+        return this;
+    }
+    toTool() {
+        const maxLength = this._maxLength ?? 10000;
+        return toolDefinition({
+            name: 'web_fetch',
+            description: 'Fetch a web page and return its text content.',
+            inputSchema: z.object({
+                url: z.string().url().describe('The URL to fetch'),
+            }),
+            meta: {
+                providerNative: true,
+                type: 'web_fetch',
+                maxLength,
+            },
+        }).server(async ({ url: targetUrl }) => {
+            try {
+                const res = await fetch(targetUrl, {
+                    headers: { 'User-Agent': 'RudderJS/1.0' },
+                    signal: AbortSignal.timeout(10000),
+                });
+                const html = await res.text();
+                const text = html
+                    .replace(/<script[^>]*>[\s\S]*?<\/script>/gi, '')
+                    .replace(/<style[^>]*>[\s\S]*?<\/style>/gi, '')
+                    .replace(/<[^>]*>/g, ' ')
+                    .replace(/\s+/g, ' ')
+                    .trim()
+                    .slice(0, maxLength);
+                return { content: text, url: targetUrl, status: res.status };
+            }
+            catch (err) {
+                return { error: err instanceof Error ? err.message : 'Fetch failed', url: targetUrl };
+            }
+        });
+    }
+}
+/**
+ * Code execution tool — uses provider-native code interpreter when available.
+ * No server-side fallback for security — returns an error if the provider
+ * doesn't support native code execution.
+ */
+export class CodeExecution {
+    constructor() { }
+    static make() {
+        return new CodeExecution();
+    }
+    toTool() {
+        return toolDefinition({
+            name: 'code_execution',
+            description: 'Execute code to perform calculations, data analysis, or generate outputs.',
+            inputSchema: z.object({
+                code: z.string().describe('The code to execute'),
+                language: z.string().default('javascript').describe('Programming language'),
+            }),
+            meta: {
+                providerNative: true,
+                type: 'code_execution',
+            },
+        }).server(async ({ code, language }) => {
+            // No server-side fallback for security — must be handled by provider
+            return {
+                error: `Code execution requires a provider with native code interpreter support. Language: ${language}, code length: ${code.length}`,
+            };
+        });
+    }
+}
+//# sourceMappingURL=provider-tools.js.map

package/dist/provider-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-tools.js","sourceRoot":"","sources":["../src/provider-tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAE1C;;;GAGG;AACH,MAAM,OAAO,SAAS;IACZ,QAAQ,CAAsB;IAC9B,WAAW,CAAoB;IAEvC,gBAAuB,CAAC;IAExB,MAAM,CAAC,IAAI;QACT,OAAO,IAAI,SAAS,EAAE,CAAA;IACxB,CAAC;IAED,OAAO,CAAC,OAAiB;QACvB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAA;QACvB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,UAAU,CAAC,CAAS;QAClB,IAAI,CAAC,WAAW,GAAG,CAAC,CAAA;QACpB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,+EAA+E;IAC/E,MAAM;QACJ,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAA;QAEnC,OAAO,cAAc,CAAC;YACpB,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,yCAAyC;YACtD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;aAC/C,CAAC;YACF,IAAI,EAAE;gBACJ,cAAc,EAAE,IAAI;gBACpB,IAAI,EAAE,YAAY;gBAClB,OAAO;gBACP,UAAU;aACX;SACF,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;YAC5B,gDAAgD;YAChD,uDAAuD;YACvD,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,mCAAmC,CAAC,CAAA;gBACxD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;gBACxF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,cAAc,EAAE,EAAE,CAAC,CAAA;gBACtF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;gBAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;gBACrF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAA;YAC1B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAA;YAC5C,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,QAAQ;IACX,UAAU,CAAoB;IAEtC,gBAAuB,CAAC;IAExB,MAAM,CAAC,IAAI;QACT,OAAO,IAAI,QAAQ,EAAE,CAAA;IACvB,CAAC;IAED,SAAS,CAAC,CAAS;QACjB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAA;QACnB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM;QACJ,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,IAAI,KAAK,CAAA;QAE1C,OAAO,cAAc,CAAC;YACpB,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,+CAA+C;YAC5D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;aACnD,CAAC;YACF,IAAI,EAAE;gBACJ,cAAc,EAAE,IAAI;gBACpB,IAAI,EAAE,WAAW;gBACjB,SAAS;aACV;SACF,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE;YACrC,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE;oBACjC,OAAO,EAAE,EAAE,YAAY,EAAE,cAAc,EAAE;oBACzC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;iBACnC,CAAC,CAAA;gBACF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;gBAC7B,MAAM,IAAI,GAAG,IAAI;qBACd,OAAO,CAAC,mCAAmC,EAAE,EAAE,CAAC;qBAChD,OAAO,CAAC,iCAAiC,EAAE,EAAE,CAAC;qBAC9C,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC;qBACxB,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;qBACpB,IAAI,EAAE;qBACN,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAA;gBACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAA;YAC9D,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,EAAE,GAAG,EAAE,SAAS,EAAE,CAAA;YACvF,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AAED;;;;GAIG;AACH,MAAM,OAAO,aAAa;IACxB,gBAAuB,CAAC;IAExB,MAAM,CAAC,IAAI;QACT,OAAO,IAAI,aAAa,EAAE,CAAA;IAC5B,CAAC;IAED,MAAM;QACJ,OAAO,cAAc,CAAC;YACpB,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,2EAA2E;YACxF,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;gBAChD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,QAAQ,CAAC,sBAAsB,CAAC;aAC5E,CAAC;YACF,IAAI,EAAE;gBACJ,cAAc,EAAE,IAAI;gBACpB,IAAI,EAAE,gBAAgB;aACvB;SACF,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;YACrC,qEAAqE;YACrE,OAAO;gBACL,KAAK,EAAE,sFAAsF,QAAQ,kBAAkB,IAAI,CAAC,MAAM,EAAE;aACrI,CAAA;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;CACF"}

package/dist/provider.d.ts

@@ -0,0 +1,12 @@
+import { ServiceProvider, type Application } from '@rudderjs/core';
+import type { AiConfig } from './types.js';
+/**
+ * Create the AI ServiceProvider from config.
+ *
+ * @example
+ * // bootstrap/providers.ts
+ * import { ai } from '@rudderjs/ai'
+ * export default [ai(configs.ai), ...]
+ */
+export declare function ai(config: AiConfig): new (app: Application) => ServiceProvider;
+//# sourceMappingURL=provider.d.ts.map

package/dist/provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../src/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,KAAK,WAAW,EAAE,MAAM,gBAAgB,CAAA;AAGlE,OAAO,KAAK,EAAE,QAAQ,EAAqB,MAAM,YAAY,CAAA;AAE7D;;;;;;;GAOG;AACH,wBAAgB,EAAE,CAAC,MAAM,EAAE,QAAQ,GAAG,KAAK,GAAG,EAAE,WAAW,KAAK,eAAe,CA8E9E"}

package/dist/provider.js

@@ -0,0 +1,94 @@
+import { ServiceProvider } from '@rudderjs/core';
+import { AiRegistry } from './registry.js';
+import { setConversationStore } from './agent.js';
+/**
+ * Create the AI ServiceProvider from config.
+ *
+ * @example
+ * // bootstrap/providers.ts
+ * import { ai } from '@rudderjs/ai'
+ * export default [ai(configs.ai), ...]
+ */
+export function ai(config) {
+    class AiServiceProvider extends ServiceProvider {
+        register() { }
+        async boot() {
+            for (const [name, providerConfig] of Object.entries(config.providers)) {
+                const driver = providerConfig.driver ?? name;
+                if (driver === 'anthropic') {
+                    const { AnthropicProvider } = await import('./providers/anthropic.js');
+                    AiRegistry.register(new AnthropicProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+                else if (driver === 'openai') {
+                    const { OpenAIProvider } = await import('./providers/openai.js');
+                    AiRegistry.register(new OpenAIProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                        organization: providerConfig['organization'],
+                    }));
+                }
+                else if (driver === 'google') {
+                    const { GoogleProvider } = await import('./providers/google.js');
+                    AiRegistry.register(new GoogleProvider({
+                        apiKey: providerConfig.apiKey,
+                    }));
+                }
+                else if (driver === 'ollama') {
+                    const { OllamaProvider } = await import('./providers/ollama.js');
+                    AiRegistry.register(new OllamaProvider({
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+                else if (driver === 'deepseek') {
+                    const { DeepSeekProvider } = await import('./providers/deepseek.js');
+                    AiRegistry.register(new DeepSeekProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+                else if (driver === 'xai') {
+                    const { XaiProvider } = await import('./providers/xai.js');
+                    AiRegistry.register(new XaiProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+                else if (driver === 'groq') {
+                    const { GroqProvider } = await import('./providers/groq.js');
+                    AiRegistry.register(new GroqProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+                else if (driver === 'mistral') {
+                    const { MistralProvider } = await import('./providers/mistral.js');
+                    AiRegistry.register(new MistralProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+                else if (driver === 'azure') {
+                    const { AzureOpenAIProvider } = await import('./providers/azure.js');
+                    AiRegistry.register(new AzureOpenAIProvider({
+                        apiKey: providerConfig.apiKey,
+                        baseUrl: providerConfig.baseUrl,
+                    }));
+                }
+            }
+            AiRegistry.setDefault(config.default);
+            AiRegistry.setModels(config.models ?? []);
+            this.app.instance('ai.registry', AiRegistry);
+            // Register conversation store if provided in config
+            if (config.conversations) {
+                const store = config.conversations;
+                setConversationStore(store);
+                this.app.instance('ai.conversations', store);
+            }
+        }
+    }
+    return AiServiceProvider;
+}
+//# sourceMappingURL=provider.js.map

package/dist/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../src/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAoB,MAAM,gBAAgB,CAAA;AAClE,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAC1C,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AAGjD;;;;;;;GAOG;AACH,MAAM,UAAU,EAAE,CAAC,MAAgB;IACjC,MAAM,iBAAkB,SAAQ,eAAe;QAC7C,QAAQ,KAAU,CAAC;QAEnB,KAAK,CAAC,IAAI;YACR,KAAK,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBACtE,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,IAAI,CAAA;gBAE5C,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;oBAC3B,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAA;oBACtE,UAAU,CAAC,QAAQ,CAAC,IAAI,iBAAiB,CAAC;wBACxC,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;qBAChC,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAA;oBAChE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;wBACrC,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;wBAC/B,YAAY,EAAE,cAAc,CAAC,cAAc,CAAuB;qBACnE,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAA;oBAChE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;wBACrC,MAAM,EAAE,cAAc,CAAC,MAAO;qBAC/B,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAA;oBAChE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;wBACrC,OAAO,EAAE,cAAc,CAAC,OAAO;qBAChC,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;oBACjC,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAA;oBACpE,UAAU,CAAC,QAAQ,CAAC,IAAI,gBAAgB,CAAC;wBACvC,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;qBAChC,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;oBAC5B,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAA;oBAC1D,UAAU,CAAC,QAAQ,CAAC,IAAI,WAAW,CAAC;wBAClC,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;qBAChC,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;oBAC7B,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,CAAA;oBAC5D,UAAU,CAAC,QAAQ,CAAC,IAAI,YAAY,CAAC;wBACnC,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;qBAChC,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oBAChC,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;oBAClE,UAAU,CAAC,QAAQ,CAAC,IAAI,eAAe,CAAC;wBACtC,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;qBAChC,CAAC,CAAC,CAAA;gBACL,CAAC;qBAAM,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;oBAC9B,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAA;oBACpE,UAAU,CAAC,QAAQ,CAAC,IAAI,mBAAmB,CAAC;wBAC1C,MAAM,EAAE,cAAc,CAAC,MAAO;wBAC9B,OAAO,EAAE,cAAc,CAAC,OAAQ;qBACjC,CAAC,CAAC,CAAA;gBACL,CAAC;YACH,CAAC;YAED,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YACrC,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAA;YACzC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;YAE5C,oDAAoD;YACpD,IAAK,MAA2D,CAAC,aAAa,EAAE,CAAC;gBAC/E,MAAM,KAAK,GAAI,MAA2D,CAAC,aAAc,CAAA;gBACzF,oBAAoB,CAAC,KAAK,CAAC,CAAA;gBAC3B,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAA;YAC9C,CAAC;QACH,CAAC;KACF;IAED,OAAO,iBAAiB,CAAA;AAC1B,CAAC"}

package/dist/providers/anthropic.d.ts

@@ -0,0 +1,12 @@
+import type { ProviderFactory, ProviderAdapter } from '../types.js';
+export interface AnthropicConfig {
+    apiKey: string;
+    baseUrl?: string | undefined;
+}
+export declare class AnthropicProvider implements ProviderFactory {
+    readonly name = "anthropic";
+    private readonly config;
+    constructor(config: AnthropicConfig);
+    create(model: string): ProviderAdapter;
+}
+//# sourceMappingURL=anthropic.d.ts.map

package/dist/providers/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,eAAe,EAQhB,MAAM,aAAa,CAAA;AAEpB,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;CAC7B;AAED,qBAAa,iBAAkB,YAAW,eAAe;IACvD,QAAQ,CAAC,IAAI,eAAc;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAiB;gBAE5B,MAAM,EAAE,eAAe;IAInC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,eAAe;CAGvC"}