@primero.ai/temporal-graph-tools 1.4.0 → 1.4.2

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

@@ -1 +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;AAmBD,wBAAgB,iBAAiB,CAC/B,OAAO,GAAE,IAAI,CAAC,uBAAuB,EAAE,UAAU,GAAG,iBAAiB,CAAM,GAC1E,mBAAmB,CAqBrB;AAED,wBAAgB,cAAc,CAC5B,OAAO,GAAE,IAAI,CAAC,uBAAuB,EAAE,OAAO,GAAG,cAAc,CAAM,GACpE,MAAM,CAaR;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"}
+{"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

@@ -16,11 +16,28 @@ function normalizeString(value) {
     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;
-    const candidate = (_c = (_b = (_a = normalizeString(options.provider)) !== null && _a !== void 0 ? _a : normalizeString(process.env.MASTRA_AI_PROVIDER)) !== null && _b !== void 0 ? _b : options.defaultProvider) !== null && _c !== void 0 ? _c : DEFAULT_PROVIDER;
+    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 = (_d = PROVIDER_ALIASES[lowered]) !== null && _d !== void 0 ? _d : lowered;
+    const resolved = (_e = PROVIDER_ALIASES[lowered]) !== null && _e !== void 0 ? _e : lowered;
     if (resolved === 'openai' ||
         resolved === 'anthropic' ||
         resolved === 'google' ||
@@ -32,10 +49,11 @@ export function resolveAiProvider(options = {}) {
     throw new Error(`Unsupported AI provider '${candidate}'.`);
 }
 export function resolveAiModel(options = {}) {
-    var _a, _b;
-    const model = (_b = (_a = normalizeString(options.model)) !== null && _a !== void 0 ? _a : normalizeString(process.env.MASTRA_AI_MODEL)) !== null && _b !== void 0 ? _b : normalizeString(options.defaultModel);
+    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 or set MASTRA_AI_MODEL in the environment.');
+        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;
 }
@@ -65,5 +83,5 @@ export function selectAiModel(options = {}) {
     if (provider === 'azure') {
         return azure(model);
     }
-    throw new Error(`Unsupported AI provider '${provider}'.`);
+    throw new Error('Unsupported AI provider.');
 }

package/dist/src/mastra/workflow-compiler.d.ts

@@ -3,6 +3,8 @@ export type MastraWorkflowLike = {
     id?: string;
     engineType?: string;
     stepGraph: Array<Record<string, unknown>>;
+    steps?: Record<string, unknown>;
+    stepById?: Record<string, unknown>;
 };
 export type CompileMastraWorkflowOptions = Omit<TemporalWorkflowBuildOptions, 'workflowName'> & {
     workflowName?: string;

package/dist/src/mastra/workflow-compiler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"workflow-compiler.d.ts","sourceRoot":"","sources":["../../../src/mastra/workflow-compiler.ts"],"names":[],"mappings":"AAEA,OAAO,EAIL,KAAK,4BAA4B,EACjC,KAAK,mBAAmB,EACzB,MAAM,aAAa,CAAA;AAQpB,MAAM,MAAM,kBAAkB,GAAG;IAC/B,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;CAC1C,CAAA;AAED,MAAM,MAAM,4BAA4B,GAAG,IAAI,CAAC,4BAA4B,EAAE,cAAc,CAAC,GAAG;IAC9F,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACxC;;;;;OAKG;IACH,8BAA8B,CAAC,EAAE,OAAO,CAAA;CACzC,CAAA;AAED,MAAM,MAAM,8BAA8B,GAAG;IAC3C,QAAQ,EAAE,kBAAkB,CAAA;IAC5B,OAAO,CAAC,EAAE,4BAA4B,CAAA;CACvC,CAAA;AAED,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,kBAAkB,EAC5B,OAAO,GAAE,4BAAiC,GACzC,mBAAmB,CA+DrB"}
+{"version":3,"file":"workflow-compiler.d.ts","sourceRoot":"","sources":["../../../src/mastra/workflow-compiler.ts"],"names":[],"mappings":"AAEA,OAAO,EAIL,KAAK,4BAA4B,EACjC,KAAK,mBAAmB,EACzB,MAAM,aAAa,CAAA;AAYpB,MAAM,MAAM,kBAAkB,GAAG;IAC/B,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;IACzC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CACnC,CAAA;AAED,MAAM,MAAM,4BAA4B,GAAG,IAAI,CAAC,4BAA4B,EAAE,cAAc,CAAC,GAAG;IAC9F,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACxC;;;;;OAKG;IACH,8BAA8B,CAAC,EAAE,OAAO,CAAA;CACzC,CAAA;AAED,MAAM,MAAM,8BAA8B,GAAG;IAC3C,QAAQ,EAAE,kBAAkB,CAAA;IAC5B,OAAO,CAAC,EAAE,4BAA4B,CAAA;CACvC,CAAA;AAED,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,kBAAkB,EAC5B,OAAO,GAAE,4BAAiC,GACzC,mBAAmB,CAwFrB"}

package/dist/src/mastra/workflow-compiler.js

@@ -13,8 +13,16 @@ export function compileMastraWorkflow(workflow, options = {}) {
         ...(options.proxyOptions ? { proxyOptions: options.proxyOptions } : {}),
     });
     workflow.stepGraph.forEach((entry, stageIndex) => {
-        if (entry.type === 'step' && isMastraStepLike(entry.step)) {
-            const reference = resolveStepReference(workflow, entry.step, options.stepActivities, options.allowUnsafeStepExecuteFallback === true);
+        if (entry.type === 'step') {
+            const step = resolveMastraStep(entry.step, workflow);
+            if (!step) {
+                const stepId = getMastraStepId(entry.step);
+                throw new Error(stepId
+                    ? `Unable to resolve Mastra step '${stepId}' at stage ${stageIndex}. ` +
+                        `Ensure workflow.steps/workflow.stepById includes executable steps or pass stepActivities explicitly.`
+                    : `Unsupported Mastra step entry at stage ${stageIndex}. Expected an executable step or step id.`);
+            }
+            const reference = resolveStepReference(workflow, step, options.stepActivities, options.allowUnsafeStepExecuteFallback === true);
             builder = builder.then(reference);
             return;
         }
@@ -22,14 +30,21 @@ export function compileMastraWorkflow(workflow, options = {}) {
             const references = entry.steps.map((child, branchIndex) => {
                 if (typeof child !== 'object' ||
                     child === null ||
-                    child.type !== 'step' ||
-                    !isMastraStepLike(child.step)) {
+                    child.type !== 'step') {
                     const childType = typeof child === 'object' && child !== null
                         ? String(child.type)
                         : typeof child;
                     throw new Error(`Unsupported Mastra parallel child type '${childType}' at stage ${stageIndex}, branch ${branchIndex}.`);
                 }
-                return resolveStepReference(workflow, child.step, options.stepActivities, options.allowUnsafeStepExecuteFallback === true);
+                const step = resolveMastraStep(child.step, workflow);
+                if (!step) {
+                    const stepId = getMastraStepId(child.step);
+                    throw new Error(stepId
+                        ? `Unable to resolve Mastra parallel step '${stepId}' at stage ${stageIndex}, branch ${branchIndex}. ` +
+                            `Ensure workflow.steps/workflow.stepById includes executable steps or pass stepActivities explicitly.`
+                        : `Unsupported Mastra parallel step at stage ${stageIndex}, branch ${branchIndex}.`);
+                }
+                return resolveStepReference(workflow, step, options.stepActivities, options.allowUnsafeStepExecuteFallback === true);
             });
             builder = builder.parallel(references);
             return;
@@ -45,6 +60,38 @@ function isMastraStepLike(value) {
     const step = value;
     return typeof step.id === 'string' && typeof step.execute === 'function';
 }
+function isMastraStepIdentifier(value) {
+    if (typeof value !== 'object' || value === null) {
+        return false;
+    }
+    const step = value;
+    return typeof step.id === 'string' && step.id.trim().length > 0;
+}
+function resolveMastraStep(stepCandidate, workflow) {
+    var _a, _b;
+    if (isMastraStepLike(stepCandidate)) {
+        return stepCandidate;
+    }
+    if (!isMastraStepIdentifier(stepCandidate)) {
+        return null;
+    }
+    const { id } = stepCandidate;
+    const fromSteps = (_a = workflow.steps) === null || _a === void 0 ? void 0 : _a[id];
+    if (isMastraStepLike(fromSteps)) {
+        return fromSteps;
+    }
+    const fromStepById = (_b = workflow.stepById) === null || _b === void 0 ? void 0 : _b[id];
+    if (isMastraStepLike(fromStepById)) {
+        return fromStepById;
+    }
+    return null;
+}
+function getMastraStepId(stepCandidate) {
+    if (!isMastraStepIdentifier(stepCandidate)) {
+        return null;
+    }
+    return stepCandidate.id;
+}
 function resolveWorkflowName(workflow, explicitName) {
     const explicit = typeof explicitName === 'string' ? explicitName.trim() : '';
     if (explicit.length > 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@primero.ai/temporal-graph-tools",
-  "version": "1.4.0",
+  "version": "1.4.2",
   "description": "Utilities for working with DAG workflows.",
   "keywords": [],
   "homepage": "https://github.com/primero-ai/temporal-graph-tools#readme",