@salesforce/magen-mcp-workflow 0.0.4 → 0.0.5

package/dist/common/metadata.d.ts

@@ -80,6 +80,22 @@ export interface NodeGuidanceData<TResultSchema extends z.ZodObject<z.ZodRawShap
      * LLM compliance with the expected structure.
      */
     exampleOutput?: string;
+    /**
+     * Optional custom guidance for the LLM to return results to the orchestrator.
+     *
+     * When provided, this replaces the orchestrator's default "return to orchestrator"
+     * prompt. The function receives only `workflowStateData` (the runtime session state
+     * that the producer doesn't have at construction time). The producer already owns
+     * `resultSchema` and `exampleOutput` as sibling properties on this same struct,
+     * so they can be captured in the closure if needed.
+     *
+     * Ensure this custom guidance properly instructs the LLM to return the workflow
+     * to the orchestrator, or the workflow will likely be broken.
+     *
+     * @param workflowStateData - The workflow state data to round-trip back to the orchestrator
+     * @returns The return guidance prompt string
+     */
+    returnGuidance?: (workflowStateData: WorkflowStateData) => string;
 }
 /**
  * Union type for all interrupt data types.

package/dist/common/metadata.js.map

@@ -1 +1 @@
-{"version":3,"file":"metadata.js","sourceRoot":"","sources":["../../src/common/metadata.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB;;;GAGG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;CAC7E,CAAC,CAAC;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,iBAAiB,EAAE,mBAAmB;IACtC,SAAS,EAAE,WAAW;CACd,CAAC;AAEX;;;GAGG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,CAAC,uBAAuB,CAAC,iBAAiB,CAAC,EAAE,0BAA0B,CAAC,QAAQ,CAC9E,kKAAkK,CACnK;CACF,CAAC,CAAC;AA0DH;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAGhC,IAAgD;IAChD,OAAO,cAAc,IAAI,IAAI,IAAI,cAAc,IAAI,IAAI,IAAI,QAAQ,IAAI,IAAI,CAAC;AAC9E,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,YAAY,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,CAAC,gEAAgE,CAAC;IAC7E,YAAY,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,CAAC,8EAA8E,CAAC;CAC5F,CAAC,CAAC"}
+{"version":3,"file":"metadata.js","sourceRoot":"","sources":["../../src/common/metadata.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB;;;GAGG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;CAC7E,CAAC,CAAC;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,iBAAiB,EAAE,mBAAmB;IACtC,SAAS,EAAE,WAAW;CACd,CAAC;AAEX;;;GAGG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,CAAC,uBAAuB,CAAC,iBAAiB,CAAC,EAAE,0BAA0B,CAAC,QAAQ,CAC9E,kKAAkK,CACnK;CACF,CAAC,CAAC;AA0EH;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAGhC,IAAgD;IAChD,OAAO,cAAc,IAAI,IAAI,IAAI,cAAc,IAAI,IAAI,IAAI,QAAQ,IAAI,IAAI,CAAC;AAC9E,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,YAAY,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,CAAC,gEAAgE,CAAC;IAC7E,YAAY,EAAE,CAAC;SACZ,MAAM,EAAE;SACR,QAAQ,CAAC,8EAA8E,CAAC;CAC5F,CAAC,CAAC"}

package/dist/index.d.ts

@@ -9,9 +9,9 @@ export { executeToolWithLogging } from './utils/toolExecutionUtils.js';
 export { AbstractTool, AbstractWorkflowTool } from './tools/base/index.js';
 export { GET_INPUT_WORKFLOW_RESULT_SCHEMA } from './tools/utilities/index.js';
 export { type BaseGraphConfig, type WorkflowRunnableConfig } from './common/graphConfig.js';
-export { BaseNode, AbstractToolNode, createGetUserInputNode, createUserInputExtractionNode, type GetUserInputNodeOptions, type UserInputExtractionNodeOptions, } from './nodes/index.js';
+export { BaseNode, AbstractToolNode, UserInputExtractionNode, createGetUserInputNode, createUserInputExtractionNode, type GetUserInputNodeOptions, type UserInputExtractionNodeOptions, } from './nodes/index.js';
 export { CheckPropertiesFulfilledRouter } from './routers/index.js';
-export { AbstractService } from './services/index.js';
+export { AbstractService, InputExtractionService } from './services/index.js';
 export { JsonCheckpointSaver, WorkflowStatePersistence, WorkflowStateManager, type WorkflowStateManagerConfig, type WorkflowEnvironment, } from './checkpointing/index.js';
-export { OrchestratorTool, createOrchestratorToolMetadata, type OrchestratorConfig, type OrchestratorInput, type OrchestratorOutput, type OrchestratorToolMetadata, ORCHESTRATOR_INPUT_SCHEMA, ORCHESTRATOR_OUTPUT_SCHEMA, } from './tools/orchestrator/index.js';
+export { OrchestratorTool, createOrchestratorToolMetadata, type DefaultOrchestratorInputSchema, type OrchestratorConfig, type OrchestratorInput, type OrchestratorOutput, type OrchestratorToolMetadata, ORCHESTRATOR_INPUT_SCHEMA, ORCHESTRATOR_OUTPUT_SCHEMA, } from './tools/orchestrator/index.js';
 export { ProgressReporter, MCPProgressReporter, CommandRunner, DefaultCommandRunner, type Command, type CommandResult, type ProgressParseResult, type ProgressParser, type CommandExecutionOptions, } from './execution/index.js';

package/dist/index.js

@@ -21,11 +21,11 @@ export { AbstractTool, AbstractWorkflowTool } from './tools/base/index.js';
 // Get Input Result Schema (for NodeGuidanceData / direct guidance mode services)
 export { GET_INPUT_WORKFLOW_RESULT_SCHEMA } from './tools/utilities/index.js';
 // Base Node Classes
-export { BaseNode, AbstractToolNode, createGetUserInputNode, createUserInputExtractionNode, } from './nodes/index.js';
+export { BaseNode, AbstractToolNode, UserInputExtractionNode, createGetUserInputNode, createUserInputExtractionNode, } from './nodes/index.js';
 // Routers
 export { CheckPropertiesFulfilledRouter } from './routers/index.js';
 // Base Service Classes
-export { AbstractService } from './services/index.js';
+export { AbstractService, InputExtractionService } from './services/index.js';
 // Checkpointing Infrastructure
 export { JsonCheckpointSaver, WorkflowStatePersistence, WorkflowStateManager, } from './checkpointing/index.js';
 // Orchestrator Tool

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,iCAAiC;AACjC,OAAO,EAA6B,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAE7F,iCAAiC;AACjC,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAEhB,yBAAyB,GAC1B,MAAM,iCAAiC,CAAC;AAEzC,UAAU;AACV,OAAO,EAEL,UAAU,EACV,YAAY,EACZ,qBAAqB,EACrB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAE7B,kBAAkB;AAClB,OAAO,EACL,0BAA0B,EAC1B,uBAAuB,EACvB,+BAA+B,EAC/B,+BAA+B,EAK/B,kBAAkB,GAInB,MAAM,sBAAsB,CAAC;AAU9B,gCAAgC;AAChC,OAAO,EAAqB,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAEnF,uBAAuB;AACvB,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,oBAAoB;AACpB,OAAO,EAAE,YAAY,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE3E,iFAAiF;AACjF,OAAO,EAAE,gCAAgC,EAAE,MAAM,4BAA4B,CAAC;AAK9E,oBAAoB;AACpB,OAAO,EACL,QAAQ,EACR,gBAAgB,EAChB,sBAAsB,EACtB,6BAA6B,GAG9B,MAAM,kBAAkB,CAAC;AAE1B,UAAU;AACV,OAAO,EAAE,8BAA8B,EAAE,MAAM,oBAAoB,CAAC;AAEpE,uBAAuB;AACvB,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,+BAA+B;AAC/B,OAAO,EACL,mBAAmB,EACnB,wBAAwB,EACxB,oBAAoB,GAGrB,MAAM,0BAA0B,CAAC;AAElC,oBAAoB;AACpB,OAAO,EACL,gBAAgB,EAChB,8BAA8B,EAK9B,yBAAyB,EACzB,0BAA0B,GAC3B,MAAM,+BAA+B,CAAC;AAEvC,2BAA2B;AAC3B,OAAO,EAEL,mBAAmB,EAEnB,oBAAoB,GAMrB,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,iCAAiC;AACjC,OAAO,EAA6B,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAE7F,iCAAiC;AACjC,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAEhB,yBAAyB,GAC1B,MAAM,iCAAiC,CAAC;AAEzC,UAAU;AACV,OAAO,EAEL,UAAU,EACV,YAAY,EACZ,qBAAqB,EACrB,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAE7B,kBAAkB;AAClB,OAAO,EACL,0BAA0B,EAC1B,uBAAuB,EACvB,+BAA+B,EAC/B,+BAA+B,EAK/B,kBAAkB,GAInB,MAAM,sBAAsB,CAAC;AAU9B,gCAAgC;AAChC,OAAO,EAAqB,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAEnF,uBAAuB;AACvB,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAEvE,oBAAoB;AACpB,OAAO,EAAE,YAAY,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE3E,iFAAiF;AACjF,OAAO,EAAE,gCAAgC,EAAE,MAAM,4BAA4B,CAAC;AAK9E,oBAAoB;AACpB,OAAO,EACL,QAAQ,EACR,gBAAgB,EAChB,uBAAuB,EACvB,sBAAsB,EACtB,6BAA6B,GAG9B,MAAM,kBAAkB,CAAC;AAE1B,UAAU;AACV,OAAO,EAAE,8BAA8B,EAAE,MAAM,oBAAoB,CAAC;AAEpE,uBAAuB;AACvB,OAAO,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAE9E,+BAA+B;AAC/B,OAAO,EACL,mBAAmB,EACnB,wBAAwB,EACxB,oBAAoB,GAGrB,MAAM,0BAA0B,CAAC;AAElC,oBAAoB;AACpB,OAAO,EACL,gBAAgB,EAChB,8BAA8B,EAM9B,yBAAyB,EACzB,0BAA0B,GAC3B,MAAM,+BAA+B,CAAC;AAEvC,2BAA2B;AAC3B,OAAO,EAEL,mBAAmB,EAEnB,oBAAoB,GAMrB,MAAM,sBAAsB,CAAC"}

package/dist/services/inputExtractionService.d.ts

@@ -51,5 +51,8 @@ export declare class InputExtractionService extends AbstractService implements I
      * @param propertiesToExtract - Array of properties to extract
      * @returns The guidance prompt string
      */
-    private generateTaskGuidance;
+    protected generateTaskGuidance(userUtterance: unknown, propertiesToExtract: Array<{
+        propertyName: string;
+        description: string;
+    }>): string;
 }

package/dist/services/inputExtractionService.js.map

@@ -1 +1 @@
-{"version":3,"file":"inputExtractionService.js","sourceRoot":"","sources":["../../src/services/inputExtractionService.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AA4BvD;;;;;;GAMG;AACH,MAAM,OAAO,sBACX,SAAQ,eAAe;IAWJ;IARnB;;;;;;OAMG;IACH,YACmB,MAAc,EAC/B,YAA2B,EAC3B,MAAe;QAEf,KAAK,CAAC,wBAAwB,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAJrC,WAAM,GAAN,MAAM,CAAQ;IAKjC,CAAC;IAED,iBAAiB,CAAC,SAAkB,EAAE,UAAsC;QAC1E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;YAChD,SAAS;YACT,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM;SAC9C,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,IAAI,CAAC,8BAA8B,CAAC,UAAU,CAAC,CAAC;QAC5E,MAAM,YAAY,GAAG,IAAI,CAAC,4BAA4B,CAAC,UAAU,CAAC,CAAC;QAEnE,0EAA0E;QAC1E,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,MAAM,CAClD,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACZ,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,cAAc,IAAI,CAAC,YAAY,iBAAiB,CAAC;YAC1E,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAA4B,CAC7B,CAAC;QAEF,mDAAmD;QACnD,MAAM,gBAAgB,GAA0C;YAC9D,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,mBAAmB,CAAC;YACvE,YAAY,EAAE,YAAY;YAC1B,kFAAkF;YAClF,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,CAAC;SAC1E,CAAC;QAEF,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CACjD,gBAAgB,EAChB,YAAY,EACZ,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,CACnF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;YAChD,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,MAAM;YACvE,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC;SAC7D,CAAC,CAAC;QAEH,OAAO,eAAe,CAAC;IACzB,CAAC;IAEO,8BAA8B,CACpC,UAAsC;QAEtC,MAAM,mBAAmB,GAAyD,EAAE,CAAC;QAErF,KAAK,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,mBAAmB,CAAC,IAAI,CAAC;gBACvB,YAAY;gBACZ,WAAW,EAAE,QAAQ,CAAC,WAAW;aAClC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;YACtD,KAAK,EAAE,mBAAmB,CAAC,MAAM;YACjC,UAAU,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;SACzD,CAAC,CAAC;QAEH,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAEO,4BAA4B,CAClC,UAAsC;QAEtC,MAAM,wBAAwB,GAA8B,EAAE,CAAC;QAE/D,KAAK,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,wBAAwB,CAAC,YAAY,CAAC,GAAG,QAAQ,CAAC,OAAO;iBACtD,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;iBAC9B,QAAQ,EAAE;iBACV,KAAK,CAAC,CAAC,GAAuB,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CAAC,EAAE,mBAAmB,EAAE,CAAC,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IAC7F,CAAC;IAEO,uBAAuB,CAC7B,SAAkB,EAClB,UAAsC,EACtC,YAA8E;QAE9E,MAAM,kBAAkB,GAAG,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACzD,MAAM,EAAE,mBAAmB,EAAE,GAAG,kBAAkB,CAAC;QAEnD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;YACnD,mBAAmB;SACpB,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAA4B,EAAE,CAAC;QACxD,MAAM,iBAAiB,GAAa,EAAE,CAAC;QAEvC,KAAK,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAChF,IAAI,aAAa,IAAI,IAAI,EAAE,CAAC;gBAC1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6CAA6C,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;gBACnF,SAAS;YACX,CAAC;YAED,MAAM,gBAAgB,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;YAClD,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC5E,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,cAAc,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBACrE,mBAAmB,CAAC,YAAY,CAAC,GAAG,cAAc,CAAC;gBACnD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;oBACnD,YAAY;oBACZ,KAAK,EAAE,cAAc;iBACtB,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBACrC,IAAI,KAAK,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;oBAChC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE;wBAC9C,YAAY;wBACZ,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACxE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mCAAmC,YAAY,KAAK,QAAQ,EAAE,CAAC,CAAC;oBAClF,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,EAAE;gBACpD,iBAAiB;gBACjB,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,MAAM;aACpD,CAAC,CAAC;QACL,CAAC;QAED,OAAO,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,CAAC;IACtD,CAAC;IAED;;;;;;OAMG;IACK,oBAAoB,CAC1B,aAAsB,EACtB,mBAAyE;QAEzE,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;EAyBT,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;;;;EAI7B,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuCpC,CAAC;IACA,CAAC;CACF"}
+{"version":3,"file":"inputExtractionService.js","sourceRoot":"","sources":["../../src/services/inputExtractionService.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AA4BvD;;;;;;GAMG;AACH,MAAM,OAAO,sBACX,SAAQ,eAAe;IAWJ;IARnB;;;;;;OAMG;IACH,YACmB,MAAc,EAC/B,YAA2B,EAC3B,MAAe;QAEf,KAAK,CAAC,wBAAwB,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAJrC,WAAM,GAAN,MAAM,CAAQ;IAKjC,CAAC;IAED,iBAAiB,CAAC,SAAkB,EAAE,UAAsC;QAC1E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;YAChD,SAAS;YACT,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM;SAC9C,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,IAAI,CAAC,8BAA8B,CAAC,UAAU,CAAC,CAAC;QAC5E,MAAM,YAAY,GAAG,IAAI,CAAC,4BAA4B,CAAC,UAAU,CAAC,CAAC;QAEnE,0EAA0E;QAC1E,MAAM,iBAAiB,GAAG,mBAAmB,CAAC,MAAM,CAClD,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACZ,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,cAAc,IAAI,CAAC,YAAY,iBAAiB,CAAC;YAC1E,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAA4B,CAC7B,CAAC;QAEF,mDAAmD;QACnD,MAAM,gBAAgB,GAA0C;YAC9D,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,mBAAmB,CAAC;YACvE,YAAY,EAAE,YAAY;YAC1B,kFAAkF;YAClF,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,CAAC;SAC1E,CAAC;QAEF,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CACjD,gBAAgB,EAChB,YAAY,EACZ,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,CACnF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;YAChD,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,MAAM;YACvE,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC;SAC7D,CAAC,CAAC;QAEH,OAAO,eAAe,CAAC;IACzB,CAAC;IAEO,8BAA8B,CACpC,UAAsC;QAEtC,MAAM,mBAAmB,GAAyD,EAAE,CAAC;QAErF,KAAK,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,mBAAmB,CAAC,IAAI,CAAC;gBACvB,YAAY;gBACZ,WAAW,EAAE,QAAQ,CAAC,WAAW;aAClC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;YACtD,KAAK,EAAE,mBAAmB,CAAC,MAAM;YACjC,UAAU,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;SACzD,CAAC,CAAC;QAEH,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAEO,4BAA4B,CAClC,UAAsC;QAEtC,MAAM,wBAAwB,GAA8B,EAAE,CAAC;QAE/D,KAAK,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,wBAAwB,CAAC,YAAY,CAAC,GAAG,QAAQ,CAAC,OAAO;iBACtD,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;iBAC9B,QAAQ,EAAE;iBACV,KAAK,CAAC,CAAC,GAAuB,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CAAC,EAAE,mBAAmB,EAAE,CAAC,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IAC7F,CAAC;IAEO,uBAAuB,CAC7B,SAAkB,EAClB,UAAsC,EACtC,YAA8E;QAE9E,MAAM,kBAAkB,GAAG,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACzD,MAAM,EAAE,mBAAmB,EAAE,GAAG,kBAAkB,CAAC;QAEnD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;YACnD,mBAAmB;SACpB,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAA4B,EAAE,CAAC;QACxD,MAAM,iBAAiB,GAAa,EAAE,CAAC;QAEvC,KAAK,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAChF,IAAI,aAAa,IAAI,IAAI,EAAE,CAAC;gBAC1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6CAA6C,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;gBACnF,SAAS;YACX,CAAC;YAED,MAAM,gBAAgB,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;YAClD,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC5E,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,cAAc,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBACrE,mBAAmB,CAAC,YAAY,CAAC,GAAG,cAAc,CAAC;gBACnD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;oBACnD,YAAY;oBACZ,KAAK,EAAE,cAAc;iBACtB,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBACrC,IAAI,KAAK,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;oBAChC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE;wBAC9C,YAAY;wBACZ,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACxE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mCAAmC,YAAY,KAAK,QAAQ,EAAE,CAAC,CAAC;oBAClF,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,EAAE;gBACpD,iBAAiB;gBACjB,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,MAAM;aACpD,CAAC,CAAC;QACL,CAAC;QAED,OAAO,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,CAAC;IACtD,CAAC;IAED;;;;;;OAMG;IACO,oBAAoB,CAC5B,aAAsB,EACtB,mBAAyE;QAEzE,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;EAyBT,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;;;;EAI7B,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAuCpC,CAAC;IACA,CAAC;CACF"}

package/dist/tools/orchestrator/config.d.ts

@@ -1,10 +1,18 @@
+import z from 'zod';
 import { StateGraph } from '@langchain/langgraph';
 import { Logger } from '../../logging/logger.js';
 import { WorkflowStateManager } from '../../checkpointing/workflowStateManager.js';
+import type { DefaultOrchestratorInputSchema } from './metadata.js';
 /**
  * Orchestrator configuration interface
  *
- * Example usage:
+ * @template TInputSchema - The Zod input schema type for the orchestrator MCP tool.
+ *   Defaults to the standard ORCHESTRATOR_INPUT_SCHEMA. When providing a custom schema,
+ *   the OrchestratorTool subclass MUST also override extractUserInput() and
+ *   extractWorkflowStateData() to map the custom schema's properties to the
+ *   orchestrator's semantic needs.
+ *
+ * Example usage (default schema):
  * ```
  * const MyWorkflowState = Annotation.Root({ messages: Annotation<string[]> });
  * const workflow = new StateGraph(MyWorkflowState)
@@ -19,8 +27,24 @@ import { WorkflowStateManager } from '../../checkpointing/workflowStateManager.j
  *   workflow,
  * };
  * ```
+ *
+ * Example usage (custom schema):
+ * ```
+ * const MY_CUSTOM_SCHEMA = z.object({
+ *   payload: z.unknown().optional(),
+ *   sessionState: z.object({ thread_id: z.string() }).default({ thread_id: '' }),
+ * });
+ *
+ * const config: OrchestratorConfig<typeof MY_CUSTOM_SCHEMA> = {
+ *   toolId: 'my-orchestrator',
+ *   title: 'My Orchestrator',
+ *   description: 'Orchestrates my workflow',
+ *   workflow,
+ *   inputSchema: MY_CUSTOM_SCHEMA,
+ * };
+ * ```
  */
-export interface OrchestratorConfig {
+export interface OrchestratorConfig<TInputSchema extends z.ZodObject<z.ZodRawShape> = DefaultOrchestratorInputSchema> {
     /** Unique tool identifier for MCP registration */
     toolId: string;
     /** Extended tool title for display */
@@ -35,6 +59,17 @@ export interface OrchestratorConfig {
      * with the appropriate checkpointer.
      */
     workflow: StateGraph<any, any, any, any, any, any, any, any>;
+    /**
+     * Custom Zod input schema for the orchestrator MCP tool.
+     *
+     * Optional - defaults to ORCHESTRATOR_INPUT_SCHEMA, which provides the standard
+     * `userInput` and `workflowStateData` properties.
+     *
+     * When providing a custom schema, the OrchestratorTool subclass MUST also override
+     * `extractUserInput()` and `extractWorkflowStateData()` to map the custom schema's
+     * properties to the orchestrator's semantic needs.
+     */
+    inputSchema?: TInputSchema;
     /**
      * Workflow state manager for checkpointing and persistence
      *

package/dist/tools/orchestrator/index.d.ts

@@ -1,3 +1,3 @@
 export { type OrchestratorConfig } from './config.js';
-export { type OrchestratorInput, type OrchestratorOutput, type OrchestratorToolMetadata, ORCHESTRATOR_INPUT_SCHEMA, ORCHESTRATOR_OUTPUT_SCHEMA, createOrchestratorToolMetadata, } from './metadata.js';
+export { type DefaultOrchestratorInputSchema, type OrchestratorInput, type OrchestratorOutput, type OrchestratorToolMetadata, ORCHESTRATOR_INPUT_SCHEMA, ORCHESTRATOR_OUTPUT_SCHEMA, createOrchestratorToolMetadata, } from './metadata.js';
 export { OrchestratorTool } from './orchestratorTool.js';

package/dist/tools/orchestrator/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/orchestrator/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAIL,yBAAyB,EACzB,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/orchestrator/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAKL,yBAAyB,EACzB,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/tools/orchestrator/metadata.d.ts

@@ -85,17 +85,30 @@ export declare const ORCHESTRATOR_OUTPUT_SCHEMA: z.ZodObject<{
     orchestrationInstructionsPrompt: string;
 }>;
 export type OrchestratorOutput = z.infer<typeof ORCHESTRATOR_OUTPUT_SCHEMA>;
+/**
+ * Type alias for the default orchestrator input schema type.
+ *
+ * Used as the default generic parameter for OrchestratorConfig, OrchestratorToolMetadata,
+ * and OrchestratorTool to avoid circular imports between config.ts and metadata.ts.
+ */
+export type DefaultOrchestratorInputSchema = typeof ORCHESTRATOR_INPUT_SCHEMA;
 /**
  * Orchestrator tool metadata type
  * The metadata for the orchestrator tool (inputs/outputs)
+ *
+ * @template TInputSchema - The Zod input schema type. Defaults to ORCHESTRATOR_INPUT_SCHEMA.
  */
-export type OrchestratorToolMetadata = ToolMetadata<typeof ORCHESTRATOR_INPUT_SCHEMA, typeof ORCHESTRATOR_OUTPUT_SCHEMA>;
+export type OrchestratorToolMetadata<TInputSchema extends z.ZodObject<z.ZodRawShape> = DefaultOrchestratorInputSchema> = ToolMetadata<TInputSchema, typeof ORCHESTRATOR_OUTPUT_SCHEMA>;
 /**
- * Factory function to create orchestrator tool metadata from configuration
+ * Factory function to create orchestrator tool metadata from configuration.
  * Takes the consumer-provided config and creates the tool metadata with
- * standardized input/output schemas.
+ * the appropriate input/output schemas.
+ *
+ * When a custom inputSchema is provided in the config, it is used instead of
+ * the default ORCHESTRATOR_INPUT_SCHEMA.
  *
+ * @template TInputSchema - The Zod input schema type. Defaults to ORCHESTRATOR_INPUT_SCHEMA.
  * @param config - The orchestrator configuration
- * @returns Tool metadata with the specified toolId, title, description
+ * @returns Tool metadata with the specified toolId, title, description, and schemas
  */
-export declare function createOrchestratorToolMetadata(config: OrchestratorConfig): OrchestratorToolMetadata;
+export declare function createOrchestratorToolMetadata<TInputSchema extends z.ZodObject<z.ZodRawShape> = DefaultOrchestratorInputSchema>(config: OrchestratorConfig<TInputSchema>): OrchestratorToolMetadata<TInputSchema>;

package/dist/tools/orchestrator/metadata.js

@@ -57,19 +57,28 @@ export const ORCHESTRATOR_OUTPUT_SCHEMA = z.object({
         .describe('The prompt describing the next workflow action for the LLM to execute.'),
 });
 /**
- * Factory function to create orchestrator tool metadata from configuration
+ * Factory function to create orchestrator tool metadata from configuration.
  * Takes the consumer-provided config and creates the tool metadata with
- * standardized input/output schemas.
+ * the appropriate input/output schemas.
  *
+ * When a custom inputSchema is provided in the config, it is used instead of
+ * the default ORCHESTRATOR_INPUT_SCHEMA.
+ *
+ * @template TInputSchema - The Zod input schema type. Defaults to ORCHESTRATOR_INPUT_SCHEMA.
  * @param config - The orchestrator configuration
- * @returns Tool metadata with the specified toolId, title, description
+ * @returns Tool metadata with the specified toolId, title, description, and schemas
  */
 export function createOrchestratorToolMetadata(config) {
+    // The cast is necessary because TypeScript cannot prove that when config.inputSchema
+    // is undefined, TInputSchema must be DefaultOrchestratorInputSchema (i.e. the generic
+    // default and the runtime default independently track the same fallback). The cast is
+    // safe: when inputSchema is omitted, TInputSchema defaults to typeof ORCHESTRATOR_INPUT_SCHEMA.
+    const effectiveInputSchema = (config.inputSchema ?? ORCHESTRATOR_INPUT_SCHEMA);
     return {
         toolId: config.toolId,
         title: config.title,
         description: config.description,
-        inputSchema: ORCHESTRATOR_INPUT_SCHEMA,
+        inputSchema: effectiveInputSchema,
         outputSchema: ORCHESTRATOR_OUTPUT_SCHEMA,
     };
 }

package/dist/tools/orchestrator/metadata.js.map

@@ -1 +1 @@
-{"version":3,"file":"metadata.js","sourceRoot":"","sources":["../../../src/tools/orchestrator/metadata.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAEL,uBAAuB,EACvB,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAGlC;;;GAGG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAClD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;CACjF,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC;KAC1C,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;KAC/B,QAAQ,CAAC,8EAA8E,CAAC,CAAC;AAE5F;;;;GAIG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC;KAC/B,KAAK,CAAC,CAAC,2BAA2B,EAAE,4BAA4B,CAAC,CAAC;KAClE,QAAQ,CACP,uIAAuI,CACxI,CAAC;AAEJ;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,CAAC,uBAAuB,CAAC,SAAS,CAAC,EAAE,iBAAiB,CAAC,QAAQ,EAAE;IACjE,CAAC,uBAAuB,CAAC,iBAAiB,CAAC,EAAE,0BAA0B,CAAC,OAAO,CAAC;QAC9E,SAAS,EAAE,EAAE;KACd,CAAC,CAAC,QAAQ,CAAC,oFAAoF,CAAC;CAClG,CAAC,CAAC;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,+BAA+B,EAAE,CAAC;SAC/B,MAAM,EAAE;SACR,QAAQ,CAAC,wEAAwE,CAAC;CACtF,CAAC,CAAC;AAaH;;;;;;;GAOG;AACH,MAAM,UAAU,8BAA8B,CAC5C,MAA0B;IAE1B,OAAO;QACL,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,WAAW,EAAE,yBAAyB;QACtC,YAAY,EAAE,0BAA0B;KACzC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"metadata.js","sourceRoot":"","sources":["../../../src/tools/orchestrator/metadata.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,OAAO,EAEL,uBAAuB,EACvB,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAGlC;;;GAGG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAClD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;CACjF,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC;KAC1C,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;KAC/B,QAAQ,CAAC,8EAA8E,CAAC,CAAC;AAE5F;;;;GAIG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC;KAC/B,KAAK,CAAC,CAAC,2BAA2B,EAAE,4BAA4B,CAAC,CAAC;KAClE,QAAQ,CACP,uIAAuI,CACxI,CAAC;AAEJ;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,CAAC,uBAAuB,CAAC,SAAS,CAAC,EAAE,iBAAiB,CAAC,QAAQ,EAAE;IACjE,CAAC,uBAAuB,CAAC,iBAAiB,CAAC,EAAE,0BAA0B,CAAC,OAAO,CAAC;QAC9E,SAAS,EAAE,EAAE;KACd,CAAC,CAAC,QAAQ,CAAC,oFAAoF,CAAC;CAClG,CAAC,CAAC;AAIH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,+BAA+B,EAAE,CAAC;SAC/B,MAAM,EAAE;SACR,QAAQ,CAAC,wEAAwE,CAAC;CACtF,CAAC,CAAC;AAsBH;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,8BAA8B,CAE5C,MAAwC;IACxC,qFAAqF;IACrF,sFAAsF;IACtF,sFAAsF;IACtF,gGAAgG;IAChG,MAAM,oBAAoB,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,yBAAyB,CAAiB,CAAC;IAC/F,OAAO;QACL,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,WAAW,EAAE,oBAAoB;QACjC,YAAY,EAAE,0BAA0B;KACzC,CAAC;AACJ,CAAC"}

package/dist/tools/orchestrator/orchestratorTool.d.ts

@@ -1,11 +1,13 @@
+import z from 'zod';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import type { ServerRequest, ServerNotification } from '@modelcontextprotocol/sdk/types.js';
 import type { RequestHandlerExtra } from '@modelcontextprotocol/sdk/shared/protocol.js';
 import { AbstractTool } from '../base/abstractTool.js';
+import { MCPToolInvocationData, NodeGuidanceData, WorkflowStateData } from '../../common/metadata.js';
 import type { WorkflowRunnableConfig } from '../../common/graphConfig.js';
 import { type ProgressReporter } from '../../execution/progressReporter.js';
-import { OrchestratorConfig } from './config.js';
-import { OrchestratorInput, OrchestratorOutput, OrchestratorToolMetadata } from './metadata.js';
+import type { OrchestratorConfig } from './config.js';
+import { type OrchestratorOutput, type OrchestratorToolMetadata, type DefaultOrchestratorInputSchema } from './metadata.js';
 /**
  * Orchestrator Tool
  *
@@ -19,15 +21,15 @@ import { OrchestratorInput, OrchestratorOutput, OrchestratorToolMetadata } from
  *
  * All state management and checkpointing responsibilities are delegated to WorkflowStateManager.
  */
-export declare class OrchestratorTool extends AbstractTool<OrchestratorToolMetadata> {
+export declare class OrchestratorTool<TInputSchema extends z.ZodObject<z.ZodRawShape> = DefaultOrchestratorInputSchema> extends AbstractTool<OrchestratorToolMetadata<TInputSchema>> {
     private readonly config;
     private readonly stateManager;
     private currentProgressReporter;
-    constructor(server: McpServer, config: OrchestratorConfig);
+    constructor(server: McpServer, config: OrchestratorConfig<TInputSchema>);
     /**
      * Handle orchestrator requests - manages workflow state and execution
      */
-    handleRequest: (input: OrchestratorInput, extra: RequestHandlerExtra<ServerRequest, ServerNotification>) => Promise<{
+    handleRequest: (input: z.infer<TInputSchema>, extra: RequestHandlerExtra<ServerRequest, ServerNotification>) => Promise<{
         content: {
             type: "text";
             text: string;
@@ -45,7 +47,7 @@ export declare class OrchestratorTool extends AbstractTool<OrchestratorToolMetad
      * @returns A progress reporter instance, or undefined if progress reporting is disabled
      */
     protected createProgressReporter(extra: RequestHandlerExtra<ServerRequest, ServerNotification>): ProgressReporter | undefined;
-    protected processRequest(input: OrchestratorInput): Promise<OrchestratorOutput>;
+    protected processRequest(input: z.infer<TInputSchema>): Promise<OrchestratorOutput>;
     /**
      * Create the thread configuration for LangGraph workflow invocation.
      *
@@ -64,17 +66,46 @@ export declare class OrchestratorTool extends AbstractTool<OrchestratorToolMetad
      */
     protected getProgressReporter(): ProgressReporter | undefined;
     /**
-     * Create orchestration prompt for LLM with embedded tool invocation data and workflow state
+     * Extract the user input value from the orchestrator input.
+     *
+     * Default implementation assumes the standard ORCHESTRATOR_INPUT_SCHEMA structure
+     * and reads the `userInput` property directly. Subclasses with custom input schemas
+     * MUST override this method to map their custom properties.
+     *
+     * @param input - The parsed orchestrator input
+     * @returns The user input value, or undefined if not present
+     */
+    protected extractUserInput(input: z.infer<TInputSchema>): unknown | undefined;
+    /**
+     * Extract the workflow state data from the orchestrator input.
+     *
+     * Default implementation assumes the standard ORCHESTRATOR_INPUT_SCHEMA structure
+     * and reads the `workflowStateData` property directly. Subclasses with custom input
+     * schemas MUST override this method to map their custom properties.
+     *
+     * @param input - The parsed orchestrator input
+     * @returns The workflow state data, or undefined if not present or invalid
+     */
+    protected extractWorkflowStateData(input: z.infer<TInputSchema>): WorkflowStateData | undefined;
+    /**
+     * Create orchestration prompt for LLM with embedded tool invocation data and workflow state.
      * Used in delegate mode - instructs LLM to call a separate MCP tool.
+     *
+     * Subclasses with custom input schemas may override this to adjust property
+     * name references in the generated prompt.
      */
-    private createOrchestrationPrompt;
+    protected createOrchestrationPrompt(mcpToolInvocationData: MCPToolInvocationData<z.ZodObject<z.ZodRawShape>>, workflowStateData: WorkflowStateData): string;
     /**
      * Create a direct guidance prompt for the LLM.
      * Used in direct guidance mode - provides guidance inline without an intermediate tool call.
      *
+     * Subclasses with custom input schemas may override this to adjust property
+     * name references in the generated prompt.
+     *
      * @param nodeGuidanceData - The node guidance data containing task guidance and schemas
      * @param workflowStateData - The workflow state data to round-trip back to the orchestrator
      * @returns A prompt with the task guidance and post-task instructions
      */
-    private createDirectGuidancePrompt;
+    protected createDirectGuidancePrompt(nodeGuidanceData: NodeGuidanceData<z.ZodObject<z.ZodRawShape>>, workflowStateData: WorkflowStateData): string;
+    private defaultReturnGuidance;
 }

package/dist/tools/orchestrator/orchestratorTool.js

@@ -47,6 +47,9 @@ export class OrchestratorTool extends AbstractTool {
     /**
      * Handle orchestrator requests - manages workflow state and execution
      */
+    // @ts-expect-error TS2416 - z.infer<TInputSchema> is structurally equivalent to
+    // ShapeOutput<TInputSchema['shape']> (the type ToolCallback resolves to) for Zod object
+    // schemas, but TypeScript cannot prove this for generic type parameters.
     handleRequest = async (input, extra) => {
         // Create progress reporter from MCP context
         this.currentProgressReporter = this.createProgressReporter(extra);
@@ -87,26 +90,29 @@ export class OrchestratorTool extends AbstractTool {
         return new MCPProgressReporter(sendNotification, progressToken);
     }
     async processRequest(input) {
-        // Generate or use existing thread ID for workflow session
-        let threadId = '';
+        // Resolve workflow state: extract from input or create fresh for a new session
+        let workflowStateData;
         try {
             const parsedInput = this.toolMetadata.inputSchema.parse(input);
-            threadId = parsedInput.workflowStateData.thread_id;
+            workflowStateData = this.extractWorkflowStateData(parsedInput);
         }
         catch (error) {
             this.logger.error('Error parsing orchestrator input. Starting a new workflow.', error);
         }
-        if (threadId === '') {
-            threadId = generateUniqueThreadId();
+        const isResumption = workflowStateData && workflowStateData.thread_id !== '';
+        if (!isResumption) {
+            workflowStateData = { thread_id: generateUniqueThreadId() };
         }
-        const workflowStateData = { thread_id: threadId };
+        workflowStateData = workflowStateData;
+        // Extract user input using the (overridable) extractor method
+        const userInput = this.extractUserInput(input);
         this.logger.info('Processing orchestrator request', {
-            threadId,
-            hasUserInput: !!input.userInput,
-            isResumption: !!input.workflowStateData?.thread_id,
+            threadId: workflowStateData.thread_id,
+            hasUserInput: !!userInput,
+            isResumption,
         });
         // Thread configuration for LangGraph - includes optional progress reporter
-        const threadConfig = this.createThreadConfig(threadId, this.getProgressReporter());
+        const threadConfig = this.createThreadConfig(workflowStateData.thread_id, this.getProgressReporter());
         // Get checkpointer from state manager
         const checkpointer = await this.stateManager.createCheckpointer();
         // Compile workflow with checkpointer
@@ -122,13 +128,13 @@ export class OrchestratorTool extends AbstractTool {
                 interrupts: interruptedTask.interrupts.length,
             });
             // Resume workflow with user input from previous tool execution
-            result = await compiledWorkflow.invoke(new Command({ resume: input.userInput }), threadConfig);
+            result = await compiledWorkflow.invoke(new Command({ resume: userInput }), threadConfig);
         }
         else {
             // Start new workflow session
             this.logger.info('Starting new workflow execution');
             result = await compiledWorkflow.invoke({
-                userInput: input.userInput,
+                userInput,
             }, threadConfig);
         }
         this.logger.debug('Processing workflow result');
@@ -194,8 +200,37 @@ export class OrchestratorTool extends AbstractTool {
         return this.currentProgressReporter;
     }
     /**
-     * Create orchestration prompt for LLM with embedded tool invocation data and workflow state
+     * Extract the user input value from the orchestrator input.
+     *
+     * Default implementation assumes the standard ORCHESTRATOR_INPUT_SCHEMA structure
+     * and reads the `userInput` property directly. Subclasses with custom input schemas
+     * MUST override this method to map their custom properties.
+     *
+     * @param input - The parsed orchestrator input
+     * @returns The user input value, or undefined if not present
+     */
+    extractUserInput(input) {
+        return input.userInput;
+    }
+    /**
+     * Extract the workflow state data from the orchestrator input.
+     *
+     * Default implementation assumes the standard ORCHESTRATOR_INPUT_SCHEMA structure
+     * and reads the `workflowStateData` property directly. Subclasses with custom input
+     * schemas MUST override this method to map their custom properties.
+     *
+     * @param input - The parsed orchestrator input
+     * @returns The workflow state data, or undefined if not present or invalid
+     */
+    extractWorkflowStateData(input) {
+        return input.workflowStateData;
+    }
+    /**
+     * Create orchestration prompt for LLM with embedded tool invocation data and workflow state.
      * Used in delegate mode - instructs LLM to call a separate MCP tool.
+     *
+     * Subclasses with custom input schemas may override this to adjust property
+     * name references in the generated prompt.
      */
     createOrchestrationPrompt(mcpToolInvocationData, workflowStateData) {
         return `
@@ -243,22 +278,17 @@ instructions for continuing the workflow.
      * Create a direct guidance prompt for the LLM.
      * Used in direct guidance mode - provides guidance inline without an intermediate tool call.
      *
+     * Subclasses with custom input schemas may override this to adjust property
+     * name references in the generated prompt.
+     *
      * @param nodeGuidanceData - The node guidance data containing task guidance and schemas
      * @param workflowStateData - The workflow state data to round-trip back to the orchestrator
      * @returns A prompt with the task guidance and post-task instructions
      */
     createDirectGuidancePrompt(nodeGuidanceData, workflowStateData) {
-        const resultSchemaJson = JSON.stringify(zodToJsonSchema(nodeGuidanceData.resultSchema), null, 2);
-        // Build example section if provided
-        const exampleSection = nodeGuidanceData.exampleOutput
-            ? `
-For example, a properly formatted result should look like:
-
-\`\`\`json
-${nodeGuidanceData.exampleOutput}
-\`\`\`
-`
-            : '';
+        const returnGuidance = nodeGuidanceData.returnGuidance
+            ? nodeGuidanceData.returnGuidance(workflowStateData)
+            : this.defaultReturnGuidance(workflowStateData, nodeGuidanceData.resultSchema, nodeGuidanceData.exampleOutput);
         return `
 # ROLE
 
@@ -269,7 +299,22 @@ you with direct guidance for the current task.
 
 ${nodeGuidanceData.taskGuidance}
 
-# CRITICAL: REQUIRED NEXT STEP
+${returnGuidance}
+`;
+    }
+    defaultReturnGuidance(workflowStateData, resultSchema, exampleOutput) {
+        const resultSchemaJson = JSON.stringify(zodToJsonSchema(resultSchema), null, 2);
+        // Build example section if provided
+        const exampleSection = exampleOutput
+            ? `
+For example, a properly formatted result should look like:
+
+\`\`\`json
+${exampleOutput}
+\`\`\`
+`
+            : '';
+        return `# CRITICAL: REQUIRED NEXT STEP
 
 After completing the task above, you **MUST** invoke the \`${this.toolMetadata.toolId}\` tool 
 to continue the workflow. The workflow CANNOT proceed without this tool call.
@@ -302,7 +347,7 @@ Here is an example of the EXACT format your tool call should follow:
 \`\`\`
 Tool: ${this.toolMetadata.toolId}
 Parameters:
-  ${WORKFLOW_PROPERTY_NAMES.userInput}: ${nodeGuidanceData.exampleOutput ? nodeGuidanceData.exampleOutput.replaceAll('\n', '\n    ') : '{ /* your result object here */ }'}
+  ${WORKFLOW_PROPERTY_NAMES.userInput}: ${exampleOutput ? exampleOutput.replaceAll('\n', '\n    ') : '{ /* your result object here */ }'}
   ${WORKFLOW_PROPERTY_NAMES.workflowStateData}: ${JSON.stringify(workflowStateData)}
 \`\`\`
 `;

package/dist/tools/orchestrator/orchestratorTool.js.map

@@ -1 +1 @@
-{"version":3,"file":"orchestratorTool.js","sourceRoot":"","sources":["../../../src/tools/orchestrator/orchestratorTool.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAIrD,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAIL,kBAAkB,EAClB,uBAAuB,GAExB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,mBAAmB,EAAyB,MAAM,qCAAqC,CAAC;AACjG,OAAO,EAAE,oBAAoB,EAAE,MAAM,6CAA6C,CAAC;AAEnF,OAAO,EAIL,8BAA8B,GAC/B,MAAM,eAAe,CAAC;AAEvB;;GAEG;AACH,SAAS,sBAAsB;IAC7B,OAAO,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC3E,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,gBAAiB,SAAQ,YAAsC;IAMvD;IALF,YAAY,CAAuB;IAC5C,uBAAuB,CAA+B;IAE9D,YACE,MAAiB,EACA,MAA0B;QAE3C,oEAAoE;QACpE,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,IAAI,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;QAClF,KAAK,CAAC,MAAM,EAAE,8BAA8B,CAAC,MAAM,CAAC,EAAE,kBAAkB,EAAE,eAAe,CAAC,CAAC;QAJ1E,WAAM,GAAN,MAAM,CAAoB;QAM3C,2EAA2E;QAC3E,IAAI,CAAC,YAAY;YACf,MAAM,CAAC,YAAY,IAAI,IAAI,oBAAoB,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC;IACnF,CAAC;IAED;;OAEG;IACI,aAAa,GAAG,KAAK,EAC1B,KAAwB,EACxB,KAA6D,EAC7D,EAAE;QACF,4CAA4C;QAC5C,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAElE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;QAChE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,MAAM,CAAC,CAAC;YAE3D,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;qBAC7B;iBACF;gBACD,iBAAiB,EAAE,MAAM;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAc,CAAC,CAAC;YAC1E,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,kDAAkD;YAClD,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;QAC3C,CAAC;IACH,CAAC,CAAC;IAEF;;;;;;;OAOG;IACO,sBAAsB,CAC9B,KAA6D;QAE7D,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAC1C,MAAM,aAAa,GAAG,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrF,OAAO,IAAI,mBAAmB,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;IAClE,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,KAAwB;QACrD,0DAA0D;QAC1D,IAAI,QAAQ,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC/D,QAAQ,GAAG,WAAW,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,4DAA4D,EAC5D,KAAc,CACf,CAAC;QACJ,CAAC;QACD,IAAI,QAAQ,KAAK,EAAE,EAAE,CAAC;YACpB,QAAQ,GAAG,sBAAsB,EAAE,CAAC;QACtC,CAAC;QACD,MAAM,iBAAiB,GAAsB,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;QAErE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;YAClD,QAAQ;YACR,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,SAAS;YAC/B,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,iBAAiB,EAAE,SAAS;SACnD,CAAC,CAAC;QAEH,2EAA2E;QAC3E,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAEnF,sCAAsC;QACtC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,CAAC;QAElE,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC;QAExE,uCAAuC;QACvC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,IAAI,UAAU,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC/D,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAElF,IAAI,MAAM,CAAC;QACX,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;gBAChD,MAAM,EAAE,eAAe,CAAC,EAAE;gBAC1B,UAAU,EAAE,eAAe,CAAC,UAAU,CAAC,MAAM;aAC9C,CAAC,CAAC;YAEH,+DAA+D;YAC/D,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,CACpC,IAAI,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,EACxC,YAAY,CACb,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,6BAA6B;YAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACpD,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,CACpC;gBACE,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,UAAU,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC3D,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,mCAAmC;YACnC,MAAM,aAAa,GAGjB,eAAe,IAAI,MAAM;gBACvB,CAAC,CACG,MAAM,CAAC,aAGR,CAAC,CAAC,CAAC,CAAC,KAAK;gBACZ,CAAC,CAAC,SAAS,CAAC;YAEhB,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;gBACzE,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;YAC3E,CAAC;YAED,+CAA+C;YAC/C,IAAI,mBAA2B,CAAC;YAChC,IAAI,kBAAkB,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;oBAC7C,MAAM,EAAE,aAAa,CAAC,MAAM;iBAC7B,CAAC,CAAC;gBACH,mBAAmB,GAAG,IAAI,CAAC,0BAA0B,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YAC1F,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;oBACtC,QAAQ,EAAE,aAAa,CAAC,WAAW,EAAE,IAAI;iBAC1C,CAAC,CAAC;gBACH,mBAAmB,GAAG,IAAI,CAAC,yBAAyB,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YACzF,CAAC;YAED,2BAA2B;YAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;YAE5D,OAAO;gBACL,+BAA+B,EAAE,mBAAmB;aACrD,CAAC;QACJ,CAAC;QAED,sBAAsB;QACtB,OAAO;YACL,+BAA+B,EAC7B,0EAA0E;SAC7E,CAAC;IACJ,CAAC;IAED;;;;;;;;;OASG;IACO,kBAAkB,CAC1B,QAAgB,EAChB,gBAAmC;QAEnC,OAAO;YACL,YAAY,EAAE;gBACZ,SAAS,EAAE,QAAQ;gBACnB,gBAAgB;aACjB;SACF,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACO,mBAAmB;QAC3B,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAED;;;OAGG;IACK,yBAAyB,CAC/B,qBAAwE,EACxE,iBAAoC;QAEpC,OAAO;;;;KAIN,IAAI,CAAC,YAAY,CAAC,MAAM;;;;;;;;4BAQD,qBAAqB,CAAC,WAAW,EAAE,IAAI;;;EAGjE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,qBAAqB,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;;;;EAI9E,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,KAAK,CAAC;;;yBAGpB,uBAAuB,CAAC,iBAAiB;;IAE9D,uBAAuB,CAAC,iBAAiB;;;;;EAK3C,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;;;;IAI/B,IAAI,CAAC,YAAY,CAAC,MAAM;;;;;;CAM3B,CAAC;IACA,CAAC;IAED;;;;;;;OAOG;IACK,0BAA0B,CAChC,gBAA8D,EAC9D,iBAAoC;QAEpC,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CACrC,eAAe,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAC9C,IAAI,EACJ,CAAC,CACF,CAAC;QAEF,oCAAoC;QACpC,MAAM,cAAc,GAAG,gBAAgB,CAAC,aAAa;YACnD,CAAC,CAAC;;;;EAIN,gBAAgB,CAAC,aAAa;;CAE/B;YACK,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO;;;;;;;;EAQT,gBAAgB,CAAC,YAAY;;;;6DAI8B,IAAI,CAAC,YAAY,CAAC,MAAM;;;;;;;;0BAQ3D,IAAI,CAAC,YAAY,CAAC,MAAM;;;;MAI5C,uBAAuB,CAAC,SAAS;MACjC,uBAAuB,CAAC,iBAAiB,UAAU,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;;;;QAIlF,uBAAuB,CAAC,SAAS;;;EAGvC,gBAAgB;;EAEhB,cAAc;;;;;;;QAOR,IAAI,CAAC,YAAY,CAAC,MAAM;;IAE5B,uBAAuB,CAAC,SAAS,KAAK,gBAAgB,CAAC,aAAa,CAAC,CAAC,CAAC,gBAAgB,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,mCAAmC;IACtK,uBAAuB,CAAC,iBAAiB,KAAK,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;;CAElF,CAAC;IACA,CAAC;CACF"}
+{"version":3,"file":"orchestratorTool.js","sourceRoot":"","sources":["../../../src/tools/orchestrator/orchestratorTool.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAIrD,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAIL,kBAAkB,EAClB,uBAAuB,GAExB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,mBAAmB,EAAyB,MAAM,qCAAqC,CAAC;AACjG,OAAO,EAAE,oBAAoB,EAAE,MAAM,6CAA6C,CAAC;AAEnF,OAAO,EAKL,8BAA8B,GAC/B,MAAM,eAAe,CAAC;AAEvB;;GAEG;AACH,SAAS,sBAAsB;IAC7B,OAAO,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC3E,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,gBAEX,SAAQ,YAAoD;IAMzC;IALF,YAAY,CAAuB;IAC5C,uBAAuB,CAA+B;IAE9D,YACE,MAAiB,EACA,MAAwC;QAEzD,oEAAoE;QACpE,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,IAAI,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;QAClF,KAAK,CAAC,MAAM,EAAE,8BAA8B,CAAC,MAAM,CAAC,EAAE,kBAAkB,EAAE,eAAe,CAAC,CAAC;QAJ1E,WAAM,GAAN,MAAM,CAAkC;QAMzD,2EAA2E;QAC3E,IAAI,CAAC,YAAY;YACf,MAAM,CAAC,YAAY,IAAI,IAAI,oBAAoB,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC;IACnF,CAAC;IAED;;OAEG;IACH,gFAAgF;IAChF,wFAAwF;IACxF,yEAAyE;IAClE,aAAa,GAAG,KAAK,EAC1B,KAA4B,EAC5B,KAA6D,EAC7D,EAAE;QACF,4CAA4C;QAC5C,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAElE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;QAChE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,MAAM,CAAC,CAAC;YAE3D,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;qBAC7B;iBACF;gBACD,iBAAiB,EAAE,MAAM;aAC1B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAc,CAAC,CAAC;YAC1E,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,kDAAkD;YAClD,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;QAC3C,CAAC;IACH,CAAC,CAAC;IAEF;;;;;;;OAOG;IACO,sBAAsB,CAC9B,KAA6D;QAE7D,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC;QAC1C,MAAM,aAAa,GAAG,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrF,OAAO,IAAI,mBAAmB,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;IAClE,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,KAA4B;QACzD,+EAA+E;QAC/E,IAAI,iBAAgD,CAAC;QACrD,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC/D,iBAAiB,GAAG,IAAI,CAAC,wBAAwB,CAAC,WAAW,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,4DAA4D,EAC5D,KAAc,CACf,CAAC;QACJ,CAAC;QACD,MAAM,YAAY,GAAG,iBAAiB,IAAI,iBAAiB,CAAC,SAAS,KAAK,EAAE,CAAC;QAC7E,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,iBAAiB,GAAG,EAAE,SAAS,EAAE,sBAAsB,EAAE,EAAE,CAAC;QAC9D,CAAC;QACD,iBAAiB,GAAG,iBAAkB,CAAC;QAEvC,8DAA8D;QAC9D,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAE/C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;YAClD,QAAQ,EAAE,iBAAiB,CAAC,SAAS;YACrC,YAAY,EAAE,CAAC,CAAC,SAAS;YACzB,YAAY;SACb,CAAC,CAAC;QAEH,2EAA2E;QAC3E,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAC1C,iBAAiB,CAAC,SAAS,EAC3B,IAAI,CAAC,mBAAmB,EAAE,CAC3B,CAAC;QAEF,sCAAsC;QACtC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,CAAC;QAElE,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC;QAExE,uCAAuC;QACvC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,IAAI,UAAU,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC/D,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAElF,IAAI,MAAM,CAAC;QACX,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;gBAChD,MAAM,EAAE,eAAe,CAAC,EAAE;gBAC1B,UAAU,EAAE,eAAe,CAAC,UAAU,CAAC,MAAM;aAC9C,CAAC,CAAC;YAEH,+DAA+D;YAC/D,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;QAC3F,CAAC;aAAM,CAAC;YACN,6BAA6B;YAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACpD,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,CACpC;gBACE,SAAS;aACV,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,UAAU,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC3D,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,mCAAmC;YACnC,MAAM,aAAa,GAGjB,eAAe,IAAI,MAAM;gBACvB,CAAC,CACG,MAAM,CAAC,aAGR,CAAC,CAAC,CAAC,CAAC,KAAK;gBACZ,CAAC,CAAC,SAAS,CAAC;YAEhB,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;gBACzE,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;YAC3E,CAAC;YAED,+CAA+C;YAC/C,IAAI,mBAA2B,CAAC;YAChC,IAAI,kBAAkB,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;oBAC7C,MAAM,EAAE,aAAa,CAAC,MAAM;iBAC7B,CAAC,CAAC;gBACH,mBAAmB,GAAG,IAAI,CAAC,0BAA0B,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YAC1F,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;oBACtC,QAAQ,EAAE,aAAa,CAAC,WAAW,EAAE,IAAI;iBAC1C,CAAC,CAAC;gBACH,mBAAmB,GAAG,IAAI,CAAC,yBAAyB,CAAC,aAAa,EAAE,iBAAiB,CAAC,CAAC;YACzF,CAAC;YAED,2BAA2B;YAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;YAE5D,OAAO;gBACL,+BAA+B,EAAE,mBAAmB;aACrD,CAAC;QACJ,CAAC;QAED,sBAAsB;QACtB,OAAO;YACL,+BAA+B,EAC7B,0EAA0E;SAC7E,CAAC;IACJ,CAAC;IAED;;;;;;;;;OASG;IACO,kBAAkB,CAC1B,QAAgB,EAChB,gBAAmC;QAEnC,OAAO;YACL,YAAY,EAAE;gBACZ,SAAS,EAAE,QAAQ;gBACnB,gBAAgB;aACjB;SACF,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACO,mBAAmB;QAC3B,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAED;;;;;;;;;OASG;IACO,gBAAgB,CAAC,KAA4B;QACrD,OAAQ,KAA2B,CAAC,SAAS,CAAC;IAChD,CAAC;IAED;;;;;;;;;OASG;IACO,wBAAwB,CAAC,KAA4B;QAC7D,OAAQ,KAA2B,CAAC,iBAAiB,CAAC;IACxD,CAAC;IAED;;;;;;OAMG;IACO,yBAAyB,CACjC,qBAAwE,EACxE,iBAAoC;QAEpC,OAAO;;;;KAIN,IAAI,CAAC,YAAY,CAAC,MAAM;;;;;;;;4BAQD,qBAAqB,CAAC,WAAW,EAAE,IAAI;;;EAGjE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,qBAAqB,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;;;;EAI9E,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,KAAK,CAAC;;;yBAGpB,uBAAuB,CAAC,iBAAiB;;IAE9D,uBAAuB,CAAC,iBAAiB;;;;;EAK3C,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;;;;IAI/B,IAAI,CAAC,YAAY,CAAC,MAAM;;;;;;CAM3B,CAAC;IACA,CAAC;IAED;;;;;;;;;;OAUG;IACO,0BAA0B,CAClC,gBAA8D,EAC9D,iBAAoC;QAEpC,MAAM,cAAc,GAAG,gBAAgB,CAAC,cAAc;YACpD,CAAC,CAAC,gBAAgB,CAAC,cAAc,CAAC,iBAAiB,CAAC;YACpD,CAAC,CAAC,IAAI,CAAC,qBAAqB,CACxB,iBAAiB,EACjB,gBAAgB,CAAC,YAAY,EAC7B,gBAAgB,CAAC,aAAa,CAC/B,CAAC;QAEN,OAAO;;;;;;;;EAQT,gBAAgB,CAAC,YAAY;;EAE7B,cAAc;CACf,CAAC;IACA,CAAC;IAEO,qBAAqB,CAC3B,iBAAoC,EACpC,YAAwC,EACxC,aAAsB;QAEtB,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAEhF,oCAAoC;QACpC,MAAM,cAAc,GAAG,aAAa;YAClC,CAAC,CAAC;;;;EAIN,aAAa;;CAEd;YACK,CAAC,CAAC,EAAE,CAAC;QACP,OAAO;;6DAEkD,IAAI,CAAC,YAAY,CAAC,MAAM;;;;;;;;0BAQ3D,IAAI,CAAC,YAAY,CAAC,MAAM;;;;MAI5C,uBAAuB,CAAC,SAAS;MACjC,uBAAuB,CAAC,iBAAiB,UAAU,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;;;;QAIlF,uBAAuB,CAAC,SAAS;;;EAGvC,gBAAgB;;EAEhB,cAAc;;;;;;;QAOR,IAAI,CAAC,YAAY,CAAC,MAAM;;IAE5B,uBAAuB,CAAC,SAAS,KAAK,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,mCAAmC;IACpI,uBAAuB,CAAC,iBAAiB,KAAK,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;;CAElF,CAAC;IACA,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@salesforce/magen-mcp-workflow",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "type": "module",
   "files": [
     "dist",
@@ -24,14 +24,18 @@
     "lint": "eslint --max-warnings 0 --config ../../eslint.config.mjs \"{src,tests}/**/*.ts\"",
     "format": "prettier --write \"{src,tests}/**/*.ts\""
   },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/forcedotcom/mobile-mcp-tools.git"
+  },
   "dependencies": {
     "@langchain/langgraph": "^0.4.9",
     "@langchain/langgraph-checkpoint": "^0.0.11",
     "@langchain/core": "^0.3.80",
-    "@modelcontextprotocol/sdk": "^1.24.0",
+    "@modelcontextprotocol/sdk": "^1.26.0",
     "pino": "^9.9.5",
     "zod": "^3.25.67",
-    "zod-to-json-schema": "^3.24.6"
+    "zod-to-json-schema": "^3.25.1"
   },
   "devDependencies": {
     "@types/node": "^24.8.1",