Customization of guidance

Author: khawkins

packages/mcp-workflow/src/common/metadata.ts

@@ -77,6 +77,15 @@ export interface NodeGuidanceData<TResultSchema extends z.ZodObject<z.ZodRawShap
    * LLM compliance with the expected structure.
    */
   exampleOutput?: string;
+  /**
+   * Optional guidance for the LLM to return to the orchestrator.
+   * When provided, this guidance will replace the default "return to orchestrator"
+   * guidance that the orchestrator normally provides to the LLM, to return the
+   * workflow to the orchestrator after task completion. Make sure this custom
+   * guidance can properly return the workflow to the orchestrator, or the workflow
+   * will likely be broken.
+   */
+  returnGuidance?: (workflowStateData: WorkflowStateData) => string;
 }
 
 /**

packages/mcp-workflow/src/index.ts

@@ -68,6 +68,7 @@ export { type BaseGraphConfig, type WorkflowRunnableConfig } from './common/grap
 export {
   BaseNode,
   AbstractToolNode,
+  UserInputExtractionNode,
   createGetUserInputNode,
   createUserInputExtractionNode,
   type GetUserInputNodeOptions,
@@ -78,7 +79,7 @@ export {
 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 {

packages/mcp-workflow/src/services/inputExtractionService.ts

@@ -204,7 +204,7 @@ export class InputExtractionService
    * @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 {