@rudderjs/ai 0.0.1

package/dist/facade.js

@@ -0,0 +1,100 @@
+import { AiRegistry } from './registry.js';
+import { agent as agentHelper } from './agent.js';
+import { ImageGenerator } from './image.js';
+import { AudioGenerator } from './audio.js';
+import { Transcription } from './transcription.js';
+import { CachedEmbeddingAdapter } from './cached-embedding.js';
+/**
+ * AI facade — static entry point for quick prompts, embeddings, and image generation.
+ *
+ * @example
+ * const response = await AI.prompt('Hello')
+ * const a = AI.agent('You are helpful.')
+ * const result = await AI.embed('Some text')
+ * const image = await AI.image('A sunset').generate()
+ */
+export class AI {
+    /** Quick prompt with default model */
+    static async prompt(input, options) {
+        const opts = {
+            instructions: 'You are a helpful assistant.',
+        };
+        if (options?.model)
+            opts.model = options.model;
+        return agentHelper(opts).prompt(input);
+    }
+    /** Create an anonymous agent */
+    static agent(instructionsOrOptions) {
+        return agentHelper(instructionsOrOptions);
+    }
+    /** Create an image generator with a fluent API */
+    static image(prompt) {
+        return ImageGenerator.of(prompt);
+    }
+    /** Create a text-to-speech audio generator */
+    static audio(text) {
+        return AudioGenerator.of(text);
+    }
+    /** Create a speech-to-text transcription */
+    static transcribe(pathOrBuffer) {
+        return typeof pathOrBuffer === 'string'
+            ? Transcription.fromPath(pathOrBuffer)
+            : Transcription.fromBuffer(pathOrBuffer);
+    }
+    /**
+     * Generate embeddings for text.
+     *
+     * Large arrays (100+ items) are automatically chunked into batches.
+     * Pass `cache: true` to enable in-memory caching of embeddings.
+     *
+     * @example
+     * const result = await AI.embed('Hello world')
+     * const result = await AI.embed(['text1', 'text2'])
+     * const result = await AI.embed('text', { model: 'openai/text-embedding-3-small' })
+     * const result = await AI.embed('text', { cache: true })
+     */
+    static async embed(input, options) {
+        const modelString = options?.model ?? AiRegistry.getDefault();
+        const [providerName, modelId] = AiRegistry.parseModelString(modelString);
+        const factory = AiRegistry.getFactory(providerName);
+        if (!factory.createEmbedding) {
+            throw new Error(`[RudderJS AI] Provider "${providerName}" does not support embeddings. ` +
+                `Use a provider that implements createEmbedding() (e.g. openai, google, mistral).`);
+        }
+        let adapter = factory.createEmbedding(modelId);
+        if (options?.cache) {
+            adapter = AI.getCachedAdapter(adapter);
+        }
+        // Batch chunking for large arrays
+        const inputs = Array.isArray(input) ? input : [input];
+        if (inputs.length > 100) {
+            const batches = [];
+            for (let i = 0; i < inputs.length; i += 100) {
+                batches.push(inputs.slice(i, i + 100));
+            }
+            const results = [];
+            for (const batch of batches) {
+                results.push(await adapter.embed(batch, modelId));
+            }
+            return {
+                embeddings: results.flatMap(r => r.embeddings),
+                usage: {
+                    promptTokens: results.reduce((sum, r) => sum + r.usage.promptTokens, 0),
+                    totalTokens: results.reduce((sum, r) => sum + r.usage.totalTokens, 0),
+                },
+            };
+        }
+        return adapter.embed(inputs.length === 1 ? inputs[0] : inputs, modelId);
+    }
+    /** Cache adapter instances so the in-memory cache persists across calls. */
+    static cachedAdapters = new WeakMap();
+    static getCachedAdapter(inner) {
+        let cached = AI.cachedAdapters.get(inner);
+        if (!cached) {
+            cached = new CachedEmbeddingAdapter(inner);
+            AI.cachedAdapters.set(inner, cached);
+        }
+        return cached;
+    }
+}
+//# sourceMappingURL=facade.js.map

package/dist/facade.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"facade.js","sourceRoot":"","sources":["../src/facade.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAC1C,OAAO,EAAE,KAAK,IAAI,WAAW,EAAE,MAAM,YAAY,CAAA;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAA;AAI9D;;;;;;;;GAQG;AACH,MAAM,OAAO,EAAE;IACb,sCAAsC;IACtC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAwC;QACzE,MAAM,IAAI,GAAyD;YACjE,YAAY,EAAE,8BAA8B;SAC7C,CAAA;QACD,IAAI,OAAO,EAAE,KAAK;YAAE,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAA;QAC9C,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACxC,CAAC;IAED,gCAAgC;IAChC,MAAM,CAAC,KAAK,CACV,qBAKC;QAED,OAAO,WAAW,CAAC,qBAAqB,CAAC,CAAA;IAC3C,CAAC;IAED,kDAAkD;IAClD,MAAM,CAAC,KAAK,CAAC,MAAc;QACzB,OAAO,cAAc,CAAC,EAAE,CAAC,MAAM,CAAC,CAAA;IAClC,CAAC;IAED,8CAA8C;IAC9C,MAAM,CAAC,KAAK,CAAC,IAAY;QACvB,OAAO,cAAc,CAAC,EAAE,CAAC,IAAI,CAAC,CAAA;IAChC,CAAC;IAED,4CAA4C;IAC5C,MAAM,CAAC,UAAU,CAAC,YAA6B;QAC7C,OAAO,OAAO,YAAY,KAAK,QAAQ;YACrC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC;YACtC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,YAAY,CAAC,CAAA;IAC5C,CAAC;IAED;;;;;;;;;;;OAWG;IACH,MAAM,CAAC,KAAK,CAAC,KAAK,CAChB,KAAwB,EACxB,OAAqE;QAErE,MAAM,WAAW,GAAG,OAAO,EAAE,KAAK,IAAI,UAAU,CAAC,UAAU,EAAE,CAAA;QAC7D,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC,GAAG,UAAU,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAA;QACxE,MAAM,OAAO,GAAG,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,CAAA;QAEnD,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CACb,2BAA2B,YAAY,iCAAiC;gBACxE,kFAAkF,CACnF,CAAA;QACH,CAAC;QAED,IAAI,OAAO,GAAqB,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;QAEhE,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,OAAO,GAAG,EAAE,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAA;QACxC,CAAC;QAED,kCAAkC;QAClC,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAA;QACrD,IAAI,MAAM,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YACxB,MAAM,OAAO,GAAe,EAAE,CAAA;YAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC;gBAC5C,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAA;YACxC,CAAC;YACD,MAAM,OAAO,GAAsB,EAAE,CAAA;YACrC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,OAAO,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAA;YACnD,CAAC;YACD,OAAO;gBACL,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC9C,KAAK,EAAE;oBACL,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;oBACvE,WAAW,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;iBACtE;aACF,CAAA;QACH,CAAC;QAED,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC1E,CAAC;IAED,4EAA4E;IACpE,MAAM,CAAC,cAAc,GAAG,IAAI,OAAO,EAA4C,CAAA;IAE/E,MAAM,CAAC,gBAAgB,CAAC,KAAuB;QACrD,IAAI,MAAM,GAAG,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACzC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,IAAI,sBAAsB,CAAC,KAAK,CAAC,CAAA;YAC1C,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;QACtC,CAAC;QACD,OAAO,MAAM,CAAA;IACf,CAAC"}

package/dist/fake.d.ts

@@ -0,0 +1,55 @@
+import type { ProviderRequestOptions, ImageGenerationOptions, TextToSpeechOptions, SpeechToTextOptions } from './types.js';
+/**
+ * Testing fake for @rudderjs/ai.
+ *
+ * @example
+ * const fake = AiFake.fake()
+ * fake.respondWith('Mocked response')
+ *
+ * const response = await AI.prompt('Hello')
+ * assert.strictEqual(response.text, 'Mocked response')
+ *
+ * fake.assertPrompted(input => input.includes('Hello'))
+ * fake.restore()
+ */
+export declare class AiFake {
+    private readonly calls;
+    private readonly imageCalls;
+    private readonly ttsCalls;
+    private readonly sttCalls;
+    private _response;
+    private _imageResponse;
+    private _ttsResponse;
+    private _sttResponse;
+    /** Set the text response the fake will return */
+    respondWith(text: string): void;
+    /** Set the base64 image the fake will return */
+    respondWithImage(base64: string): void;
+    /** Set the audio buffer the TTS fake will return */
+    respondWithAudio(audio: Buffer): void;
+    /** Set the transcription text the STT fake will return */
+    respondWithTranscription(text: string): void;
+    /** Install the fake — replaces all registered providers with a mock */
+    static fake(): AiFake;
+    /** Assert at least one prompt was sent, optionally matching a predicate */
+    assertPrompted(predicate?: (input: string) => boolean): void;
+    /** Assert no prompts were sent */
+    assertNothingPrompted(): void;
+    /** Assert at least one image generation was made */
+    assertImageGenerated(predicate?: (prompt: string) => boolean): void;
+    /** Assert at least one TTS generation was made */
+    assertAudioGenerated(predicate?: (text: string) => boolean): void;
+    /** Assert at least one transcription was made */
+    assertTranscribed(predicate?: (opts: SpeechToTextOptions) => boolean): void;
+    /** Get all recorded provider calls */
+    getCalls(): ProviderRequestOptions[];
+    /** Get all recorded image generation calls */
+    getImageCalls(): ImageGenerationOptions[];
+    /** Get all recorded TTS calls */
+    getTtsCalls(): TextToSpeechOptions[];
+    /** Get all recorded STT calls */
+    getSttCalls(): SpeechToTextOptions[];
+    /** Restore — clears the fake (user must re-register real providers) */
+    restore(): void;
+}
+//# sourceMappingURL=fake.d.ts.map

package/dist/fake.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fake.d.ts","sourceRoot":"","sources":["../src/fake.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAGV,sBAAsB,EAItB,sBAAsB,EAGtB,mBAAmB,EAGnB,mBAAmB,EAEpB,MAAM,YAAY,CAAA;AAEnB;;;;;;;;;;;;GAYG;AACH,qBAAa,MAAM;IACjB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAA+B;IACrD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA+B;IAC1D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA4B;IACrD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA4B;IACrD,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,cAAc,CAAqB;IAC3C,OAAO,CAAC,YAAY,CAAoC;IACxD,OAAO,CAAC,YAAY,CAAuB;IAE3C,iDAAiD;IACjD,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAI/B,gDAAgD;IAChD,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAItC,oDAAoD;IACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAIrC,0DAA0D;IAC1D,wBAAwB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAI5C,uEAAuE;IACvE,MAAM,CAAC,IAAI,IAAI,MAAM;IAiErB,2EAA2E;IAC3E,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,IAAI;IAa5D,kCAAkC;IAClC,qBAAqB,IAAI,IAAI;IAM7B,oDAAoD;IACpD,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,GAAG,IAAI;IAQnE,kDAAkD;IAClD,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,GAAG,IAAI;IAQjE,iDAAiD;IACjD,iBAAiB,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,mBAAmB,KAAK,OAAO,GAAG,IAAI;IAQ3E,sCAAsC;IACtC,QAAQ,IAAI,sBAAsB,EAAE;IAIpC,8CAA8C;IAC9C,aAAa,IAAI,sBAAsB,EAAE;IAIzC,iCAAiC;IACjC,WAAW,IAAI,mBAAmB,EAAE;IAIpC,iCAAiC;IACjC,WAAW,IAAI,mBAAmB,EAAE;IAIpC,uEAAuE;IACvE,OAAO,IAAI,IAAI;CAGhB"}

package/dist/fake.js

@@ -0,0 +1,172 @@
+import { AiRegistry } from './registry.js';
+/**
+ * Testing fake for @rudderjs/ai.
+ *
+ * @example
+ * const fake = AiFake.fake()
+ * fake.respondWith('Mocked response')
+ *
+ * const response = await AI.prompt('Hello')
+ * assert.strictEqual(response.text, 'Mocked response')
+ *
+ * fake.assertPrompted(input => input.includes('Hello'))
+ * fake.restore()
+ */
+export class AiFake {
+    calls = [];
+    imageCalls = [];
+    ttsCalls = [];
+    sttCalls = [];
+    _response = 'fake response';
+    _imageResponse = 'ZmFrZS1pbWFnZQ=='; // base64 of 'fake-image'
+    _ttsResponse = Buffer.from('fake-audio');
+    _sttResponse = 'fake transcription';
+    /** Set the text response the fake will return */
+    respondWith(text) {
+        this._response = text;
+    }
+    /** Set the base64 image the fake will return */
+    respondWithImage(base64) {
+        this._imageResponse = base64;
+    }
+    /** Set the audio buffer the TTS fake will return */
+    respondWithAudio(audio) {
+        this._ttsResponse = audio;
+    }
+    /** Set the transcription text the STT fake will return */
+    respondWithTranscription(text) {
+        this._sttResponse = text;
+    }
+    /** Install the fake — replaces all registered providers with a mock */
+    static fake() {
+        const fake = new AiFake();
+        const adapter = {
+            async generate(opts) {
+                fake.calls.push(opts);
+                return {
+                    message: { role: 'assistant', content: fake._response },
+                    usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
+                    finishReason: 'stop',
+                };
+            },
+            async *stream(opts) {
+                fake.calls.push(opts);
+                yield { type: 'text-delta', text: fake._response };
+                yield { type: 'finish', finishReason: 'stop', usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 } };
+            },
+        };
+        const imageAdapter = {
+            async generate(opts) {
+                fake.imageCalls.push(opts);
+                return {
+                    images: [{ base64: fake._imageResponse }],
+                    model: opts.model ?? 'fake-image',
+                };
+            },
+        };
+        const ttsAdapter = {
+            async generate(opts) {
+                fake.ttsCalls.push(opts);
+                return {
+                    audio: fake._ttsResponse,
+                    format: opts.format ?? 'mp3',
+                    model: opts.model ?? 'fake-tts',
+                };
+            },
+        };
+        const sttAdapter = {
+            async transcribe(opts) {
+                fake.sttCalls.push(opts);
+                return {
+                    text: fake._sttResponse,
+                    language: opts.language,
+                    model: opts.model ?? 'fake-stt',
+                };
+            },
+        };
+        const factory = {
+            name: '__fake__',
+            create: () => adapter,
+            createImage: () => imageAdapter,
+            createTts: () => ttsAdapter,
+            createStt: () => sttAdapter,
+        };
+        AiRegistry.reset();
+        AiRegistry.register(factory);
+        AiRegistry.setDefault('__fake__/default');
+        return fake;
+    }
+    /** Assert at least one prompt was sent, optionally matching a predicate */
+    assertPrompted(predicate) {
+        if (this.calls.length === 0)
+            throw new Error('[RudderJS AI] Expected at least one prompt, but none were sent.');
+        if (predicate) {
+            const match = this.calls.some(c => {
+                const userMsg = c.messages.find(m => m.role === 'user');
+                if (!userMsg)
+                    return false;
+                const text = typeof userMsg.content === 'string' ? userMsg.content : userMsg.content.filter(p => p.type === 'text').map(p => p.text).join('');
+                return predicate(text);
+            });
+            if (!match)
+                throw new Error('[RudderJS AI] No prompt matched the predicate.');
+        }
+    }
+    /** Assert no prompts were sent */
+    assertNothingPrompted() {
+        if (this.calls.length > 0) {
+            throw new Error(`[RudderJS AI] Expected no prompts, but ${this.calls.length} were sent.`);
+        }
+    }
+    /** Assert at least one image generation was made */
+    assertImageGenerated(predicate) {
+        if (this.imageCalls.length === 0)
+            throw new Error('[RudderJS AI] Expected at least one image generation, but none were made.');
+        if (predicate) {
+            const match = this.imageCalls.some(c => predicate(c.prompt));
+            if (!match)
+                throw new Error('[RudderJS AI] No image generation matched the predicate.');
+        }
+    }
+    /** Assert at least one TTS generation was made */
+    assertAudioGenerated(predicate) {
+        if (this.ttsCalls.length === 0)
+            throw new Error('[RudderJS AI] Expected at least one audio generation, but none were made.');
+        if (predicate) {
+            const match = this.ttsCalls.some(c => predicate(c.text));
+            if (!match)
+                throw new Error('[RudderJS AI] No audio generation matched the predicate.');
+        }
+    }
+    /** Assert at least one transcription was made */
+    assertTranscribed(predicate) {
+        if (this.sttCalls.length === 0)
+            throw new Error('[RudderJS AI] Expected at least one transcription, but none were made.');
+        if (predicate) {
+            const match = this.sttCalls.some(c => predicate(c));
+            if (!match)
+                throw new Error('[RudderJS AI] No transcription matched the predicate.');
+        }
+    }
+    /** Get all recorded provider calls */
+    getCalls() {
+        return [...this.calls];
+    }
+    /** Get all recorded image generation calls */
+    getImageCalls() {
+        return [...this.imageCalls];
+    }
+    /** Get all recorded TTS calls */
+    getTtsCalls() {
+        return [...this.ttsCalls];
+    }
+    /** Get all recorded STT calls */
+    getSttCalls() {
+        return [...this.sttCalls];
+    }
+    /** Restore — clears the fake (user must re-register real providers) */
+    restore() {
+        AiRegistry.reset();
+    }
+}
+//# sourceMappingURL=fake.js.map

package/dist/fake.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fake.js","sourceRoot":"","sources":["../src/fake.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAkB1C;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,MAAM;IACA,KAAK,GAA6B,EAAE,CAAA;IACpC,UAAU,GAA6B,EAAE,CAAA;IACzC,QAAQ,GAA0B,EAAE,CAAA;IACpC,QAAQ,GAA0B,EAAE,CAAA;IAC7C,SAAS,GAAG,eAAe,CAAA;IAC3B,cAAc,GAAG,kBAAkB,CAAA,CAAE,yBAAyB;IAC9D,YAAY,GAAW,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IAChD,YAAY,GAAG,oBAAoB,CAAA;IAE3C,iDAAiD;IACjD,WAAW,CAAC,IAAY;QACtB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA;IACvB,CAAC;IAED,gDAAgD;IAChD,gBAAgB,CAAC,MAAc;QAC7B,IAAI,CAAC,cAAc,GAAG,MAAM,CAAA;IAC9B,CAAC;IAED,oDAAoD;IACpD,gBAAgB,CAAC,KAAa;QAC5B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;IAC3B,CAAC;IAED,0DAA0D;IAC1D,wBAAwB,CAAC,IAAY;QACnC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;IAC1B,CAAC;IAED,uEAAuE;IACvE,MAAM,CAAC,IAAI;QACT,MAAM,IAAI,GAAG,IAAI,MAAM,EAAE,CAAA;QAEzB,MAAM,OAAO,GAAoB;YAC/B,KAAK,CAAC,QAAQ,CAAC,IAA4B;gBACzC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACrB,OAAO;oBACL,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,EAAE;oBACvD,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;oBAC/D,YAAY,EAAE,MAAM;iBACrB,CAAA;YACH,CAAC;YACD,KAAK,CAAC,CAAC,MAAM,CAAC,IAA4B;gBACxC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACrB,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,CAAA;gBAClD,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,EAAE,CAAA;YACjH,CAAC;SACF,CAAA;QAED,MAAM,YAAY,GAA2B;YAC3C,KAAK,CAAC,QAAQ,CAAC,IAA4B;gBACzC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBAC1B,OAAO;oBACL,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;oBACzC,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,YAAY;iBAClC,CAAA;YACH,CAAC;SACF,CAAA;QAED,MAAM,UAAU,GAAwB;YACtC,KAAK,CAAC,QAAQ,CAAC,IAAyB;gBACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACxB,OAAO;oBACL,KAAK,EAAE,IAAI,CAAC,YAAY;oBACxB,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,KAAK;oBAC5B,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,UAAU;iBAChC,CAAA;YACH,CAAC;SACF,CAAA;QAED,MAAM,UAAU,GAAwB;YACtC,KAAK,CAAC,UAAU,CAAC,IAAyB;gBACxC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACxB,OAAO;oBACL,IAAI,EAAE,IAAI,CAAC,YAAY;oBACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,UAAU;iBAChC,CAAA;YACH,CAAC;SACF,CAAA;QAED,MAAM,OAAO,GAAoB;YAC/B,IAAI,EAAE,UAAU;YAChB,MAAM,EAAE,GAAG,EAAE,CAAC,OAAO;YACrB,WAAW,EAAE,GAAG,EAAE,CAAC,YAAY;YAC/B,SAAS,EAAE,GAAG,EAAE,CAAC,UAAU;YAC3B,SAAS,EAAE,GAAG,EAAE,CAAC,UAAU;SAC5B,CAAA;QAED,UAAU,CAAC,KAAK,EAAE,CAAA;QAClB,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAA;QAC5B,UAAU,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAA;QACzC,OAAO,IAAI,CAAA;IACb,CAAC;IAED,2EAA2E;IAC3E,cAAc,CAAC,SAAsC;QACnD,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAA;QAC/G,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE;gBAChC,MAAM,OAAO,GAAG,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAA;gBACvD,IAAI,CAAC,OAAO;oBAAE,OAAO,KAAK,CAAA;gBAC1B,MAAM,IAAI,GAAG,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAsB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;gBACnK,OAAO,SAAS,CAAC,IAAI,CAAC,CAAA;YACxB,CAAC,CAAC,CAAA;YACF,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAA;QAC/E,CAAC;IACH,CAAC;IAED,kCAAkC;IAClC,qBAAqB;QACnB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,0CAA0C,IAAI,CAAC,KAAK,CAAC,MAAM,aAAa,CAAC,CAAA;QAC3F,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,oBAAoB,CAAC,SAAuC;QAC1D,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC,CAAA;QAC9H,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;YAC5D,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAA;QACzF,CAAC;IACH,CAAC;IAED,kDAAkD;IAClD,oBAAoB,CAAC,SAAqC;QACxD,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,2EAA2E,CAAC,CAAA;QAC5H,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;YACxD,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAA;QACzF,CAAC;IACH,CAAC;IAED,iDAAiD;IACjD,iBAAiB,CAAC,SAAkD;QAClE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC,CAAA;QACzH,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAA;YACnD,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAA;QACtF,CAAC;IACH,CAAC;IAED,sCAAsC;IACtC,QAAQ;QACN,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAA;IACxB,CAAC;IAED,8CAA8C;IAC9C,aAAa;QACX,OAAO,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,CAAA;IAC7B,CAAC;IAED,iCAAiC;IACjC,WAAW;QACT,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAA;IAC3B,CAAC;IAED,iCAAiC;IACjC,WAAW;QACT,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAA;IAC3B,CAAC;IAED,uEAAuE;IACvE,OAAO;QACL,UAAU,CAAC,KAAK,EAAE,CAAA;IACpB,CAAC;CACF"}

package/dist/image.d.ts

@@ -0,0 +1,27 @@
+import type { ImageGenerationResult } from './types.js';
+/**
+ * Fluent image generation facade.
+ *
+ * @example
+ * const result = await ImageGenerator.of('A sunset over mountains').size('landscape').generate()
+ * const path = await ImageGenerator.of('A logo').model('openai/dall-e-3').store('images/logo.png')
+ */
+export declare class ImageGenerator {
+    private readonly _prompt;
+    private _model;
+    private _size;
+    private _quality;
+    private _style;
+    private _n;
+    private constructor();
+    static of(prompt: string): ImageGenerator;
+    model(model: string): this;
+    size(size: string): this;
+    quality(quality: 'standard' | 'hd'): this;
+    style(style: 'natural' | 'vivid'): this;
+    count(n: number): this;
+    generate(): Promise<ImageGenerationResult>;
+    /** Generate and store the first image to storage. Requires @rudderjs/storage. */
+    store(path: string): Promise<string>;
+}
+//# sourceMappingURL=image.d.ts.map

package/dist/image.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.d.ts","sourceRoot":"","sources":["../src/image.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAA0B,qBAAqB,EAAE,MAAM,YAAY,CAAA;AAE/E;;;;;;GAMG;AACH,qBAAa,cAAc;IAOL,OAAO,CAAC,QAAQ,CAAC,OAAO;IAN5C,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,KAAK,CAAoB;IACjC,OAAO,CAAC,QAAQ,CAA+B;IAC/C,OAAO,CAAC,MAAM,CAAiC;IAC/C,OAAO,CAAC,EAAE,CAAoB;IAE9B,OAAO;IAEP,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc;IAIzC,KAAK,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAK1B,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAKxB,OAAO,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,GAAG,IAAI;IAKzC,KAAK,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,GAAG,IAAI;IAKvC,KAAK,CAAC,CAAC,EAAE,MAAM,GAAG,IAAI;IAKhB,QAAQ,IAAI,OAAO,CAAC,qBAAqB,CAAC;IAuBhD,iFAAiF;IAC3E,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAwB3C"}

package/dist/image.js

@@ -0,0 +1,90 @@
+import { AiRegistry } from './registry.js';
+/**
+ * Fluent image generation facade.
+ *
+ * @example
+ * const result = await ImageGenerator.of('A sunset over mountains').size('landscape').generate()
+ * const path = await ImageGenerator.of('A logo').model('openai/dall-e-3').store('images/logo.png')
+ */
+export class ImageGenerator {
+    _prompt;
+    _model;
+    _size;
+    _quality;
+    _style;
+    _n;
+    constructor(_prompt) {
+        this._prompt = _prompt;
+    }
+    static of(prompt) {
+        return new ImageGenerator(prompt);
+    }
+    model(model) {
+        this._model = model;
+        return this;
+    }
+    size(size) {
+        this._size = size;
+        return this;
+    }
+    quality(quality) {
+        this._quality = quality;
+        return this;
+    }
+    style(style) {
+        this._style = style;
+        return this;
+    }
+    count(n) {
+        this._n = n;
+        return this;
+    }
+    async generate() {
+        const modelStr = this._model ?? AiRegistry.getDefault();
+        const [providerName, modelName] = AiRegistry.parseModelString(modelStr);
+        const factory = AiRegistry.getFactory(providerName);
+        if (!factory.createImage) {
+            throw new Error(`[RudderJS AI] Provider "${providerName}" does not support image generation.`);
+        }
+        const adapter = factory.createImage(modelName);
+        const options = {
+            prompt: this._prompt,
+            model: modelStr,
+        };
+        if (this._size !== undefined)
+            options.size = this._size;
+        if (this._quality !== undefined)
+            options.quality = this._quality;
+        if (this._style !== undefined)
+            options.style = this._style;
+        if (this._n !== undefined)
+            options.n = this._n;
+        return adapter.generate(options);
+    }
+    /** Generate and store the first image to storage. Requires @rudderjs/storage. */
+    async store(path) {
+        const result = await this.generate();
+        const image = result.images[0];
+        if (!image)
+            throw new Error('[RudderJS AI] No image generated.');
+        try {
+            // eslint-disable-next-line @typescript-eslint/no-require-imports
+            const mod = await import(/* @vite-ignore */ '@rudderjs/storage');
+            const Storage = mod.Storage;
+            if (image.base64) {
+                const buffer = Buffer.from(image.base64, 'base64');
+                await Storage.put(path, buffer);
+            }
+            else if (image.url) {
+                const response = await fetch(image.url);
+                const buffer = Buffer.from(await response.arrayBuffer());
+                await Storage.put(path, buffer);
+            }
+            return path;
+        }
+        catch {
+            throw new Error('[RudderJS AI] Image storage requires @rudderjs/storage to be installed.');
+        }
+    }
+}
+//# sourceMappingURL=image.js.map

package/dist/image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.js","sourceRoot":"","sources":["../src/image.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAG1C;;;;;;GAMG;AACH,MAAM,OAAO,cAAc;IAOY;IAN7B,MAAM,CAAoB;IAC1B,KAAK,CAAoB;IACzB,QAAQ,CAA+B;IACvC,MAAM,CAAiC;IACvC,EAAE,CAAoB;IAE9B,YAAqC,OAAe;QAAf,YAAO,GAAP,OAAO,CAAQ;IAAG,CAAC;IAExD,MAAM,CAAC,EAAE,CAAC,MAAc;QACtB,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAA;IACnC,CAAC;IAED,KAAK,CAAC,KAAa;QACjB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACnB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,IAAI,CAAC,IAAY;QACf,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;QACjB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,OAAO,CAAC,OAA0B;QAChC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAA;QACvB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,KAAK,CAAC,KAA0B;QAC9B,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;QACnB,OAAO,IAAI,CAAA;IACb,CAAC;IAED,KAAK,CAAC,CAAS;QACb,IAAI,CAAC,EAAE,GAAG,CAAC,CAAA;QACX,OAAO,IAAI,CAAA;IACb,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,IAAI,UAAU,CAAC,UAAU,EAAE,CAAA;QACvD,MAAM,CAAC,YAAY,EAAE,SAAS,CAAC,GAAG,UAAU,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAA;QACvE,MAAM,OAAO,GAAG,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,CAAA;QAEnD,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,2BAA2B,YAAY,sCAAsC,CAAC,CAAA;QAChG,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAA;QAE9C,MAAM,OAAO,GAA2B;YACtC,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,KAAK,EAAE,QAAQ;SAChB,CAAA;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS;YAAE,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAA;QACvD,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS;YAAE,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAA;QAChE,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAA;QAC1D,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS;YAAE,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,CAAA;QAE9C,OAAO,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAA;IAClC,CAAC;IAED,iFAAiF;IACjF,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAA;QACpC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;QAC9B,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAA;QAEhE,IAAI,CAAC;YACH,iEAAiE;YACjE,MAAM,GAAG,GAAQ,MAAM,MAAM,CAAC,kBAAkB,CAAC,mBAA6B,CAAC,CAAA;YAC/E,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAA;YAE3B,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;gBAClD,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;YACjC,CAAC;iBAAM,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;gBACrB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;gBACvC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAA;gBACxD,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;YACjC,CAAC;YAED,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAA;QAC5F,CAAC;IACH,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,30 @@
+export { DocumentAttachment as Document, ImageAttachment as Image, getMessageText, attachmentsToContentParts } from './attachment.js';
+export type { AgentPromptOptions, Attachment, AiMessage, ContentPart, AiConfig, AiModelConfig, AiProviderConfig, AiMiddleware, AgentResponse, AgentStep, AgentStreamResponse, AnyTool, BeforeToolCallResult, ClientTool, ConversationStore, ConversationStoreMeta, FinishReason, HasMiddleware, HasMemory, HasStructuredOutput, HasTools, MiddlewareConfigResult, MiddlewareContext, PrepareStepResult, ProviderAdapter, ProviderFactory, ProviderRequestOptions, ProviderResponse, ServerTool, StopCondition, Tool, StreamChunk, TokenUsage, ToolCall, ToolCallContext, ToolChoice, ToolDefinitionOptions, ToolDefinitionSchema, ToolExecuteFn, ToolNeedsApproval, ToolResult, EmbeddingAdapter, EmbeddingResult, ImageGenerationAdapter, ImageGenerationOptions, ImageGenerationResult, TextToSpeechAdapter, TextToSpeechOptions, TextToSpeechResult, SpeechToTextAdapter, SpeechToTextOptions, SpeechToTextResult, } from './types.js';
+export { AiRegistry } from './registry.js';
+export { AnthropicProvider, type AnthropicConfig } from './providers/anthropic.js';
+export { OpenAIProvider, OpenAIAdapter, type OpenAIConfig } from './providers/openai.js';
+export { GoogleProvider, type GoogleConfig } from './providers/google.js';
+export { OllamaProvider, type OllamaConfig } from './providers/ollama.js';
+export { DeepSeekProvider, type DeepSeekConfig } from './providers/deepseek.js';
+export { XaiProvider, type XaiConfig } from './providers/xai.js';
+export { GroqProvider, type GroqConfig } from './providers/groq.js';
+export { MistralProvider, type MistralConfig } from './providers/mistral.js';
+export { AzureOpenAIProvider, type AzureOpenAIConfig } from './providers/azure.js';
+export { toolDefinition, dynamicTool, ToolBuilder, toolToSchema, pauseForClientTools, isPauseForClientToolsChunk, } from './tool.js';
+export type { PauseForClientToolsChunk } from './tool.js';
+export { zodToJsonSchema } from './zod-to-json-schema.js';
+export { Agent, ConversableAgent, agent, stepCountIs, hasToolCall, setConversationStore } from './agent.js';
+export { QueuedPromptBuilder } from './queue-job.js';
+export { runOnConfig, runOnChunk, runOnBeforeToolCall, runOnAfterToolCall, runSequential, runOnUsage, runOnAbort, runOnError } from './middleware.js';
+export { Output, type OutputWrapper } from './output.js';
+export { MemoryConversationStore } from './conversation.js';
+export { AI } from './facade.js';
+export { ai } from './provider.js';
+export { ImageGenerator } from './image.js';
+export { AudioGenerator } from './audio.js';
+export { Transcription } from './transcription.js';
+export { WebSearch, WebFetch, CodeExecution } from './provider-tools.js';
+export { toVercelDataStream, toVercelResponse } from './vercel-protocol.js';
+export { CachedEmbeddingAdapter } from './cached-embedding.js';
+export { AiFake } from './fake.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,kBAAkB,IAAI,QAAQ,EAAE,eAAe,IAAI,KAAK,EAAE,cAAc,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAA;AAGrI,YAAY,EACV,kBAAkB,EAClB,UAAU,EACV,SAAS,EACT,WAAW,EACX,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,YAAY,EACZ,aAAa,EACb,SAAS,EACT,mBAAmB,EACnB,OAAO,EACP,oBAAoB,EACpB,UAAU,EACV,iBAAiB,EACjB,qBAAqB,EACrB,YAAY,EACZ,aAAa,EACb,SAAS,EACT,mBAAmB,EACnB,QAAQ,EACR,sBAAsB,EACtB,iBAAiB,EACjB,iBAAiB,EACjB,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,gBAAgB,EAChB,UAAU,EACV,aAAa,EACb,IAAI,EACJ,WAAW,EACX,UAAU,EACV,QAAQ,EACR,eAAe,EACf,UAAU,EACV,qBAAqB,EACrB,oBAAoB,EACpB,aAAa,EACb,iBAAiB,EACjB,UAAU,EACV,gBAAgB,EAChB,eAAe,EACf,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,EACrB,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,EAClB,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,YAAY,CAAA;AAGnB,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAG1C,OAAO,EAAE,iBAAiB,EAAE,KAAK,eAAe,EAAE,MAAM,0BAA0B,CAAA;AAClF,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAA;AACxF,OAAO,EAAE,cAAc,EAAE,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAA;AACzE,OAAO,EAAE,cAAc,EAAE,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAA;AACzE,OAAO,EAAE,gBAAgB,EAAE,KAAK,cAAc,EAAE,MAAM,yBAAyB,CAAA;AAC/E,OAAO,EAAE,WAAW,EAAE,KAAK,SAAS,EAAE,MAAM,oBAAoB,CAAA;AAChE,OAAO,EAAE,YAAY,EAAE,KAAK,UAAU,EAAE,MAAM,qBAAqB,CAAA;AACnE,OAAO,EAAE,eAAe,EAAE,KAAK,aAAa,EAAE,MAAM,wBAAwB,CAAA;AAC5E,OAAO,EAAE,mBAAmB,EAAE,KAAK,iBAAiB,EAAE,MAAM,sBAAsB,CAAA;AAGlF,OAAO,EACL,cAAc,EACd,WAAW,EACX,WAAW,EACX,YAAY,EACZ,mBAAmB,EACnB,0BAA0B,GAC3B,MAAM,WAAW,CAAA;AAClB,YAAY,EAAE,wBAAwB,EAAE,MAAM,WAAW,CAAA;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AAGzD,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AAC3G,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAA;AAGpD,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,aAAa,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAGrJ,OAAO,EAAE,MAAM,EAAE,KAAK,aAAa,EAAE,MAAM,aAAa,CAAA;AAGxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAG3D,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAGhC,OAAO,EAAE,EAAE,EAAE,MAAM,eAAe,CAAA;AAGlC,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAG3C,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAGlD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AAGxE,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AAG3E,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAA;AAG9D,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA"}

package/dist/index.js

@@ -0,0 +1,45 @@
+// @rudderjs/ai — AI engine
+// Attachments
+export { DocumentAttachment as Document, ImageAttachment as Image, getMessageText, attachmentsToContentParts } from './attachment.js';
+// Registry
+export { AiRegistry } from './registry.js';
+// Providers
+export { AnthropicProvider } from './providers/anthropic.js';
+export { OpenAIProvider, OpenAIAdapter } from './providers/openai.js';
+export { GoogleProvider } from './providers/google.js';
+export { OllamaProvider } from './providers/ollama.js';
+export { DeepSeekProvider } from './providers/deepseek.js';
+export { XaiProvider } from './providers/xai.js';
+export { GroqProvider } from './providers/groq.js';
+export { MistralProvider } from './providers/mistral.js';
+export { AzureOpenAIProvider } from './providers/azure.js';
+// Tools
+export { toolDefinition, dynamicTool, ToolBuilder, toolToSchema, pauseForClientTools, isPauseForClientToolsChunk, } from './tool.js';
+export { zodToJsonSchema } from './zod-to-json-schema.js';
+// Agent
+export { Agent, ConversableAgent, agent, stepCountIs, hasToolCall, setConversationStore } from './agent.js';
+export { QueuedPromptBuilder } from './queue-job.js';
+// Middleware
+export { runOnConfig, runOnChunk, runOnBeforeToolCall, runOnAfterToolCall, runSequential, runOnUsage, runOnAbort, runOnError } from './middleware.js';
+// Structured Output
+export { Output } from './output.js';
+// Conversation
+export { MemoryConversationStore } from './conversation.js';
+// Facade
+export { AI } from './facade.js';
+// ServiceProvider factory
+export { ai } from './provider.js';
+// Image Generation
+export { ImageGenerator } from './image.js';
+// Audio (TTS & STT)
+export { AudioGenerator } from './audio.js';
+export { Transcription } from './transcription.js';
+// Provider Tools
+export { WebSearch, WebFetch, CodeExecution } from './provider-tools.js';
+// Vercel AI Protocol
+export { toVercelDataStream, toVercelResponse } from './vercel-protocol.js';
+// Cached Embeddings
+export { CachedEmbeddingAdapter } from './cached-embedding.js';
+// Testing
+export { AiFake } from './fake.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAE3B,cAAc;AACd,OAAO,EAAE,kBAAkB,IAAI,QAAQ,EAAE,eAAe,IAAI,KAAK,EAAE,cAAc,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAA;AA0DrI,WAAW;AACX,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAE1C,YAAY;AACZ,OAAO,EAAE,iBAAiB,EAAwB,MAAM,0BAA0B,CAAA;AAClF,OAAO,EAAE,cAAc,EAAE,aAAa,EAAqB,MAAM,uBAAuB,CAAA;AACxF,OAAO,EAAE,cAAc,EAAqB,MAAM,uBAAuB,CAAA;AACzE,OAAO,EAAE,cAAc,EAAqB,MAAM,uBAAuB,CAAA;AACzE,OAAO,EAAE,gBAAgB,EAAuB,MAAM,yBAAyB,CAAA;AAC/E,OAAO,EAAE,WAAW,EAAkB,MAAM,oBAAoB,CAAA;AAChE,OAAO,EAAE,YAAY,EAAmB,MAAM,qBAAqB,CAAA;AACnE,OAAO,EAAE,eAAe,EAAsB,MAAM,wBAAwB,CAAA;AAC5E,OAAO,EAAE,mBAAmB,EAA0B,MAAM,sBAAsB,CAAA;AAElF,QAAQ;AACR,OAAO,EACL,cAAc,EACd,WAAW,EACX,WAAW,EACX,YAAY,EACZ,mBAAmB,EACnB,0BAA0B,GAC3B,MAAM,WAAW,CAAA;AAElB,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAA;AAEzD,QAAQ;AACR,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AAC3G,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAA;AAEpD,aAAa;AACb,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,aAAa,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAErJ,oBAAoB;AACpB,OAAO,EAAE,MAAM,EAAsB,MAAM,aAAa,CAAA;AAExD,eAAe;AACf,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAE3D,SAAS;AACT,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAEhC,0BAA0B;AAC1B,OAAO,EAAE,EAAE,EAAE,MAAM,eAAe,CAAA;AAElC,mBAAmB;AACnB,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAE3C,oBAAoB;AACpB,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAElD,iBAAiB;AACjB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AAExE,qBAAqB;AACrB,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AAE3E,oBAAoB;AACpB,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAA;AAE9D,UAAU;AACV,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA"}

package/dist/middleware.d.ts

@@ -0,0 +1,18 @@
+import type { AiMiddleware, BeforeToolCallResult, MiddlewareConfigResult, MiddlewareContext, StreamChunk, TokenUsage } from './types.js';
+/** Run onConfig hooks — piped (each transforms config, next sees result) */
+export declare function runOnConfig(middlewares: AiMiddleware[], ctx: MiddlewareContext, config: MiddlewareConfigResult, phase: 'init' | 'beforeModel'): MiddlewareConfigResult;
+/** Run onChunk hooks — piped (each can transform or drop by returning null) */
+export declare function runOnChunk(middlewares: AiMiddleware[], ctx: MiddlewareContext, chunk: StreamChunk): StreamChunk | null;
+/** Run onBeforeToolCall hooks — first non-void result wins */
+export declare function runOnBeforeToolCall(middlewares: AiMiddleware[], ctx: MiddlewareContext, toolName: string, args: Record<string, unknown>): Promise<BeforeToolCallResult>;
+/** Run a sequential hook (onStart, onIteration, onAfterToolCall, etc.) */
+export declare function runSequential(middlewares: AiMiddleware[], hook: 'onStart' | 'onIteration' | 'onToolPhaseComplete' | 'onFinish', ctx: MiddlewareContext): Promise<void>;
+/** Run onAfterToolCall hooks sequentially */
+export declare function runOnAfterToolCall(middlewares: AiMiddleware[], ctx: MiddlewareContext, toolName: string, args: Record<string, unknown>, result: unknown): Promise<void>;
+/** Run onUsage hooks sequentially */
+export declare function runOnUsage(middlewares: AiMiddleware[], ctx: MiddlewareContext, usage: TokenUsage): Promise<void>;
+/** Run onAbort hooks sequentially */
+export declare function runOnAbort(middlewares: AiMiddleware[], ctx: MiddlewareContext, reason: string): Promise<void>;
+/** Run onError hooks sequentially */
+export declare function runOnError(middlewares: AiMiddleware[], ctx: MiddlewareContext, error: unknown): Promise<void>;
+//# sourceMappingURL=middleware.d.ts.map

package/dist/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,YAAY,EACZ,oBAAoB,EACpB,sBAAsB,EACtB,iBAAiB,EACjB,WAAW,EACX,UAAU,EACX,MAAM,YAAY,CAAA;AAEnB,4EAA4E;AAC5E,wBAAgB,WAAW,CACzB,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,MAAM,EAAE,sBAAsB,EAC9B,KAAK,EAAE,MAAM,GAAG,aAAa,GAC5B,sBAAsB,CASxB;AAED,+EAA+E;AAC/E,wBAAgB,UAAU,CACxB,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,KAAK,EAAE,WAAW,GACjB,WAAW,GAAG,IAAI,CASpB;AAED,8DAA8D;AAC9D,wBAAsB,mBAAmB,CACvC,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,OAAO,CAAC,oBAAoB,CAAC,CAQ/B;AAED,0EAA0E;AAC1E,wBAAsB,aAAa,CACjC,WAAW,EAAE,YAAY,EAAE,EAC3B,IAAI,EAAE,SAAS,GAAG,aAAa,GAAG,qBAAqB,GAAG,UAAU,EACpE,GAAG,EAAE,iBAAiB,GACrB,OAAO,CAAC,IAAI,CAAC,CAKf;AAED,6CAA6C;AAC7C,wBAAsB,kBAAkB,CACtC,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,MAAM,EAAE,OAAO,GACd,OAAO,CAAC,IAAI,CAAC,CAIf;AAED,qCAAqC;AACrC,wBAAsB,UAAU,CAC9B,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,KAAK,EAAE,UAAU,GAChB,OAAO,CAAC,IAAI,CAAC,CAIf;AAED,qCAAqC;AACrC,wBAAsB,UAAU,CAC9B,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC,CAIf;AAED,qCAAqC;AACrC,wBAAsB,UAAU,CAC9B,WAAW,EAAE,YAAY,EAAE,EAC3B,GAAG,EAAE,iBAAiB,EACtB,KAAK,EAAE,OAAO,GACb,OAAO,CAAC,IAAI,CAAC,CAIf"}