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

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

@@ -9,5 +9,5 @@ export type AiClassification = {
     summary: string;
     recommendedFix: string;
 };
-export declare const classifyOutageWithAi: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<TicketIntake, AiClassification, "classifyOutageWithAi">;
+export declare const classifyOutageWithAi: import("@primero.ai/temporal-graph-tools").CreateActivity<TicketIntake, AiClassification>;
 //# sourceMappingURL=classify-outage-with-ai.d.ts.map

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":"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;AA0FD,eAAO,MAAM,oBAAoB,gIAqDhC,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,14 +1,5 @@
-var _a;
-import { openai } from '@ai-sdk/openai';
 import { z } from 'zod';
-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'}`);
-}
+import { resolveAiSelection, selectAiModel, wrapActivityWithMastraAgent, } from '@primero.ai/temporal-graph-tools';
 function inferIntent(message) {
     const normalized = message.toLowerCase();
     if (normalized.includes('invoice') ||
@@ -47,16 +38,18 @@ 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 classificationAgentContext = {};
-const classificationMastraInput = {
-    agentInput: {
+export const classifyOutageWithAi = wrapActivityWithMastraAgent(async ({ input: ticket, agent }) => {
+    const aiSelection = resolveAiSelection({
+        defaultModel: 'gpt-4o-mini',
+        defaultProvider: 'openai',
+    });
+    const classificationAgentInput = {
         id: 'mastra-outage-classifier',
         name: 'Mastra Outage Classifier',
         instructions: [
@@ -64,17 +57,8 @@ const classificationMastraInput = {
             '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';
+        model: selectAiModel(aiSelection),
+    };
     const fallbackIntent = inferIntent(ticket.outageDescription);
     const fallbackUrgency = inferUrgency(ticket.outageDescription);
     const fallbackSummary = ticket.outageDescription.slice(0, 180);
@@ -84,6 +68,7 @@ export const classifyOutageWithAi = createActivityWithMastra({
         : 'Collect diagnostics and continue support triage.';
     try {
         const parsed = await agent({
+            agentInput: classificationAgentInput,
             prompt: [
                 `Ticket ID: ${ticket.ticketId}`,
                 `Client: ${ticket.client.name} (${ticket.client.accountId}) at ${ticket.client.site}`,
@@ -93,8 +78,8 @@ export const classifyOutageWithAi = createActivityWithMastra({
             schema: classificationOutputSchema,
         });
         return {
-            provider,
-            model: modelId,
+            provider: aiSelection.provider,
+            model: aiSelection.model,
             intent: parsed.intent,
             urgency: parsed.urgency,
             summary: parsed.summary,
@@ -105,8 +90,8 @@ export const classifyOutageWithAi = createActivityWithMastra({
         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

@@ -8,5 +8,5 @@ export type DraftResponse = {
     subject: string;
     body: string;
 };
-export declare const draftResponseWithAi: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<ParallelAnalysis, DraftResponse, "draftResponseWithAi">;
+export declare const draftResponseWithAi: import("@primero.ai/temporal-graph-tools").CreateActivity<ParallelAnalysis, DraftResponse>;
 //# sourceMappingURL=draft-response-with-ai.d.ts.map

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":"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"}
+{"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,37 +1,19 @@
-var _a;
-import { openai } from '@ai-sdk/openai';
 import { z } from 'zod';
-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'}`);
-}
-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 responseAgentContext = {};
-const responseMastraInput = {
-    agentInput: {
+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: openai(modelId),
-    },
-    agentContext: responseAgentContext,
-};
-export const draftResponseWithAi = createActivityWithMastra({
-    id: 'draftResponseWithAi',
-    onMastraContext: logMastraContext,
-    mastraInput: responseMastraInput,
-}, async ({ input: analysis, agent }) => {
+        model: selectAiModel({ defaultModel: 'gpt-4o-mini' }),
+    };
     const { classifyOutageWithAi: ai, enrichOutageDescription: enriched } = analysis;
     const fallbackSubject = `[${ai.intent.toUpperCase()}][${ai.urgency}] Internet Incident`;
     const fallbackBody = [
@@ -46,6 +28,7 @@ export const draftResponseWithAi = createActivityWithMastra({
     ].join('\n');
     try {
         const parsed = await agent({
+            agentInput: responseAgentInput,
             prompt: [
                 `Client: ${enriched.client.name} (${enriched.client.accountId}) at ${enriched.client.site}`,
                 `Classification Intent: ${ai.intent}`,

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

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

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

@@ -1 +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"}
+{"version":3,"file":"enrich-outage-description.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/enrich-outage-description.ts"],"names":[],"mappings":"AAAA,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;AAkBD,wBAAsB,uBAAuB,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAchG"}

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

@@ -1,11 +1,3 @@
-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);
@@ -18,10 +10,7 @@ function inferRootCause(message) {
     }
     return 'unknown';
 }
-export const enrichOutageDescription = createActivityWithMastra({
-    id: 'enrichOutageDescription',
-    onMastraContext: logMastraContext,
-}, async ({ input: ticket }) => {
+export async function enrichOutageDescription(ticket) {
     const rootCause = inferRootCause(ticket.outageDescription);
     const rewrittenDescription = [
         `Client ${ticket.client.name} (${ticket.client.accountId}) at ${ticket.client.site} reports internet connectivity failure.`,
@@ -34,4 +23,4 @@ export const enrichOutageDescription = createActivityWithMastra({
         rewrittenDescription,
         inferredRootCause: rootCause,
     };
-});
+}

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

@@ -3,8 +3,8 @@ 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 { AiClassification } from './classify-outage-with-ai';
+export type { DraftResponse } 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';

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

@@ -1 +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"}
+{"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,MAAM,2BAA2B,CAAA;AACjE,YAAY,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AAC7D,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/submit-response.d.ts

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

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

@@ -1 +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,4HAa1B,CAAA"}
+{"version":3,"file":"submit-response.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/submit-response.ts"],"names":[],"mappings":"AAAA,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;AAED,wBAAsB,cAAc,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAKrF"}

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

@@ -1,15 +1,6 @@
-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 async function submitResponse(draft) {
+    return {
+        ready: true,
+        response: draft,
+    };
 }
-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

@@ -12,5 +12,5 @@ export type TicketIntake = {
     client: SupportTicketInput['client'];
     outageDescription: string;
 };
-export declare const takeTicket: import("@primero.ai/temporal-graph-tools").ConfiguredActivityReference<SupportTicketInput, TicketIntake, "takeTicket">;
+export declare function takeTicket(input: SupportTicketInput): Promise<TicketIntake>;
 //# sourceMappingURL=take-ticket.d.ts.map

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

@@ -1 +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"}
+{"version":3,"file":"take-ticket.d.ts","sourceRoot":"","sources":["../../../../examples/mastra-ai-workflow/activities/take-ticket.ts"],"names":[],"mappings":"AAAA,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;AAED,wBAAsB,UAAU,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,YAAY,CAAC,CAMjF"}

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

@@ -1,16 +1,7 @@
-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 async function takeTicket(input) {
+    return {
+        ticketId: input.ticketId,
+        client: input.client,
+        outageDescription: input.outageDescription.trim(),
+    };
 }
-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/workflow.d.ts

@@ -0,0 +1,3 @@
+import { type SubmittedResponse, type SupportTicketInput } from './activities';
+export declare const mastraAiSupportWorkflow: import("@mastra/core/workflows").Workflow<import("@mastra/core/workflows").DefaultEngineType, import("@mastra/core/workflows").Step<string, unknown, unknown, unknown, unknown, unknown, any, unknown>[], "mastra-ai-support-orchestration", unknown, SupportTicketInput, SubmittedResponse, SubmittedResponse, unknown>;
+//# sourceMappingURL=workflow.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"workflow.d.ts","sourceRoot":"","sources":["../../../examples/mastra-ai-workflow/workflow.ts"],"names":[],"mappings":"AAEA,OAAO,EASL,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EAExB,MAAM,cAAc,CAAA;AAgGrB,eAAO,MAAM,uBAAuB,0TASzB,CAAA"}

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

@@ -0,0 +1,95 @@
+import { createStep, createWorkflow } from '@mastra/core/workflows';
+import { z } from 'zod';
+import { classifyOutageWithAi, draftResponseWithAi, enrichOutageDescription, submitResponse, takeTicket, } from './activities';
+const supportTicketInputSchema = z.object({
+    ticketId: z.string(),
+    client: z.object({
+        name: z.string(),
+        accountId: z.string(),
+        site: z.string(),
+    }),
+    outageDescription: z.string(),
+});
+const ticketIntakeSchema = z.object({
+    ticketId: z.string(),
+    client: z.object({
+        name: z.string(),
+        accountId: z.string(),
+        site: z.string(),
+    }),
+    outageDescription: z.string(),
+});
+const aiClassificationSchema = z.object({
+    provider: z.string(),
+    model: z.string(),
+    intent: z.enum(['connectivity', 'billing', 'account']),
+    urgency: z.enum(['low', 'medium', 'high']),
+    summary: z.string(),
+    recommendedFix: z.string(),
+});
+const enrichedDescriptionSchema = z.object({
+    client: z.object({
+        name: z.string(),
+        accountId: z.string(),
+        site: z.string(),
+    }),
+    rewrittenDescription: z.string(),
+    inferredRootCause: z.enum(['local-network', 'provider-outage', 'unknown']),
+});
+const parallelAnalysisSchema = z.object({
+    classifyOutageWithAi: aiClassificationSchema,
+    enrichOutageDescription: enrichedDescriptionSchema,
+});
+const draftResponseSchema = z.object({
+    subject: z.string(),
+    body: z.string(),
+});
+const submittedResponseSchema = z.object({
+    ready: z.literal(true),
+    response: draftResponseSchema,
+});
+const takeTicketStep = createStep({
+    id: 'takeTicket',
+    description: 'Normalize and validate incoming support ticket payload.',
+    inputSchema: supportTicketInputSchema,
+    outputSchema: ticketIntakeSchema,
+    execute: async ({ inputData }) => takeTicket(inputData),
+});
+const classifyOutageWithAiStep = createStep({
+    id: 'classifyOutageWithAi',
+    description: 'Classify outage intent and urgency with AI or fallback heuristics.',
+    inputSchema: ticketIntakeSchema,
+    outputSchema: aiClassificationSchema,
+    execute: async ({ inputData }) => classifyOutageWithAi(inputData),
+});
+const enrichOutageDescriptionStep = createStep({
+    id: 'enrichOutageDescription',
+    description: 'Enrich outage narrative with inferred root cause context.',
+    inputSchema: ticketIntakeSchema,
+    outputSchema: enrichedDescriptionSchema,
+    execute: async ({ inputData }) => enrichOutageDescription(inputData),
+});
+const draftResponseWithAiStep = createStep({
+    id: 'draftResponseWithAi',
+    description: 'Draft a client-ready response from the parallel analysis outputs.',
+    inputSchema: parallelAnalysisSchema,
+    outputSchema: draftResponseSchema,
+    execute: async ({ inputData }) => draftResponseWithAi(inputData),
+});
+const submitResponseStep = createStep({
+    id: 'submitResponse',
+    description: 'Wrap draft response as a final submitted payload.',
+    inputSchema: draftResponseSchema,
+    outputSchema: submittedResponseSchema,
+    execute: async ({ inputData }) => submitResponse(inputData),
+});
+export const mastraAiSupportWorkflow = createWorkflow({
+    id: 'mastra-ai-support-orchestration',
+    inputSchema: supportTicketInputSchema,
+    outputSchema: submittedResponseSchema,
+})
+    .then(takeTicketStep)
+    .parallel([classifyOutageWithAiStep, enrichOutageDescriptionStep])
+    .then(draftResponseWithAiStep)
+    .then(submitResponseStep)
+    .commit();

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

@@ -1,21 +1,17 @@
-import { Connection, WorkflowClient } from '@temporalio/client';
-import { builderMastraAiSupportWorkflow } from './mastra-ai-workflow/workflows';
-const DEFAULT_NAMESPACE = 'default';
-const DEFAULT_TASK_QUEUE = 'default';
+import { Client, Connection } from '@temporalio/client';
 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,
+        address: process.env.TEMPORAL_ADDRESS,
         apiKey: process.env.TEMPORAL_API_KEY,
+        tls: {},
     });
-    const client = new WorkflowClient({
+    const client = new Client({
         connection,
-        namespace: (_b = process.env.TEMPORAL_NAMESPACE) !== null && _b !== void 0 ? _b : DEFAULT_NAMESPACE,
+        namespace: process.env.TEMPORAL_NAMESPACE,
     });
-    const handle = await client.start(builderMastraAiSupportWorkflow.workflowName, {
-        taskQueue: (_c = process.env.TEMPORAL_TASK_QUEUE) !== null && _c !== void 0 ? _c : DEFAULT_TASK_QUEUE,
+    const handle = await client.workflow.start('mastraAiSupportWorkflow', {
         workflowId: `mastra-ai-support-${Date.now()}`,
+        taskQueue: process.env.TEMPORAL_TASK_QUEUE || 'default',
         args: [
             {
                 ticketId: `ticket-${Date.now()}`,
@@ -28,15 +24,14 @@ async function run() {
             },
         ],
     });
-    console.log('Mastra AI Workflow started:', {
+    const result = await handle.result();
+    console.log('Temporal Mastra AI workflow completed:', {
         workflowId: handle.workflowId,
         runId: handle.firstExecutionRunId,
+        result,
     });
-    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);
+    console.error('Failed to trigger Temporal workflow:', error);
     process.exitCode = 1;
 });

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;AAKtF,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;AAStF,wBAAgB,oBAAoB,IAAI,uBAAuB,CAM9D"}

package/dist/examples/worker.js

@@ -1,7 +1,7 @@
 import { NativeConnection, Worker } from '@temporalio/worker';
-import { bundleWorkflows, loadActivitiesFromBundle } from '@primero.ai/temporal-graph-tools';
-import { builderMastraAiSupportWorkflow } from './mastra-ai-workflow/workflows';
-import { builderHelloWorld, builderOnboarding } from './normal-workflow/workflows';
+import { bundleWorkflows, compileMastraWorkflow, loadActivitiesFromBundle, } from '@primero.ai/temporal-graph-tools';
+import * as activities from './mastra-ai-workflow/activities';
+import { mastraAiSupportWorkflow } from './mastra-ai-workflow/workflow';
 export function getConnectionOptions() {
     return {
         address: process.env.TEMPORAL_ADDRESS,
@@ -10,29 +10,38 @@ export function getConnectionOptions() {
     };
 }
 async function run() {
-    const { workflowBundle, activityBundle } = await bundleWorkflows([builderHelloWorld, builderOnboarding, builderMastraAiSupportWorkflow], {
+    var _a;
+    const bundleSigningKey = (_a = process.env.TEMPORAL_ACTIVITY_BUNDLE_KEY) === null || _a === void 0 ? void 0 : _a.trim();
+    if (!bundleSigningKey) {
+        throw new Error('TEMPORAL_ACTIVITY_BUNDLE_KEY is required to sign and verify activity bundles.');
+    }
+    const compiledMastraWorkflow = compileMastraWorkflow(mastraAiSupportWorkflow, {
+        workflowName: 'mastraAiSupportWorkflow',
+        stepActivities: {
+            takeTicket: activities.takeTicket,
+            classifyOutageWithAi: activities.classifyOutageWithAi,
+            enrichOutageDescription: activities.enrichOutageDescription,
+            draftResponseWithAi: activities.draftResponseWithAi,
+            submitResponse: activities.submitResponse,
+        },
+        proxyOptions: { startToCloseTimeout: '2 minutes' },
+    });
+    const { workflowBundle, activityBundle } = await bundleWorkflows([compiledMastraWorkflow], {
         activityBundle: {
             filename: 'examples.activities.bundle.cjs',
-            entrypoints: [
-                './examples/normal-workflow/activities.ts',
-                './examples/mastra-ai-workflow/activities/index.ts',
-            ],
+            entrypoints: ['./examples/mastra-ai-workflow/activities/index.ts'],
+            signingKey: bundleSigningKey,
         },
     });
-    if (!activityBundle.integrity) {
-        throw new Error('Expected bundleWorkflows() to return an integrity-protected activity bundle.');
-    }
-    console.log('Built activity bundle:', {
-        filename: activityBundle.filename,
-        integrity: activityBundle.integrity,
+    const bundledActivities = await loadActivitiesFromBundle(activityBundle, {
+        verificationKey: bundleSigningKey,
     });
-    const activities = await loadActivitiesFromBundle(activityBundle);
     const connection = await NativeConnection.connect(getConnectionOptions());
     const worker = await Worker.create({
         connection,
         namespace: process.env.TEMPORAL_NAMESPACE,
         taskQueue: process.env.TEMPORAL_TASK_QUEUE || 'default',
-        activities,
+        activities: bundledActivities,
         workflowBundle,
     });
     await worker.run();

package/dist/src/bundler.d.ts

@@ -1,9 +1,11 @@
+import { type MastraWorkflowCompilationInput, type MastraWorkflowLike } from './mastra/workflow-compiler.js';
 import type { WorkflowBuildResult } from './types.js';
 import type { ActivityImplementations } from './workflow-bundler.js';
 export type BundleWorkflowsOptions = {
     filename?: string;
     activityBundle?: BundleActivitiesOptions;
 };
+export type BundleWorkflowInput = WorkflowBuildResult | MastraWorkflowLike | MastraWorkflowCompilationInput;
 export type BundleWorkflowsResult = {
     activityBundle: BundledActivitiesArtifact;
     workflowBundle: {
@@ -14,13 +16,18 @@ export type BundleActivitiesOptions = {
     entrypoints?: readonly string[];
     filename?: string;
     externals?: readonly string[];
+    signingKey?: string;
 };
 export type BundledActivitiesArtifact = {
     filename: string;
     code: string;
     map?: string;
     integrity?: string;
+    signature?: string;
 };
-export declare function loadActivitiesFromBundle(bundle: BundledActivitiesArtifact): Promise<ActivityImplementations>;
-export declare function bundleWorkflows(plans: readonly WorkflowBuildResult[], options?: BundleWorkflowsOptions): Promise<BundleWorkflowsResult>;
+export type LoadActivitiesFromBundleOptions = {
+    verificationKey?: string;
+};
+export declare function loadActivitiesFromBundle(bundle: BundledActivitiesArtifact, options?: LoadActivitiesFromBundleOptions): Promise<ActivityImplementations>;
+export declare function bundleWorkflows(plans: readonly BundleWorkflowInput[], options?: BundleWorkflowsOptions): Promise<BundleWorkflowsResult>;
 //# sourceMappingURL=bundler.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"bundler.d.ts","sourceRoot":"","sources":["../../src/bundler.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAkB,mBAAmB,EAA0B,MAAM,YAAY,CAAA;AAC7F,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAA;AAIpE,MAAM,MAAM,sBAAsB,GAAG;IACnC,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,cAAc,CAAC,EAAE,uBAAuB,CAAA;CACzC,CAAA;AAED,MAAM,MAAM,qBAAqB,GAAG;IAClC,cAAc,EAAE,yBAAyB,CAAA;IACzC,cAAc,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAA;CACjC,CAAA;AAED,MAAM,MAAM,uBAAuB,GAAG;IACpC,WAAW,CAAC,EAAE,SAAS,MAAM,EAAE,CAAA;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,SAAS,CAAC,EAAE,SAAS,MAAM,EAAE,CAAA;CAC9B,CAAA;AAED,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,EAAE,MAAM,CAAA;IAChB,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,CAAC,EAAE,MAAM,CAAA;IACZ,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB,CAAA;AAoCD,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,yBAAyB,GAChC,OAAO,CAAC,uBAAuB,CAAC,CAiDlC;AAED,wBAAsB,eAAe,CACnC,KAAK,EAAE,SAAS,mBAAmB,EAAE,EACrC,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,qBAAqB,CAAC,CAsChC"}
+{"version":3,"file":"bundler.d.ts","sourceRoot":"","sources":["../../src/bundler.ts"],"names":[],"mappings":"AAMA,OAAO,EAEL,KAAK,8BAA8B,EACnC,KAAK,kBAAkB,EACxB,MAAM,+BAA+B,CAAA;AACtC,OAAO,KAAK,EAAkB,mBAAmB,EAA0B,MAAM,YAAY,CAAA;AAC7F,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAA;AAIpE,MAAM,MAAM,sBAAsB,GAAG;IACnC,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,cAAc,CAAC,EAAE,uBAAuB,CAAA;CACzC,CAAA;AAED,MAAM,MAAM,mBAAmB,GAC3B,mBAAmB,GACnB,kBAAkB,GAClB,8BAA8B,CAAA;AAElC,MAAM,MAAM,qBAAqB,GAAG;IAClC,cAAc,EAAE,yBAAyB,CAAA;IACzC,cAAc,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAA;CACjC,CAAA;AAED,MAAM,MAAM,uBAAuB,GAAG;IACpC,WAAW,CAAC,EAAE,SAAS,MAAM,EAAE,CAAA;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,SAAS,CAAC,EAAE,SAAS,MAAM,EAAE,CAAA;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB,CAAA;AAED,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,EAAE,MAAM,CAAA;IAChB,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,CAAC,EAAE,MAAM,CAAA;IACZ,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB,CAAA;AAED,MAAM,MAAM,+BAA+B,GAAG;IAC5C,eAAe,CAAC,EAAE,MAAM,CAAA;CACzB,CAAA;AAoCD,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,yBAAyB,EACjC,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,uBAAuB,CAAC,CAkDlC;AAED,wBAAsB,eAAe,CACnC,KAAK,EAAE,SAAS,mBAAmB,EAAE,EACrC,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,qBAAqB,CAAC,CAuChC"}