@primero.ai/temporal-graph-tools 1.1.1 → 1.2.0

package/README.md

@@ -37,15 +37,22 @@ import {
   bundleWorkflows,
   createActivity,
   createWorkflowBuilder,
+  wrapActivityWithMastraContext,
 } from '@primero.ai/temporal-graph-tools'
 
 type FetchUserInput = { userId: string }
 type FetchUserOutput = { profile: { id: string; name: string } }
 
 const fetchUserProfile = createActivity(
-  async ({ userId }: FetchUserInput): Promise<FetchUserOutput> => {
-    return { profile: { id: userId, name: `User ${userId}` } }
-  },
+  wrapActivityWithMastraContext(
+    async ({ userId }: FetchUserInput): Promise<FetchUserOutput> => {
+      return { profile: { id: userId, name: `User ${userId}` } }
+    },
+    async ({ context, execute }) => {
+      console.log(`[mastra] node=${context?.node.key} stage=${context?.node.stageIndex}`)
+      return execute()
+    },
+  ),
   { id: 'fetchUserProfile' },
 )
 
@@ -140,6 +147,19 @@ object is provided its `id` becomes the activity key; without options the
 function name (or an auto-incremented fallback) is used. The helper is also
 re-exported as `createActivity` for codebases that prefer plural naming.
 
+### `wrapActivityWithMastraContext(activityFn, wrapper)`
+
+Wraps an activity with a Mastra-style execution hook. The generated workflow now
+passes a second argument to each node with:
+
+- `workflow.name`
+- `graph.totalStages`
+- `node` (`key`, `stageIndex`, `stageType`, optional `branchIndex`)
+- `previousNode`
+
+Use this wrapper to add tracing, logging, metrics, or custom context plumbing
+before and after calling `execute()`.
+
 ## Workflow bundler utilities
 
 ### `bundleWorkflows(plans, options?)`

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

@@ -0,0 +1,13 @@
+import type { TicketIntake } from './take-ticket';
+export type TicketIntent = 'connectivity' | 'billing' | 'account';
+export type TicketUrgency = 'low' | 'medium' | 'high';
+export type AiClassification = {
+    provider: string;
+    model: string;
+    intent: TicketIntent;
+    urgency: TicketUrgency;
+    summary: string;
+    recommendedFix: string;
+};
+export declare const classifyOutageWithAi: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<TicketIntake, AiClassification, "classifyOutageWithAi">;
+//# sourceMappingURL=classify-outage-with-ai.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"classify-outage-with-ai.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/classify-outage-with-ai.ts"],"names":[],"mappings":"AAOA,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;AA8ED,eAAO,MAAM,oBAAoB,gIAqDhC,CAAA"}

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

@@ -0,0 +1,109 @@
+var _a;
+import { createActivityWithMastra, } from '@primero.ai/temporal-graph-tools';
+import { openai } from '@ai-sdk/openai';
+import { z } from 'zod';
+function logMastraContext(context) {
+    var _a, _b, _c, _d, _e;
+    if (!(context === null || context === void 0 ? void 0 : context.node)) {
+        return;
+    }
+    console.log(`[mastra-ai] node=${(_a = context.node.key) !== null && _a !== void 0 ? _a : 'unknown'} stage=${(_b = context.node.stageIndex) !== null && _b !== void 0 ? _b : -1} type=${(_c = context.node.stageType) !== null && _c !== void 0 ? _c : 'unknown'} previous=${(_e = (_d = context.previousNode) === null || _d === void 0 ? void 0 : _d.key) !== null && _e !== void 0 ? _e : 'none'}`);
+}
+function inferIntent(message) {
+    const normalized = message.toLowerCase();
+    if (normalized.includes('invoice') || normalized.includes('refund') || normalized.includes('charge')) {
+        return 'billing';
+    }
+    if (normalized.includes('password') || normalized.includes('login') || normalized.includes('2fa')) {
+        return 'account';
+    }
+    return 'connectivity';
+}
+function inferUrgency(message) {
+    const normalized = message.toLowerCase();
+    if (normalized.includes('urgent') || normalized.includes('down') || normalized.includes('cannot work')) {
+        return 'high';
+    }
+    if (normalized.includes('soon') || normalized.includes('today')) {
+        return 'medium';
+    }
+    return 'low';
+}
+function inferRootCause(message) {
+    const normalized = message.toLowerCase();
+    const providerOutage = /provider outage|isp outage|service outage/.test(normalized);
+    const localIssue = /router|wifi|internet not working|local network|modem/.test(normalized);
+    if (localIssue && !providerOutage) {
+        return 'local-network';
+    }
+    if (providerOutage) {
+        return 'provider-outage';
+    }
+    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 classificationAgentContext = {};
+const classificationMastraInput = {
+    agentInput: {
+        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),
+    },
+    agentContext: classificationAgentContext,
+};
+export const classifyOutageWithAi = createActivityWithMastra({
+    id: 'classifyOutageWithAi',
+    onMastraContext: logMastraContext,
+    mastraInput: classificationMastraInput,
+}, async ({ input: ticket, agent }) => {
+    var _a;
+    const provider = (_a = process.env.MASTRA_AI_PROVIDER) !== null && _a !== void 0 ? _a : 'openai';
+    const fallbackIntent = inferIntent(ticket.outageDescription);
+    const fallbackUrgency = inferUrgency(ticket.outageDescription);
+    const fallbackSummary = ticket.outageDescription.slice(0, 180);
+    const fallbackRootCause = inferRootCause(ticket.outageDescription);
+    const fallbackFix = fallbackRootCause === 'local-network'
+        ? 'Restart the router and wait 2 minutes before reconnecting devices.'
+        : 'Collect diagnostics and continue support triage.';
+    try {
+        const parsed = await agent({
+            prompt: [
+                `Ticket ID: ${ticket.ticketId}`,
+                `Client: ${ticket.client.name} (${ticket.client.accountId}) at ${ticket.client.site}`,
+                `Outage Description: ${ticket.outageDescription}`,
+                'Classify and provide recommended fix.',
+            ].join('\n'),
+            schema: classificationOutputSchema,
+        });
+        return {
+            provider,
+            model: modelId,
+            intent: parsed.intent,
+            urgency: parsed.urgency,
+            summary: parsed.summary,
+            recommendedFix: parsed.recommendedFix,
+        };
+    }
+    catch (error) {
+        console.warn('[mastra-ai] classifyOutageWithAi fallback:', error);
+    }
+    return {
+        provider,
+        model: modelId,
+        intent: fallbackIntent,
+        urgency: fallbackUrgency,
+        summary: fallbackSummary,
+        recommendedFix: fallbackFix,
+    };
+});

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

@@ -0,0 +1,12 @@
+import type { AiClassification } from './classify-outage-with-ai';
+import type { EnrichedDescription } from './enrich-outage-description';
+export type ParallelAnalysis = {
+    classifyOutageWithAi: AiClassification;
+    enrichOutageDescription: EnrichedDescription;
+};
+export type DraftResponse = {
+    subject: string;
+    body: string;
+};
+export declare const draftResponseWithAi: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<ParallelAnalysis, DraftResponse, "draftResponseWithAi">;
+//# sourceMappingURL=draft-response-with-ai.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"draft-response-with-ai.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/draft-response-with-ai.ts"],"names":[],"mappings":"AAOA,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;AA+BD,eAAO,MAAM,mBAAmB,gIAgD/B,CAAA"}

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

@@ -0,0 +1,69 @@
+var _a;
+import { createActivityWithMastra, } from '@primero.ai/temporal-graph-tools';
+import { openai } from '@ai-sdk/openai';
+import { z } from 'zod';
+function logMastraContext(context) {
+    var _a, _b, _c, _d, _e;
+    if (!(context === null || context === void 0 ? void 0 : context.node)) {
+        return;
+    }
+    console.log(`[mastra-ai] node=${(_a = context.node.key) !== null && _a !== void 0 ? _a : 'unknown'} stage=${(_b = context.node.stageIndex) !== null && _b !== void 0 ? _b : -1} type=${(_c = context.node.stageType) !== null && _c !== void 0 ? _c : 'unknown'} previous=${(_e = (_d = context.previousNode) === null || _d === void 0 ? void 0 : _d.key) !== null && _e !== void 0 ? _e : 'none'}`);
+}
+const modelId = (_a = process.env.MASTRA_AI_MODEL) !== null && _a !== void 0 ? _a : 'gpt-4o-mini';
+const responseOutputSchema = z.object({
+    subject: z.string(),
+    body: z.string(),
+});
+const responseAgentContext = {};
+const responseMastraInput = {
+    agentInput: {
+        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),
+    },
+    agentContext: responseAgentContext,
+};
+export const draftResponseWithAi = createActivityWithMastra({
+    id: 'draftResponseWithAi',
+    onMastraContext: logMastraContext,
+    mastraInput: responseMastraInput,
+}, async ({ input: analysis, agent }) => {
+    const { classifyOutageWithAi: ai, enrichOutageDescription: enriched } = analysis;
+    const fallbackSubject = `[${ai.intent.toUpperCase()}][${ai.urgency}] Internet Incident`;
+    const fallbackBody = [
+        `Hi ${enriched.client.name},`,
+        '',
+        `We analyzed your report: ${ai.summary}`,
+        `Enriched context: ${enriched.rewrittenDescription}`,
+        '',
+        `Recommended action: ${ai.recommendedFix}`,
+        '',
+        'Please confirm once the router has restarted and devices are reconnected.',
+    ].join('\n');
+    try {
+        const parsed = await agent({
+            prompt: [
+                `Client: ${enriched.client.name} (${enriched.client.accountId}) at ${enriched.client.site}`,
+                `Classification Intent: ${ai.intent}`,
+                `Classification Urgency: ${ai.urgency}`,
+                `Classification Summary: ${ai.summary}`,
+                `Enriched Description: ${enriched.rewrittenDescription}`,
+                `Required Recommendation: ${ai.recommendedFix}`,
+                'Generate a support response for the client.',
+            ].join('\n'),
+            schema: responseOutputSchema,
+        });
+        return {
+            subject: parsed.subject,
+            body: parsed.body,
+        };
+    }
+    catch (error) {
+        console.warn('[mastra-ai] draftResponseWithAi fallback:', error);
+    }
+    return { subject: fallbackSubject, body: fallbackBody };
+});

package/dist/examples/mastra-ai-workflow/activities/enrich-outage-description.d.ts

@@ -0,0 +1,8 @@
+import type { TicketIntake } from './take-ticket';
+export type EnrichedDescription = {
+    client: TicketIntake['client'];
+    rewrittenDescription: string;
+    inferredRootCause: 'local-network' | 'provider-outage' | 'unknown';
+};
+export declare const enrichOutageDescription: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<TicketIntake, EnrichedDescription, "enrichOutageDescription">;
+//# sourceMappingURL=enrich-outage-description.d.ts.map

package/dist/examples/mastra-ai-workflow/activities/enrich-outage-description.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"enrich-outage-description.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/enrich-outage-description.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA;AAEjD,MAAM,MAAM,mBAAmB,GAAG;IAChC,MAAM,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAA;IAC9B,oBAAoB,EAAE,MAAM,CAAA;IAC5B,iBAAiB,EAAE,eAAe,GAAG,iBAAiB,GAAG,SAAS,CAAA;CACnE,CAAA;AA4BD,eAAO,MAAM,uBAAuB,sIAwBnC,CAAA"}

package/dist/examples/mastra-ai-workflow/activities/enrich-outage-description.js

@@ -0,0 +1,37 @@
+import { createActivityWithMastra } from '@primero.ai/temporal-graph-tools';
+function logMastraContext(context) {
+    var _a, _b, _c, _d, _e;
+    if (!(context === null || context === void 0 ? void 0 : context.node)) {
+        return;
+    }
+    console.log(`[mastra-ai] node=${(_a = context.node.key) !== null && _a !== void 0 ? _a : 'unknown'} stage=${(_b = context.node.stageIndex) !== null && _b !== void 0 ? _b : -1} type=${(_c = context.node.stageType) !== null && _c !== void 0 ? _c : 'unknown'} previous=${(_e = (_d = context.previousNode) === null || _d === void 0 ? void 0 : _d.key) !== null && _e !== void 0 ? _e : 'none'}`);
+}
+function inferRootCause(message) {
+    const normalized = message.toLowerCase();
+    const providerOutage = /provider outage|isp outage|service outage/.test(normalized);
+    const localIssue = /router|wifi|internet not working|local network|modem/.test(normalized);
+    if (localIssue && !providerOutage) {
+        return 'local-network';
+    }
+    if (providerOutage) {
+        return 'provider-outage';
+    }
+    return 'unknown';
+}
+export const enrichOutageDescription = createActivityWithMastra({
+    id: 'enrichOutageDescription',
+    onMastraContext: logMastraContext,
+}, async ({ input: ticket }) => {
+    const rootCause = inferRootCause(ticket.outageDescription);
+    const rewrittenDescription = [
+        `Client ${ticket.client.name} (${ticket.client.accountId}) at ${ticket.client.site} reports internet connectivity failure.`,
+        `Raw report: ${ticket.outageDescription}`,
+        `AI enrichment indicates likely cause: ${rootCause}.`,
+        'No broad service outage is reported by the client.',
+    ].join(' ');
+    return {
+        client: ticket.client,
+        rewrittenDescription,
+        inferredRootCause: rootCause,
+    };
+});

package/dist/examples/mastra-ai-workflow/activities/index.d.ts

@@ -0,0 +1,11 @@
+export { classifyOutageWithAi } from './classify-outage-with-ai';
+export { draftResponseWithAi } from './draft-response-with-ai';
+export { enrichOutageDescription } from './enrich-outage-description';
+export { submitResponse } from './submit-response';
+export { takeTicket } from './take-ticket';
+export type { AiClassification, TicketIntent, TicketUrgency } from './classify-outage-with-ai';
+export type { DraftResponse, ParallelAnalysis } from './draft-response-with-ai';
+export type { EnrichedDescription } from './enrich-outage-description';
+export type { SubmittedResponse } from './submit-response';
+export type { SupportTicketInput, TicketIntake } from './take-ticket';
+//# sourceMappingURL=index.d.ts.map

package/dist/examples/mastra-ai-workflow/activities/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAA;AAChE,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAA;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,6BAA6B,CAAA;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAC1C,YAAY,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAA;AAC9F,YAAY,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAA;AAC/E,YAAY,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAA;AACtE,YAAY,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAA;AAC1D,YAAY,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,eAAe,CAAA"}

package/dist/examples/mastra-ai-workflow/activities/index.js

@@ -0,0 +1,5 @@
+export { classifyOutageWithAi } from './classify-outage-with-ai';
+export { draftResponseWithAi } from './draft-response-with-ai';
+export { enrichOutageDescription } from './enrich-outage-description';
+export { submitResponse } from './submit-response';
+export { takeTicket } from './take-ticket';

package/dist/examples/mastra-ai-workflow/activities/submit-response.d.ts

@@ -0,0 +1,7 @@
+import type { DraftResponse } from './draft-response-with-ai';
+export type SubmittedResponse = {
+    ready: true;
+    response: DraftResponse;
+};
+export declare const submitResponse: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<DraftResponse, SubmittedResponse, "submitResponse">;
+//# sourceMappingURL=submit-response.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"submit-response.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/submit-response.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AAE7D,MAAM,MAAM,iBAAiB,GAAG;IAC9B,KAAK,EAAE,IAAI,CAAA;IACX,QAAQ,EAAE,aAAa,CAAA;CACxB,CAAA;AAYD,eAAO,MAAM,cAAc,4HAS1B,CAAA"}

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

@@ -0,0 +1,15 @@
+import { createActivityWithMastra } from '@primero.ai/temporal-graph-tools';
+function logMastraContext(context) {
+    var _a, _b, _c, _d, _e;
+    if (!(context === null || context === void 0 ? void 0 : context.node)) {
+        return;
+    }
+    console.log(`[mastra-ai] node=${(_a = context.node.key) !== null && _a !== void 0 ? _a : 'unknown'} stage=${(_b = context.node.stageIndex) !== null && _b !== void 0 ? _b : -1} type=${(_c = context.node.stageType) !== null && _c !== void 0 ? _c : 'unknown'} previous=${(_e = (_d = context.previousNode) === null || _d === void 0 ? void 0 : _d.key) !== null && _e !== void 0 ? _e : 'none'}`);
+}
+export const submitResponse = createActivityWithMastra({
+    id: 'submitResponse',
+    onMastraContext: logMastraContext,
+}, async ({ input: draft }) => ({
+    ready: true,
+    response: draft,
+}));

package/dist/examples/mastra-ai-workflow/activities/take-ticket.d.ts

@@ -0,0 +1,16 @@
+export type SupportTicketInput = {
+    ticketId: string;
+    client: {
+        name: string;
+        accountId: string;
+        site: string;
+    };
+    outageDescription: string;
+};
+export type TicketIntake = {
+    ticketId: string;
+    client: SupportTicketInput['client'];
+    outageDescription: string;
+};
+export declare const takeTicket: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<SupportTicketInput, TicketIntake, "takeTicket">;
+//# sourceMappingURL=take-ticket.d.ts.map

package/dist/examples/mastra-ai-workflow/activities/take-ticket.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"take-ticket.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/take-ticket.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,kBAAkB,GAAG;IAC/B,QAAQ,EAAE,MAAM,CAAA;IAChB,MAAM,EAAE;QACN,IAAI,EAAE,MAAM,CAAA;QACZ,SAAS,EAAE,MAAM,CAAA;QACjB,IAAI,EAAE,MAAM,CAAA;KACb,CAAA;IACD,iBAAiB,EAAE,MAAM,CAAA;CAC1B,CAAA;AAED,MAAM,MAAM,YAAY,GAAG;IACzB,QAAQ,EAAE,MAAM,CAAA;IAChB,MAAM,EAAE,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IACpC,iBAAiB,EAAE,MAAM,CAAA;CAC1B,CAAA;AAYD,eAAO,MAAM,UAAU,wHAUtB,CAAA"}

package/dist/examples/mastra-ai-workflow/activities/take-ticket.js

@@ -0,0 +1,16 @@
+import { createActivityWithMastra } from '@primero.ai/temporal-graph-tools';
+function logMastraContext(context) {
+    var _a, _b, _c, _d, _e;
+    if (!(context === null || context === void 0 ? void 0 : context.node)) {
+        return;
+    }
+    console.log(`[mastra-ai] node=${(_a = context.node.key) !== null && _a !== void 0 ? _a : 'unknown'} stage=${(_b = context.node.stageIndex) !== null && _b !== void 0 ? _b : -1} type=${(_c = context.node.stageType) !== null && _c !== void 0 ? _c : 'unknown'} previous=${(_e = (_d = context.previousNode) === null || _d === void 0 ? void 0 : _d.key) !== null && _e !== void 0 ? _e : 'none'}`);
+}
+export const takeTicket = createActivityWithMastra({
+    id: 'takeTicket',
+    onMastraContext: logMastraContext,
+}, async ({ input }) => ({
+    ticketId: input.ticketId,
+    client: input.client,
+    outageDescription: input.outageDescription.trim(),
+}));

package/dist/examples/mastra-ai-workflow/workflows.d.ts

@@ -0,0 +1,2 @@
+export declare const builderMastraAiSupportWorkflow: import("@primero.ai/temporal-graph-tools").WorkflowBuildResult;
+//# sourceMappingURL=workflows.d.ts.map

package/dist/examples/mastra-ai-workflow/workflows.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflows.d.ts","sourceRoot":"","sources":["../../../examples/mastra-ai-workflow/workflows.ts"],"names":[],"mappings":"AAUA,eAAO,MAAM,8BAA8B,gEAQhC,CAAA"}

package/dist/examples/mastra-ai-workflow/workflows.js

@@ -0,0 +1,11 @@
+import { createWorkflowBuilder } from '@primero.ai/temporal-graph-tools';
+import { classifyOutageWithAi, draftResponseWithAi, enrichOutageDescription, submitResponse, takeTicket, } from './activities';
+export const builderMastraAiSupportWorkflow = createWorkflowBuilder({
+    workflowName: 'mastraAiSupportWorkflow',
+    proxyOptions: { startToCloseTimeout: '2 minutes' },
+})
+    .then(takeTicket)
+    .parallel([classifyOutageWithAi, enrichOutageDescription])
+    .then(draftResponseWithAi)
+    .then(submitResponse)
+    .commit();

package/dist/examples/mastra-example.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=mastra-example.d.ts.map

package/dist/examples/mastra-example.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastra-example.d.ts","sourceRoot":"","sources":["../../examples/mastra-example.ts"],"names":[],"mappings":""}

package/dist/examples/mastra-example.js

@@ -0,0 +1,103 @@
+import { createActivity, createWorkflowBuilder, wrapActivityWithMastraContext, } from '@primero.ai/temporal-graph-tools';
+function withMastraLogging(id, activity) {
+    return createActivity(wrapActivityWithMastraContext(activity, async ({ context, execute }) => {
+        var _a, _b;
+        if (context) {
+            console.log(`[mastra] workflow=${context.workflow.name} node=${context.node.key} type=${context.node.stageType} stage=${context.node.stageIndex} previous=${(_b = (_a = context.previousNode) === null || _a === void 0 ? void 0 : _a.key) !== null && _b !== void 0 ? _b : 'none'}`);
+        }
+        return execute();
+    }), { id });
+}
+const normalizeLead = withMastraLogging('normalizeLead', async (input) => {
+    const normalizedEmail = input.email.trim().toLowerCase();
+    return {
+        id: normalizedEmail.split('@')[0],
+        name: input.name.trim(),
+        email: normalizedEmail,
+    };
+});
+const scoreLead = withMastraLogging('scoreLead', async (lead) => {
+    const score = lead.email.endsWith('.edu') ? 82 : 64;
+    return { lead, score };
+});
+const createCrmTask = withMastraLogging('createCrmTask', async ({ lead, score }) => {
+    return { taskId: `crm-${lead.id}-${score}`, queued: true };
+});
+const notifySales = withMastraLogging('notifySales', async ({ lead, score }) => {
+    console.log(`[sales] lead=${lead.email} score=${score}`);
+    return { channel: 'slack', sent: true };
+});
+const finalizeLead = withMastraLogging('finalizeLead', async (result) => {
+    console.log(`[finalize] task=${result.createCrmTask.taskId} notify=${result.notifySales.sent}`);
+    return { completed: true };
+});
+const builderMastra = createWorkflowBuilder({
+    workflowName: 'mastraLeadWorkflow',
+    proxyOptions: { startToCloseTimeout: '2 minutes' },
+})
+    .then(normalizeLead)
+    .then(scoreLead)
+    .parallel([createCrmTask, notifySales])
+    .then(finalizeLead)
+    .commit();
+function makeContext(workflowName, totalStages, node, previousNode) {
+    return {
+        workflow: { name: workflowName },
+        graph: { totalStages },
+        node,
+        previousNode,
+    };
+}
+async function runLocalDryRun() {
+    console.log(`Workflow: ${builderMastra.workflowName}`);
+    console.log('\nGenerated workflow source:\n');
+    console.log(builderMastra.workflowSource);
+    const totalStages = 4;
+    const workflowName = builderMastra.workflowName;
+    const input = { name: '  Ada Lovelace ', email: 'Ada@Example.com ' };
+    const step0Node = {
+        key: 'normalizeLead',
+        stageIndex: 0,
+        stageType: 'step',
+    };
+    const normalized = await normalizeLead.activity(input, makeContext(workflowName, totalStages, step0Node, null));
+    const step1Node = {
+        key: 'scoreLead',
+        stageIndex: 1,
+        stageType: 'step',
+    };
+    const scored = await scoreLead.activity(normalized, makeContext(workflowName, totalStages, step1Node, step0Node));
+    const parallelNodeA = {
+        key: 'createCrmTask',
+        stageIndex: 2,
+        stageType: 'parallel',
+        branchIndex: 0,
+    };
+    const parallelNodeB = {
+        key: 'notifySales',
+        stageIndex: 2,
+        stageType: 'parallel',
+        branchIndex: 1,
+    };
+    const [crmTask, salesMessage] = await Promise.all([
+        createCrmTask.activity(scored, makeContext(workflowName, totalStages, parallelNodeA, step1Node)),
+        notifySales.activity(scored, makeContext(workflowName, totalStages, parallelNodeB, step1Node)),
+    ]);
+    const step3Node = {
+        key: 'finalizeLead',
+        stageIndex: 3,
+        stageType: 'step',
+    };
+    const parallelAggregateNode = {
+        key: 'parallel_2',
+        stageIndex: 2,
+        stageType: 'parallel',
+    };
+    const finalResult = await finalizeLead.activity({ createCrmTask: crmTask, notifySales: salesMessage }, makeContext(workflowName, totalStages, step3Node, parallelAggregateNode));
+    console.log('\nDry run result:');
+    console.log(finalResult);
+}
+runLocalDryRun().catch((error) => {
+    console.error('Mastra example failed:', error);
+    process.exitCode = 1;
+});

package/dist/examples/normal-workflow/activities.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"activities.d.ts","sourceRoot":"","sources":["../../../examples/normal-workflow/activities.ts"],"names":[],"mappings":"AAQA,MAAM,MAAM,cAAc,GAAG;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,CAAA;AAE/C,eAAO,MAAM,gBAAgB,6HA0B5B,CAAA;AAED,eAAO,MAAM,gBAAgB;UAC2B,OAAO;mBAAiB,MAAM;sBAOrF,CAAA;AAED,MAAM,MAAM,WAAW,GAAG;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAAA;AACpE,MAAM,MAAM,eAAe,GAAG;IAAE,OAAO,EAAE,WAAW,CAAA;CAAE,CAAA;AAEtD,eAAO,MAAM,aAAa;YACgC,OAAO;gBAAc,MAAM;mBAOpF,CAAA;AAED,MAAM,MAAM,cAAc,GAAG;IAC3B,gBAAgB,EAAE;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,aAAa,EAAE,MAAM,CAAA;KAAE,CAAA;IAC1D,aAAa,EAAE;QAAE,MAAM,EAAE,OAAO,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAA;CACvD,CAAA;AAED,eAAO,MAAM,aAAa,+GASzB,CAAA;AACD,eAAO,MAAM,UAAU,qGAA0E,CAAA;AACjG,eAAO,MAAM,QAAQ,mGAAuE,CAAA"}

package/dist/examples/{bundle → normal-workflow}/activities.js

@@ -1,6 +1,6 @@
 import kebabCase from 'lodash/kebabCase.js';
-import { createActivity, } from '@primero.ai/temporal-graph-tools';
-export const fetchUserProfile = createActivity(async (input) => {
+import { createActivity, wrapActivityWithMastraContext, } from '@primero.ai/temporal-graph-tools';
+export const fetchUserProfile = createActivity(wrapActivityWithMastraContext(async (input) => {
     const profileName = `User ${input.userId.slice(0, 6)}`;
     const slug = kebabCase(profileName);
     console.log(`[fetchUserProfile] fetching ${input.userId} as ${slug}`);
@@ -11,7 +11,10 @@ export const fetchUserProfile = createActivity(async (input) => {
             slug,
         },
     };
-}, {
+}, async ({ context, execute }) => {
+    console.log(`[mastra] workflow=${context === null || context === void 0 ? void 0 : context.workflow.name} node=${context === null || context === void 0 ? void 0 : context.node.key} stage=${context === null || context === void 0 ? void 0 : context.node.stageIndex}`);
+    return execute();
+}), {
     id: 'fetchUserProfile',
 });
 export const sendWelcomeEmail = createActivity(async ({ profile }) => {

package/dist/examples/normal-workflow/workflows.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflows.d.ts","sourceRoot":"","sources":["../../../examples/normal-workflow/workflows.ts"],"names":[],"mappings":"AAWA,eAAO,MAAM,iBAAiB,gEAOnB,CAAA;AAEX,eAAO,MAAM,iBAAiB,gEAGnB,CAAA"}

package/dist/examples/trigger-mastra-workflow.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=trigger-mastra-workflow.d.ts.map

package/dist/examples/trigger-mastra-workflow.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trigger-mastra-workflow.d.ts","sourceRoot":"","sources":["../../examples/trigger-mastra-workflow.ts"],"names":[],"mappings":""}

package/dist/examples/trigger-mastra-workflow.js

@@ -0,0 +1,42 @@
+import { Connection, WorkflowClient } from '@temporalio/client';
+import { builderMastraAiSupportWorkflow } from './mastra-ai-workflow/workflows';
+const DEFAULT_NAMESPACE = 'default';
+const DEFAULT_TASK_QUEUE = 'default';
+async function run() {
+    var _a, _b, _c;
+    const connection = await Connection.connect({
+        address: (_a = process.env.TEMPORAL_ADDRESS) !== null && _a !== void 0 ? _a : 'localhost:7233',
+        tls: process.env.TEMPORAL_API_KEY ? {} : undefined,
+        apiKey: process.env.TEMPORAL_API_KEY,
+    });
+    const client = new WorkflowClient({
+        connection,
+        namespace: (_b = process.env.TEMPORAL_NAMESPACE) !== null && _b !== void 0 ? _b : DEFAULT_NAMESPACE,
+    });
+    const handle = await client.start(builderMastraAiSupportWorkflow.workflowName, {
+        taskQueue: (_c = process.env.TEMPORAL_TASK_QUEUE) !== null && _c !== void 0 ? _c : DEFAULT_TASK_QUEUE,
+        workflowId: `mastra-ai-support-${Date.now()}`,
+        args: [
+            {
+                ticketId: `ticket-${Date.now()}`,
+                client: {
+                    name: 'Acme Ops',
+                    accountId: 'acct-2048',
+                    site: 'HQ - 3rd Floor',
+                },
+                outageDescription: 'Internet is not working on our office floor Wi-Fi. ISP status page is green and there is no service outage. Looks like a local router issue.',
+            },
+        ],
+    });
+    console.log('Mastra AI Workflow started:', {
+        workflowId: handle.workflowId,
+        runId: handle.firstExecutionRunId,
+    });
+    const result = await handle.result();
+    console.log('Mastra AI Workflow completed with result:', result);
+    await connection.close();
+}
+run().catch((error) => {
+    console.error('Failed to run mastra workflow trigger:', error);
+    process.exitCode = 1;
+});

package/dist/examples/trigger-normal-workflow.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=trigger-normal-workflow.d.ts.map

package/dist/examples/trigger-normal-workflow.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"trigger-normal-workflow.d.ts","sourceRoot":"","sources":["../../examples/trigger-normal-workflow.ts"],"names":[],"mappings":""}

package/dist/examples/{trigger-workflows.js → trigger-normal-workflow.js}

@@ -1,5 +1,5 @@
 import { Connection, WorkflowClient } from '@temporalio/client';
-import { builderHelloWorld, builderOnboarding } from './bundle/workflows';
+import { builderHelloWorld, builderOnboarding } from './normal-workflow/workflows';
 const DEFAULT_NAMESPACE = 'default';
 const DEFAULT_TASK_QUEUE = 'default';
 async function run() {

package/dist/examples/worker.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../examples/worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,uBAAuB,EAAU,MAAM,oBAAoB,CAAA;AAItF,wBAAgB,oBAAoB,IAAI,uBAAuB,CAM9D"}
+{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../examples/worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,uBAAuB,EAAU,MAAM,oBAAoB,CAAA;AAKtF,wBAAgB,oBAAoB,IAAI,uBAAuB,CAM9D"}