@rudderjs/ai 1.6.1 → 1.6.3

package/dist/server/provider.js

@@ -2,6 +2,94 @@ import { ServiceProvider, config } from '@rudderjs/core';
 import { AiRegistry } from '../registry.js';
 import { setConversationStore, setUserMemory } from '../agent.js';
 import { GoogleCacheRegistry } from '../providers/google-cache-registry.js';
+/**
+ * Throw with a clear pointer to the offending config key when an apiKey-
+ * requiring provider has no key set. The config type lets `apiKey` be
+ * undefined (some drivers — ollama, bedrock — don't need one), so this is
+ * the bootstrap-time enforcement point for the drivers that do.
+ */
+function requireKey(name, cfg) {
+    if (!cfg.apiKey) {
+        throw new Error(`[RudderJS AI] config('ai').providers.${name} is missing apiKey (driver "${cfg.driver ?? name}").`);
+    }
+    return cfg.apiKey;
+}
+const DRIVERS = {
+    anthropic: async (name, cfg) => {
+        const { AnthropicProvider } = await import('../providers/anthropic.js');
+        return new AnthropicProvider({ apiKey: requireKey(name, cfg), baseUrl: cfg.baseUrl });
+    },
+    openai: async (name, cfg) => {
+        const { OpenAIProvider } = await import('../providers/openai.js');
+        return new OpenAIProvider({
+            apiKey: requireKey(name, cfg),
+            baseUrl: cfg.baseUrl,
+            organization: cfg['organization'],
+        });
+    },
+    google: async (name, cfg, { googleCacheRegistry }) => {
+        const { GoogleProvider } = await import('../providers/google.js');
+        return new GoogleProvider({ apiKey: requireKey(name, cfg) }, googleCacheRegistry);
+    },
+    ollama: async (_name, cfg) => {
+        const { OllamaProvider } = await import('../providers/ollama.js');
+        return new OllamaProvider({ baseUrl: cfg.baseUrl });
+    },
+    deepseek: async (name, cfg) => {
+        const { DeepSeekProvider } = await import('../providers/deepseek.js');
+        return new DeepSeekProvider({ apiKey: requireKey(name, cfg), baseUrl: cfg.baseUrl });
+    },
+    xai: async (name, cfg) => {
+        const { XaiProvider } = await import('../providers/xai.js');
+        return new XaiProvider({ apiKey: requireKey(name, cfg), baseUrl: cfg.baseUrl });
+    },
+    groq: async (name, cfg) => {
+        const { GroqProvider } = await import('../providers/groq.js');
+        return new GroqProvider({ apiKey: requireKey(name, cfg), baseUrl: cfg.baseUrl });
+    },
+    mistral: async (name, cfg) => {
+        const { MistralProvider } = await import('../providers/mistral.js');
+        return new MistralProvider({ apiKey: requireKey(name, cfg), baseUrl: cfg.baseUrl });
+    },
+    azure: async (name, cfg) => {
+        const { AzureOpenAIProvider } = await import('../providers/azure.js');
+        if (!cfg.baseUrl) {
+            throw new Error(`[RudderJS AI] config('ai').providers.${name} is missing baseUrl (driver "azure" requires it).`);
+        }
+        return new AzureOpenAIProvider({ apiKey: requireKey(name, cfg), baseUrl: cfg.baseUrl });
+    },
+    openrouter: async (name, cfg) => {
+        const { OpenRouterProvider } = await import('../providers/openrouter.js');
+        return new OpenRouterProvider({
+            apiKey: requireKey(name, cfg),
+            baseUrl: cfg.baseUrl,
+            siteUrl: cfg['siteUrl'],
+            siteName: cfg['siteName'],
+        });
+    },
+    bedrock: async (_name, cfg) => {
+        const { BedrockProvider } = await import('../providers/bedrock.js');
+        const region = cfg['region'] ?? 'us-east-1';
+        const credentials = cfg['credentials'];
+        return new BedrockProvider(credentials ? { region, credentials } : { region });
+    },
+    elevenlabs: async (name, cfg) => {
+        const { ElevenLabsProvider } = await import('../providers/elevenlabs.js');
+        return new ElevenLabsProvider({
+            apiKey: requireKey(name, cfg),
+            ...(cfg.baseUrl ? { baseUrl: cfg.baseUrl } : {}),
+            ...(cfg['defaultTtsModelId'] ? { defaultTtsModelId: cfg['defaultTtsModelId'] } : {}),
+        });
+    },
+    voyage: async (name, cfg) => {
+        const { VoyageProvider } = await import('../providers/voyage.js');
+        return new VoyageProvider({
+            apiKey: requireKey(name, cfg),
+            ...(cfg.baseUrl ? { baseUrl: cfg.baseUrl } : {}),
+            ...(cfg['defaultInputType'] ? { defaultInputType: cfg['defaultInputType'] } : {}),
+        });
+    },
+};
 /**
  * AI ServiceProvider — reads config from `config('ai')`.
  *
@@ -17,114 +105,23 @@ export class AiProvider extends ServiceProvider {
         const googleCacheRegistry = this.buildGoogleCacheRegistry();
         for (const [name, providerConfig] of Object.entries(cfg.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,
-                }, googleCacheRegistry));
-            }
-            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,
-                }));
-            }
-            else if (driver === 'openrouter') {
-                const { OpenRouterProvider } = await import('../providers/openrouter.js');
-                AiRegistry.register(new OpenRouterProvider({
-                    apiKey: providerConfig.apiKey,
-                    baseUrl: providerConfig.baseUrl,
-                    siteUrl: providerConfig['siteUrl'],
-                    siteName: providerConfig['siteName'],
-                }));
-            }
-            else if (driver === 'bedrock') {
-                const { BedrockProvider } = await import('../providers/bedrock.js');
-                const region = providerConfig['region'] ?? 'us-east-1';
-                const credentials = providerConfig['credentials'];
-                AiRegistry.register(new BedrockProvider(credentials ? { region, credentials } : { region }));
-            }
-            else if (driver === 'elevenlabs') {
-                const { ElevenLabsProvider } = await import('../providers/elevenlabs.js');
-                AiRegistry.register(new ElevenLabsProvider({
-                    apiKey: providerConfig.apiKey,
-                    ...(providerConfig.baseUrl ? { baseUrl: providerConfig.baseUrl } : {}),
-                    ...(providerConfig['defaultTtsModelId'] ? { defaultTtsModelId: providerConfig['defaultTtsModelId'] } : {}),
-                }));
-            }
-            else if (driver === 'voyage') {
-                const { VoyageProvider } = await import('../providers/voyage.js');
-                AiRegistry.register(new VoyageProvider({
-                    apiKey: providerConfig.apiKey,
-                    ...(providerConfig.baseUrl ? { baseUrl: providerConfig.baseUrl } : {}),
-                    ...(providerConfig['defaultInputType'] ? { defaultInputType: providerConfig['defaultInputType'] } : {}),
-                }));
-            }
+            const build = DRIVERS[driver];
+            if (!build)
+                continue;
+            AiRegistry.register(await build(name, providerConfig, { googleCacheRegistry }));
         }
         AiRegistry.setDefault(cfg.default);
         AiRegistry.setModels(cfg.models ?? []);
         this.app.instance('ai.registry', AiRegistry);
         // Register conversation store if provided in config
         if (cfg.conversations) {
-            const store = cfg.conversations;
-            setConversationStore(store);
-            this.app.instance('ai.conversations', store);
+            setConversationStore(cfg.conversations);
+            this.app.instance('ai.conversations', cfg.conversations);
         }
         // Register user-memory store if provided in config (#A4)
         if (cfg.memory) {
-            const memory = cfg.memory;
-            setUserMemory(memory);
-            this.app.instance('ai.memory', memory);
+            setUserMemory(cfg.memory);
+            this.app.instance('ai.memory', cfg.memory);
         }
         // Register make:agent scaffolder
         try {

package/dist/server/provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/server/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAA;AAC3C,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAA;AACjE,OAAO,EAAE,mBAAmB,EAAuB,MAAM,uCAAuC,CAAA;AAGhG;;;;;;;GAOG;AACH,MAAM,OAAO,UAAW,SAAQ,eAAe;IAC7C,QAAQ,KAAU,CAAC;IAEnB,KAAK,CAAC,IAAI;QACR,MAAM,GAAG,GAAG,MAAM,CAAW,IAAI,CAAC,CAAA;QAClC,MAAM,mBAAmB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAA;QAE3D,KAAK,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACjE,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,IAAI,CAAA;YAE5C,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;gBAC3B,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAA;gBACvE,UAAU,CAAC,QAAQ,CAAC,IAAI,iBAAiB,CAAC;oBACxC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;iBAChC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;gBACjE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;oBACrC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,YAAY,EAAE,cAAc,CAAC,cAAc,CAAuB;iBACnE,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;gBACjE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;oBACrC,MAAM,EAAE,cAAc,CAAC,MAAO;iBAC/B,EAAE,mBAAmB,CAAC,CAAC,CAAA;YAC1B,CAAC;iBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;gBACjE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;oBACrC,OAAO,EAAE,cAAc,CAAC,OAAO;iBAChC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;gBACjC,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAA;gBACrE,UAAU,CAAC,QAAQ,CAAC,IAAI,gBAAgB,CAAC;oBACvC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;iBAChC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;gBAC5B,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,CAAA;gBAC3D,UAAU,CAAC,QAAQ,CAAC,IAAI,WAAW,CAAC;oBAClC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;iBAChC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;gBAC7B,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAA;gBAC7D,UAAU,CAAC,QAAQ,CAAC,IAAI,YAAY,CAAC;oBACnC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;iBAChC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAA;gBACnE,UAAU,CAAC,QAAQ,CAAC,IAAI,eAAe,CAAC;oBACtC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;iBAChC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;gBAC9B,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAA;gBACrE,UAAU,CAAC,QAAQ,CAAC,IAAI,mBAAmB,CAAC;oBAC1C,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAQ;iBACjC,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;gBACnC,MAAM,EAAE,kBAAkB,EAAE,GAAG,MAAM,MAAM,CAAC,4BAA4B,CAAC,CAAA;gBACzE,UAAU,CAAC,QAAQ,CAAC,IAAI,kBAAkB,CAAC;oBACzC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,OAAO,EAAE,cAAc,CAAC,SAAS,CAAuB;oBACxD,QAAQ,EAAE,cAAc,CAAC,UAAU,CAAuB;iBAC3D,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAA;gBACnE,MAAM,MAAM,GAAI,cAAc,CAAC,QAAQ,CAAwB,IAAI,WAAW,CAAA;gBAC9E,MAAM,WAAW,GAAG,cAAc,CAAC,aAAa,CAEnC,CAAA;gBACb,UAAU,CAAC,QAAQ,CAAC,IAAI,eAAe,CACrC,WAAW,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CACnD,CAAC,CAAA;YACJ,CAAC;iBAAM,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;gBACnC,MAAM,EAAE,kBAAkB,EAAE,GAAG,MAAM,MAAM,CAAC,4BAA4B,CAAC,CAAA;gBACzE,UAAU,CAAC,QAAQ,CAAC,IAAI,kBAAkB,CAAC;oBACzC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,GAAG,CAAC,cAAc,CAAC,OAAO,CAAc,CAAC,CAAC,EAAE,OAAO,EAAY,cAAc,CAAC,OAAO,EAA+B,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC1H,GAAG,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,cAAc,CAAC,mBAAmB,CAAW,EAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC3H,CAAC,CAAC,CAAA;YACL,CAAC;iBAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;gBACjE,UAAU,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC;oBACrC,MAAM,EAAE,cAAc,CAAC,MAAO;oBAC9B,GAAG,CAAC,cAAc,CAAC,OAAO,CAAa,CAAC,CAAC,EAAE,OAAO,EAAW,cAAc,CAAC,OAAO,EAA6C,CAAC,CAAC,CAAC,EAAE,CAAC;oBACtI,GAAG,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,cAAc,CAAC,kBAAkB,CAAyB,EAAS,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvI,CAAC,CAAC,CAAA;YACL,CAAC;QACH,CAAC;QAEH,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAClC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAA;QACtC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;QAE5C,oDAAoD;QACpD,IAAK,GAAwD,CAAC,aAAa,EAAE,CAAC;YAC5E,MAAM,KAAK,GAAI,GAAwD,CAAC,aAAc,CAAA;YACtF,oBAAoB,CAAC,KAAK,CAAC,CAAA;YAC3B,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAA;QAC9C,CAAC;QAED,yDAAyD;QACzD,IAAK,GAA0C,CAAC,MAAM,EAAE,CAAC;YACvD,MAAM,MAAM,GAAI,GAA0C,CAAC,MAAO,CAAA;YAClE,aAAa,CAAC,MAAM,CAAC,CAAA;YACrB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC,CAAA;QACxC,CAAC;QAED,iCAAiC;QACjC,IAAI,CAAC;YACH,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAA;YAC/D,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAA;YACnE,iBAAiB,CAAC,aAAa,CAAC,CAAA;QAClC,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;IACxC,CAAC;IAED;;;;;;OAMG;IACK,wBAAwB;QAC9B,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAiB,OAAO,CAAC,CAAA;YACpD,OAAO,IAAI,mBAAmB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;QAC3C,CAAC;QACD,OAAO,IAAI,mBAAmB,EAAE,CAAA;IAClC,CAAC;CACF"}
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/server/provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAA;AAC3C,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAA;AACjE,OAAO,EAAE,mBAAmB,EAAuB,MAAM,uCAAuC,CAAA;AAGhG;;;;;GAKG;AACH,SAAS,UAAU,CAAC,IAAY,EAAE,GAAqB;IACrD,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,wCAAwC,IAAI,+BAA+B,GAAG,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC,CAAA;IACrH,CAAC;IACD,OAAO,GAAG,CAAC,MAAM,CAAA;AACnB,CAAC;AAKD,MAAM,OAAO,GAAkC;IAC7C,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC7B,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAA;QACvE,OAAO,IAAI,iBAAiB,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACvF,CAAC;IACD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC1B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;QACjE,OAAO,IAAI,cAAc,CAAC;YACxB,MAAM,EAAQ,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;YACnC,OAAO,EAAO,GAAG,CAAC,OAAO;YACzB,YAAY,EAAE,GAAG,CAAC,cAAc,CAAuB;SACxD,CAAC,CAAA;IACJ,CAAC;IACD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE;QACnD,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;QACjE,OAAO,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,mBAAmB,CAAC,CAAA;IACnF,CAAC;IACD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QAC3B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;QACjE,OAAO,IAAI,cAAc,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACrD,CAAC;IACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC5B,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAA;QACrE,OAAO,IAAI,gBAAgB,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACtF,CAAC;IACD,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACvB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,CAAA;QAC3D,OAAO,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACjF,CAAC;IACD,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACxB,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAA;QAC7D,OAAO,IAAI,YAAY,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IAClF,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC3B,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAA;QACnE,OAAO,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACrF,CAAC;IACD,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACzB,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAA;QACrE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wCAAwC,IAAI,mDAAmD,CAAC,CAAA;QAClH,CAAC;QACD,OAAO,IAAI,mBAAmB,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAA;IACzF,CAAC;IACD,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC9B,MAAM,EAAE,kBAAkB,EAAE,GAAG,MAAM,MAAM,CAAC,4BAA4B,CAAC,CAAA;QACzE,OAAO,IAAI,kBAAkB,CAAC;YAC5B,MAAM,EAAI,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;YAC/B,OAAO,EAAG,GAAG,CAAC,OAAO;YACrB,OAAO,EAAG,GAAG,CAAC,SAAS,CAAuB;YAC9C,QAAQ,EAAE,GAAG,CAAC,UAAU,CAAuB;SAChD,CAAC,CAAA;IACJ,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QAC5B,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAA;QACnE,MAAM,MAAM,GAAI,GAAG,CAAC,QAAQ,CAAwB,IAAI,WAAW,CAAA;QACnE,MAAM,WAAW,GAAG,GAAG,CAAC,aAAa,CAExB,CAAA;QACb,OAAO,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAA;IAChF,CAAC;IACD,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC9B,MAAM,EAAE,kBAAkB,EAAE,GAAG,MAAM,MAAM,CAAC,4BAA4B,CAAC,CAAA;QACzE,OAAO,IAAI,kBAAkB,CAAC;YAC5B,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;YAC7B,GAAG,CAAC,GAAG,CAAC,OAAO,CAAc,CAAC,CAAC,EAAE,OAAO,EAAY,GAAG,CAAC,OAAO,EAA+B,CAAC,CAAC,CAAC,EAAE,CAAC;YACpG,GAAG,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,CAAC,mBAAmB,CAAW,EAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;SACrG,CAAC,CAAA;IACJ,CAAC;IACD,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC1B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAA;QACjE,OAAO,IAAI,cAAc,CAAC;YACxB,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;YAC7B,GAAG,CAAC,GAAG,CAAC,OAAO,CAAa,CAAC,CAAC,EAAE,OAAO,EAAW,GAAG,CAAC,OAAO,EAA6C,CAAC,CAAC,CAAC,EAAE,CAAC;YAChH,GAAG,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,GAAG,CAAC,kBAAkB,CAAyB,EAAS,CAAC,CAAC,CAAC,EAAE,CAAC;SACjH,CAAC,CAAA;IACJ,CAAC;CACF,CAAA;AAED;;;;;;;GAOG;AACH,MAAM,OAAO,UAAW,SAAQ,eAAe;IAC7C,QAAQ,KAAU,CAAC;IAEnB,KAAK,CAAC,IAAI;QACR,MAAM,GAAG,GAAG,MAAM,CAAW,IAAI,CAAC,CAAA;QAClC,MAAM,mBAAmB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAA;QAE3D,KAAK,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACnE,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,IAAI,CAAA;YAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAA;YAC7B,IAAI,CAAC,KAAK;gBAAE,SAAQ;YACpB,UAAU,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAA;QACjF,CAAC;QAED,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QAClC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAA;QACtC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;QAE5C,oDAAoD;QACpD,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;YACtB,oBAAoB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAA;YACvC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,CAAC,aAAa,CAAC,CAAA;QAC1D,CAAC;QAED,yDAAyD;QACzD,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACf,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;YACzB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,CAAA;QAC5C,CAAC;QAED,iCAAiC;QACjC,IAAI,CAAC;YACH,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAA;YAC/D,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAA;YACnE,iBAAiB,CAAC,aAAa,CAAC,CAAA;QAClC,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;IACxC,CAAC;IAED;;;;;;OAMG;IACK,wBAAwB;QAC9B,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAiB,OAAO,CAAC,CAAA;YACpD,OAAO,IAAI,mBAAmB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;QAC3C,CAAC;QACD,OAAO,IAAI,mBAAmB,EAAE,CAAA;IAClC,CAAC;CACF"}

package/dist/tool-execution.d.ts

@@ -0,0 +1,16 @@
+import type { LoopContext } from './agent.js';
+import type { AiMessage, StreamChunk, ToolCall, ToolResult } from './types.js';
+/**
+ * Execute the tool phase for a single agent step. Yields the same
+ * `StreamChunk` sequence (`tool-call` → `tool-update*` → `tool-result`) that
+ * the streaming caller surfaces to consumers. Non-streaming callers iterate
+ * via `.next()` and discard yields — the side effects (message pushes,
+ * pending-state mutations on `loopCtx`) are identical regardless of whether
+ * the chunks reach a consumer.
+ *
+ * Returns the step's `ToolResult[]`. The caller passes the assistant message
+ * to push before iteration so the AgentStep shape (response.message) and the
+ * final `messages` array stay in sync with the loop variant.
+ */
+export declare function executeToolPhase(loopCtx: LoopContext, toolCalls: ToolCall[], assistantMessage: AiMessage): AsyncGenerator<StreamChunk, ToolResult[], void>;
+//# sourceMappingURL=tool-execution.d.ts.map

package/dist/tool-execution.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-execution.d.ts","sourceRoot":"","sources":["../src/tool-execution.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AAkB7C,OAAO,KAAK,EAAE,SAAS,EAAW,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAEvF;;;;;;;;;;;GAWG;AACH,wBAAuB,gBAAgB,CACrC,OAAO,EAAW,WAAW,EAC7B,SAAS,EAAS,QAAQ,EAAE,EAC5B,gBAAgB,EAAE,SAAS,GAC1B,cAAc,CAAC,WAAW,EAAE,UAAU,EAAE,EAAE,IAAI,CAAC,CA8BjD"}