@primero.ai/temporal-graph-tools 1.3.0 → 1.4.1

package/dist/examples/mastra-ai-workflow/activities/classify-outage-with-ai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"classify-outage-with-ai.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/classify-outage-with-ai.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA;AAEjD,MAAM,MAAM,YAAY,GAAG,cAAc,GAAG,SAAS,GAAG,SAAS,CAAA;AACjE,MAAM,MAAM,aAAa,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAA;AAErD,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,YAAY,CAAA;IACpB,OAAO,EAAE,aAAa,CAAA;IACtB,OAAO,EAAE,MAAM,CAAA;IACf,cAAc,EAAE,MAAM,CAAA;CACvB,CAAA;AA4ED,eAAO,MAAM,oBAAoB,2FA6ChC,CAAA"}
+{"version":3,"file":"classify-outage-with-ai.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/classify-outage-with-ai.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA;AAEjD,MAAM,MAAM,YAAY,GAAG,cAAc,GAAG,SAAS,GAAG,SAAS,CAAA;AACjE,MAAM,MAAM,aAAa,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAA;AAErD,MAAM,MAAM,gBAAgB,GAAG;IAC7B,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,YAAY,CAAA;IACpB,OAAO,EAAE,aAAa,CAAA;IACtB,OAAO,EAAE,MAAM,CAAA;IACf,cAAc,EAAE,MAAM,CAAA;CACvB,CAAA;AAiED,eAAO,MAAM,oBAAoB,2FA0DhC,CAAA"}

package/dist/examples/mastra-ai-workflow/activities/classify-outage-with-ai.js

@@ -1,7 +1,5 @@
-var _a;
-import { openai } from '@ai-sdk/openai';
 import { z } from 'zod';
-import { wrapActivityWithMastraAgent } from '@primero.ai/temporal-graph-tools';
+import { resolveAiSelection, selectAiModel, wrapActivityWithMastraAgent, } from '@primero.ai/temporal-graph-tools';
 function inferIntent(message) {
     const normalized = message.toLowerCase();
     if (normalized.includes('invoice') ||
@@ -40,26 +38,27 @@ function inferRootCause(message) {
     }
     return 'unknown';
 }
-const modelId = (_a = process.env.MASTRA_AI_MODEL) !== null && _a !== void 0 ? _a : 'gpt-4o-mini';
 const classificationOutputSchema = z.object({
     intent: z.enum(['connectivity', 'billing', 'account']),
     urgency: z.enum(['low', 'medium', 'high']),
     summary: z.string(),
     recommendedFix: z.string(),
 });
-const classificationAgentInput = {
-    id: 'mastra-outage-classifier',
-    name: 'Mastra Outage Classifier',
-    instructions: [
-        'You classify support tickets and propose precise remediation.',
-        'If the outage seems local-network and no provider outage is reported, recommend restarting the router.',
-        'Return concise and operational output.',
-    ].join(' '),
-    model: openai(modelId),
-};
 export const classifyOutageWithAi = wrapActivityWithMastraAgent(async ({ input: ticket, agent }) => {
-    var _a;
-    const provider = (_a = process.env.MASTRA_AI_PROVIDER) !== null && _a !== void 0 ? _a : 'openai';
+    const aiSelection = resolveAiSelection({
+        defaultModel: 'gpt-4o-mini',
+        defaultProvider: 'openai',
+    });
+    const classificationAgentInput = {
+        id: 'mastra-outage-classifier',
+        name: 'Mastra Outage Classifier',
+        instructions: [
+            'You classify support tickets and propose precise remediation.',
+            'If the outage seems local-network and no provider outage is reported, recommend restarting the router.',
+            'Return concise and operational output.',
+        ].join(' '),
+        model: selectAiModel(aiSelection),
+    };
     const fallbackIntent = inferIntent(ticket.outageDescription);
     const fallbackUrgency = inferUrgency(ticket.outageDescription);
     const fallbackSummary = ticket.outageDescription.slice(0, 180);
@@ -79,8 +78,8 @@ export const classifyOutageWithAi = wrapActivityWithMastraAgent(async ({ input:
             schema: classificationOutputSchema,
         });
         return {
-            provider,
-            model: modelId,
+            provider: aiSelection.provider,
+            model: aiSelection.model,
             intent: parsed.intent,
             urgency: parsed.urgency,
             summary: parsed.summary,
@@ -91,8 +90,8 @@ export const classifyOutageWithAi = wrapActivityWithMastraAgent(async ({ input:
         console.warn('[mastra-ai] classifyOutageWithAi fallback:', error);
     }
     return {
-        provider,
-        model: modelId,
+        provider: aiSelection.provider,
+        model: aiSelection.model,
         intent: fallbackIntent,
         urgency: fallbackUrgency,
         summary: fallbackSummary,

package/dist/examples/mastra-ai-workflow/activities/draft-response-with-ai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"draft-response-with-ai.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/draft-response-with-ai.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAA;AACjE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAA;AAEtE,MAAM,MAAM,gBAAgB,GAAG;IAC7B,oBAAoB,EAAE,gBAAgB,CAAA;IACtC,uBAAuB,EAAE,mBAAmB,CAAA;CAC7C,CAAA;AAED,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,MAAM,CAAA;IACf,IAAI,EAAE,MAAM,CAAA;CACb,CAAA;AAiBD,eAAO,MAAM,mBAAmB,4FAwC/B,CAAA"}
+{"version":3,"file":"draft-response-with-ai.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/draft-response-with-ai.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAA;AACjE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAA;AAEtE,MAAM,MAAM,gBAAgB,GAAG;IAC7B,oBAAoB,EAAE,gBAAgB,CAAA;IACtC,uBAAuB,EAAE,mBAAmB,CAAA;CAC7C,CAAA;AAED,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,EAAE,MAAM,CAAA;IACf,IAAI,EAAE,MAAM,CAAA;CACb,CAAA;AAOD,eAAO,MAAM,mBAAmB,4FAiD/B,CAAA"}

package/dist/examples/mastra-ai-workflow/activities/draft-response-with-ai.js

@@ -1,22 +1,19 @@
-var _a;
-import { openai } from '@ai-sdk/openai';
 import { z } from 'zod';
-import { wrapActivityWithMastraAgent } from '@primero.ai/temporal-graph-tools';
-const modelId = (_a = process.env.MASTRA_AI_MODEL) !== null && _a !== void 0 ? _a : 'gpt-4o-mini';
+import { selectAiModel, wrapActivityWithMastraAgent } from '@primero.ai/temporal-graph-tools';
 const responseOutputSchema = z.object({
     subject: z.string(),
     body: z.string(),
 });
-const responseAgentInput = {
-    id: 'mastra-response-writer',
-    name: 'Mastra Response Writer',
-    instructions: [
-        'You write short, professional customer responses for internet incidents.',
-        'If evidence indicates local-network issue without provider outage, explicitly instruct router restart.',
-    ].join(' '),
-    model: openai(modelId),
-};
 export const draftResponseWithAi = wrapActivityWithMastraAgent(async ({ input: analysis, agent }) => {
+    const responseAgentInput = {
+        id: 'mastra-response-writer',
+        name: 'Mastra Response Writer',
+        instructions: [
+            'You write short, professional customer responses for internet incidents.',
+            'If evidence indicates local-network issue without provider outage, explicitly instruct router restart.',
+        ].join(' '),
+        model: selectAiModel({ defaultModel: 'gpt-4o-mini' }),
+    };
     const { classifyOutageWithAi: ai, enrichOutageDescription: enriched } = analysis;
     const fallbackSubject = `[${ai.intent.toUpperCase()}][${ai.urgency}] Internet Incident`;
     const fallbackBody = [

package/dist/src/index.d.ts

@@ -4,6 +4,7 @@ export * from './workflow-bundler.js';
 export { collectWorkflowBuildResults } from './workflow/collection.js';
 export * from './bundler.js';
 export * from './mastra/agent-wrapper.js';
+export * from './mastra/ai-model.js';
 export * from './mastra/create-activity-with-mastra.js';
 export * from './mastra/with-mastra-agent.js';
 export * from './mastra/workflow-compiler.js';

package/dist/src/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAC9E,cAAc,uBAAuB,CAAA;AACrC,OAAO,EAAE,2BAA2B,EAAE,MAAM,0BAA0B,CAAA;AACtE,cAAc,cAAc,CAAA;AAC5B,cAAc,2BAA2B,CAAA;AACzC,cAAc,yCAAyC,CAAA;AACvD,cAAc,+BAA+B,CAAA;AAC7C,cAAc,+BAA+B,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAC9E,cAAc,uBAAuB,CAAA;AACrC,OAAO,EAAE,2BAA2B,EAAE,MAAM,0BAA0B,CAAA;AACtE,cAAc,cAAc,CAAA;AAC5B,cAAc,2BAA2B,CAAA;AACzC,cAAc,sBAAsB,CAAA;AACpC,cAAc,yCAAyC,CAAA;AACvD,cAAc,+BAA+B,CAAA;AAC7C,cAAc,+BAA+B,CAAA"}

package/dist/src/index.js

@@ -4,6 +4,7 @@ export * from './workflow-bundler.js';
 export { collectWorkflowBuildResults } from './workflow/collection.js';
 export * from './bundler.js';
 export * from './mastra/agent-wrapper.js';
+export * from './mastra/ai-model.js';
 export * from './mastra/create-activity-with-mastra.js';
 export * from './mastra/with-mastra-agent.js';
 export * from './mastra/workflow-compiler.js';

package/dist/src/mastra/ai-model.d.ts

@@ -0,0 +1,16 @@
+import { openai } from '@ai-sdk/openai';
+export type SupportedAiProvider = 'anthropic' | 'azure' | 'deepseek' | 'google' | 'openai' | 'xai';
+export type AiModelSelectionOptions = {
+    provider?: string;
+    model?: string;
+    defaultProvider?: SupportedAiProvider;
+    defaultModel?: string;
+};
+export declare function resolveAiProvider(options?: Pick<AiModelSelectionOptions, 'provider' | 'defaultProvider'>): SupportedAiProvider;
+export declare function resolveAiModel(options?: Pick<AiModelSelectionOptions, 'model' | 'defaultModel'>): string;
+export declare function resolveAiSelection(options?: AiModelSelectionOptions): {
+    provider: SupportedAiProvider;
+    model: string;
+};
+export declare function selectAiModel(options?: AiModelSelectionOptions): ReturnType<typeof openai>;
+//# sourceMappingURL=ai-model.d.ts.map

package/dist/src/mastra/ai-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-model.d.ts","sourceRoot":"","sources":["../../../src/mastra/ai-model.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AAGvC,MAAM,MAAM,mBAAmB,GAAG,WAAW,GAAG,OAAO,GAAG,UAAU,GAAG,QAAQ,GAAG,QAAQ,GAAG,KAAK,CAAA;AAElG,MAAM,MAAM,uBAAuB,GAAG;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,eAAe,CAAC,EAAE,mBAAmB,CAAA;IACrC,YAAY,CAAC,EAAE,MAAM,CAAA;CACtB,CAAA;AAqCD,wBAAgB,iBAAiB,CAC/B,OAAO,GAAE,IAAI,CAAC,uBAAuB,EAAE,UAAU,GAAG,iBAAiB,CAAM,GAC1E,mBAAmB,CAuBrB;AAED,wBAAgB,cAAc,CAC5B,OAAO,GAAE,IAAI,CAAC,uBAAuB,EAAE,OAAO,GAAG,cAAc,CAAM,GACpE,MAAM,CAeR;AAED,wBAAgB,kBAAkB,CAAC,OAAO,GAAE,uBAA4B,GAAG;IACzE,QAAQ,EAAE,mBAAmB,CAAA;IAC7B,KAAK,EAAE,MAAM,CAAA;CACd,CAKA;AAED,wBAAgB,aAAa,CAAC,OAAO,GAAE,uBAA4B,GAAG,UAAU,CAAC,OAAO,MAAM,CAAC,CA4B9F"}

package/dist/src/mastra/ai-model.js

@@ -0,0 +1,87 @@
+import { anthropic } from '@ai-sdk/anthropic';
+import { azure } from '@ai-sdk/azure';
+import { deepseek } from '@ai-sdk/deepseek';
+import { google } from '@ai-sdk/google';
+import { openai } from '@ai-sdk/openai';
+import { xai } from '@ai-sdk/xai';
+const DEFAULT_PROVIDER = 'openai';
+const PROVIDER_ALIASES = Object.freeze({
+    claude: 'anthropic',
+    gemini: 'google',
+});
+function normalizeString(value) {
+    if (typeof value !== 'string') {
+        return undefined;
+    }
+    const normalized = value.trim();
+    return normalized.length > 0 ? normalized : undefined;
+}
+function readDagAiConfig() {
+    const raw = process.env.PRIMERO_DAG_AI_CONFIG;
+    if (!raw) {
+        return {};
+    }
+    try {
+        const parsed = JSON.parse(raw);
+        return {
+            provider: typeof parsed.provider === 'string' ? parsed.provider : undefined,
+            model: typeof parsed.model === 'string' ? parsed.model : undefined,
+        };
+    }
+    catch {
+        return {};
+    }
+}
+export function resolveAiProvider(options = {}) {
+    var _a, _b, _c, _d, _e;
+    const dagAiConfig = readDagAiConfig();
+    const candidate = (_d = (_c = (_b = (_a = normalizeString(options.provider)) !== null && _a !== void 0 ? _a : normalizeString(dagAiConfig.provider)) !== null && _b !== void 0 ? _b : normalizeString(process.env.AI_PROVIDER)) !== null && _c !== void 0 ? _c : options.defaultProvider) !== null && _d !== void 0 ? _d : DEFAULT_PROVIDER;
+    const lowered = candidate.toLowerCase();
+    const resolved = (_e = PROVIDER_ALIASES[lowered]) !== null && _e !== void 0 ? _e : lowered;
+    if (resolved === 'openai' ||
+        resolved === 'anthropic' ||
+        resolved === 'google' ||
+        resolved === 'xai' ||
+        resolved === 'deepseek' ||
+        resolved === 'azure') {
+        return resolved;
+    }
+    throw new Error(`Unsupported AI provider '${candidate}'.`);
+}
+export function resolveAiModel(options = {}) {
+    var _a, _b, _c;
+    const dagAiConfig = readDagAiConfig();
+    const model = (_c = (_b = (_a = normalizeString(options.model)) !== null && _a !== void 0 ? _a : normalizeString(dagAiConfig.model)) !== null && _b !== void 0 ? _b : normalizeString(process.env.AI_MODEL)) !== null && _c !== void 0 ? _c : normalizeString(options.defaultModel);
+    if (!model) {
+        throw new Error('No AI model configured. Pass a model option, set PRIMERO_DAG_AI_CONFIG, or set AI_MODEL in the environment.');
+    }
+    return model;
+}
+export function resolveAiSelection(options = {}) {
+    return {
+        provider: resolveAiProvider(options),
+        model: resolveAiModel(options),
+    };
+}
+export function selectAiModel(options = {}) {
+    const { provider, model } = resolveAiSelection(options);
+    if (provider === 'openai') {
+        return openai(model);
+    }
+    if (provider === 'anthropic') {
+        return anthropic(model);
+    }
+    if (provider === 'google') {
+        return google(model);
+    }
+    if (provider === 'xai') {
+        return xai(model);
+    }
+    if (provider === 'deepseek') {
+        return deepseek(model);
+    }
+    if (provider === 'azure') {
+        return azure(model);
+    }
+    throw new Error('Unsupported AI provider.');
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@primero.ai/temporal-graph-tools",
-  "version": "1.3.0",
+  "version": "1.4.1",
   "description": "Utilities for working with DAG workflows.",
   "keywords": [],
   "homepage": "https://github.com/primero-ai/temporal-graph-tools#readme",
@@ -73,7 +73,12 @@
     "clean": "rm -rf node_modules dist"
   },
   "dependencies": {
+    "@ai-sdk/anthropic": "^3.0.45",
+    "@ai-sdk/azure": "^3.0.31",
+    "@ai-sdk/deepseek": "^2.0.20",
+    "@ai-sdk/google": "^3.0.30",
     "@ai-sdk/openai": "^3.0.30",
+    "@ai-sdk/xai": "^3.0.57",
     "@mastra/core": "^1.4.0",
     "@temporalio/activity": "^1.15.0",
     "@temporalio/client": "1.15.0",