@swizzy_ai/kit 1.0.0 → 1.0.2

package/dist/core/wizard/bungee/executor.d.ts

@@ -0,0 +1,15 @@
+import { BungeePlan } from './types';
+export declare class BungeeExecutor {
+    private wizard;
+    private bungeeWorkers;
+    private pendingReentry;
+    constructor(wizard: any);
+    executeBungeePlan(plan: BungeePlan): Promise<void>;
+    private launchBungeeWorker;
+    private createTelescopeContext;
+    private executeWorkerStep;
+    mergeWorkerResults(updates: Record<string, any>, telescope: Record<string, any>): void;
+    private retriggerAnchor;
+    processReentries(): Promise<void>;
+}
+//# sourceMappingURL=executor.d.ts.map

package/dist/core/wizard/bungee/executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/bungee/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAGrC,qBAAa,cAAc;IASb,OAAO,CAAC,MAAM;IAR1B,OAAO,CAAC,aAAa,CAKL;IAChB,OAAO,CAAC,cAAc,CAA0B;gBAE5B,MAAM,EAAE,GAAG;IAEzB,iBAAiB,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;YA6B1C,kBAAkB;IAsChC,OAAO,CAAC,sBAAsB;YAShB,iBAAiB;IAW/B,kBAAkB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI;YAMxE,eAAe;IAOvB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAOxC"}

package/dist/core/wizard/context-manager.d.ts

@@ -0,0 +1,9 @@
+export declare class ContextManager {
+    private workflowContext;
+    setContext(context: any): void;
+    getContext(): any;
+    updateContext(updates: any): void;
+    getWorkflowContext(): any;
+    setWorkflowContext(context: any): void;
+}
+//# sourceMappingURL=context-manager.d.ts.map

package/dist/core/wizard/context-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-manager.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/context-manager.ts"],"names":[],"mappings":"AAAA,qBAAa,cAAc;IACzB,OAAO,CAAC,eAAe,CAAW;IAElC,UAAU,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAI9B,UAAU,IAAI,GAAG;IAIjB,aAAa,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAIjC,kBAAkB,IAAI,GAAG;IAIzB,kBAAkB,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;CAGvC"}

package/dist/core/wizard/logger.d.ts

@@ -0,0 +1,8 @@
+export declare class Logger {
+    private logFilePath;
+    constructor(id: string);
+    log(messageOrFn: string | (() => string)): void;
+    private appendToFile;
+    getLog(): Promise<string>;
+}
+//# sourceMappingURL=logger.d.ts.map

package/dist/core/wizard/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/logger.ts"],"names":[],"mappings":"AAGA,qBAAa,MAAM;IACjB,OAAO,CAAC,WAAW,CAAqB;gBAE5B,EAAE,EAAE,MAAM;IAStB,GAAG,CAAC,WAAW,EAAE,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,GAAG,IAAI;IAO/C,OAAO,CAAC,YAAY;IASd,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC;CAQhC"}

package/dist/core/wizard/schema-utils.d.ts

@@ -0,0 +1,16 @@
+import { z } from 'zod';
+export declare class SchemaUtils {
+    private static schemaDescriptions;
+    private static readonly maxCacheSize;
+    static describeSchema(schema: z.ZodType<any>, stepId?: string): string;
+    static getSchemaType(schema: z.ZodType<any>): string;
+    static extractSchemaFields(schema: z.ZodType<any>): Array<{
+        key: string;
+        type: string;
+        enumValues?: string[];
+    }>;
+    static getXmlExample(key: string, type: string): string;
+    static objectToXml(obj: any, rootName?: string): string;
+    static escapeXml(unsafe: string): string;
+}
+//# sourceMappingURL=schema-utils.d.ts.map

package/dist/core/wizard/schema-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema-utils.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/schema-utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAkC;IACnE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAO;IAE3C,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM;IA8BtE,MAAM,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM;IAUpD,MAAM,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAgB/G,MAAM,CAAC,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM;IAevD,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,QAAQ,GAAE,MAAkB,GAAG,MAAM;IAqBlE,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;CAQzC"}

package/dist/core/wizard/step-data-generator.d.ts

@@ -0,0 +1,24 @@
+import { Step } from './steps/base';
+import { LLMClient } from '../../services/client/index';
+import { Logger } from './logger';
+export declare class StepDataGenerator {
+    private llmClient;
+    private logger;
+    private findStep;
+    private static readonly TEMPLATE_REGEX;
+    private static readonly WIZARD_TAG_PATTERN;
+    constructor(llmClient: LLMClient, logger: Logger, findStep: (id: string) => Step | null);
+    private log;
+    generateStepData(step: Step, stepContext: any, workflowContext: any): Promise<any>;
+    private repairSchemaData;
+    private parseXmlToJson;
+    private parseXmlElementWithTagCategory;
+    private parseNumber;
+    private parseBoolean;
+    private parseArray;
+    private inferAndParseValue;
+    private objectToXml;
+    private escapeXml;
+    private applyTemplate;
+}
+//# sourceMappingURL=step-data-generator.d.ts.map

package/dist/core/wizard/step-data-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-data-generator.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/step-data-generator.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,cAAc,CAAC;AAEpC,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAGlC,qBAAa,iBAAiB;IAIhB,OAAO,CAAC,SAAS;IAAa,OAAO,CAAC,MAAM;IAAU,OAAO,CAAC,QAAQ;IAHlF,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAA8B;IACpE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAyD;gBAE/E,SAAS,EAAE,SAAS,EAAU,MAAM,EAAE,MAAM,EAAU,QAAQ,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI;IAE/G,OAAO,CAAC,GAAG,CAAqC;IAE1C,gBAAgB,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,eAAe,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;YA0G1E,gBAAgB;IAwC9B,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,8BAA8B;IAiFtC,OAAO,CAAC,WAAW;IAQnB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,UAAU;IAYlB,OAAO,CAAC,kBAAkB;IA4B1B,OAAO,CAAC,WAAW;IAqBnB,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,aAAa;CActB"}

package/dist/core/wizard/steps/base.d.ts

@@ -18,14 +18,14 @@ export type FlowControlSignal = 'NEXT' | 'STOP' | 'RETRY' | 'WAIT' | string | {
     type: 'BUNGEE_JUMP';
     plan: BungeePlan;
 };
-export type ContextFunction = (workflowContext: any) => any;
+export type Context = (workflowContext: any) => any;
 export type ContextType = 'xml' | 'template' | 'both';
 export interface StepConfig<T = any> {
     id: string;
     instruction: string;
     schema: z.ZodType<T>;
     update: (result: T, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
-    contextFunction?: ContextFunction;
+    context?: Context;
     contextType?: ContextType;
     beforeRun?: () => Promise<void> | void;
     afterRun?: (result: T) => Promise<void> | void;
@@ -36,11 +36,12 @@ export declare class Step<T = any> {
     readonly instruction: string;
     readonly schema: z.ZodType<T>;
     readonly update: (result: T, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
-    readonly contextFunction?: ContextFunction;
+    readonly context?: Context;
     readonly contextType?: ContextType;
     readonly beforeRun?: () => Promise<void> | void;
     readonly afterRun?: (result: T) => Promise<void> | void;
     readonly model: string;
+    executionCount: number;
     constructor(config: StepConfig<T>);
     validate(data: unknown): T;
     getContext(workflowContext: any): any;

package/dist/core/wizard/steps/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/steps/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAErD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;IACtD,SAAS,EAAE,SAAS,CAAC;IACrB,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,iBAAiB,CAAC;IAC5C,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,KAAK,EAAE,MAAM,iBAAiB,CAAC;IAC/B,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,MAAM,EAAE;QACN,IAAI,EAAE,MAAM,aAAa,CAAC;KAC3B,CAAC;CACH;AAED,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAE7C,MAAM,MAAM,iBAAiB,GACzB,MAAM,GACN,MAAM,GACN,OAAO,GACP,MAAM,GACN,MAAM,GACN;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,IAAI,EAAE,UAAU,CAAA;CAAE,CAAC;AAE9C,MAAM,MAAM,eAAe,GAAG,CAAC,eAAe,EAAE,GAAG,KAAK,GAAG,CAAC;AAE5D,MAAM,MAAM,WAAW,GAAG,KAAK,GAAG,UAAU,GAAG,MAAM,CAAC;AAEtD,MAAM,WAAW,UAAU,CAAC,CAAC,GAAG,GAAG;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAChH,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC/C,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,IAAI,CAAC,CAAC,GAAG,GAAG;IACvB,SAAgB,EAAE,EAAE,MAAM,CAAC;IAC3B,SAAgB,WAAW,EAAE,MAAM,CAAC;IACpC,SAAgB,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACrC,SAAgB,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAChI,SAAgB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClD,SAAgB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1C,SAAgB,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvD,SAAgB,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC/D,SAAgB,KAAK,EAAE,MAAM,CAAC;gBAElB,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;IAYjC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,CAAC;IAQ1B,UAAU,CAAC,eAAe,EAAE,GAAG,GAAG,GAAG;CAGtC"}
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/steps/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAErD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;IACtD,SAAS,EAAE,SAAS,CAAC;IACrB,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,iBAAiB,CAAC;IAC5C,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,KAAK,EAAE,MAAM,iBAAiB,CAAC;IAC/B,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,MAAM,EAAE;QACN,IAAI,EAAE,MAAM,aAAa,CAAC;KAC3B,CAAC;CACH;AAED,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAE7C,MAAM,MAAM,iBAAiB,GACzB,MAAM,GACN,MAAM,GACN,OAAO,GACP,MAAM,GACN,MAAM,GACN;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,IAAI,EAAE,UAAU,CAAA;CAAE,CAAC;AAE9C,MAAM,MAAM,OAAO,GAAG,CAAC,eAAe,EAAE,GAAG,KAAK,GAAG,CAAC;AAEpD,MAAM,MAAM,WAAW,GAAG,KAAK,GAAG,UAAU,GAAG,MAAM,CAAC;AAEtD,MAAM,WAAW,UAAU,CAAC,CAAC,GAAG,GAAG;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACrB,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAChH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC/C,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,IAAI,CAAC,CAAC,GAAG,GAAG;IACvB,SAAgB,EAAE,EAAE,MAAM,CAAC;IAC3B,SAAgB,WAAW,EAAE,MAAM,CAAC;IACpC,SAAgB,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACrC,SAAgB,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAChI,SAAgB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClC,SAAgB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1C,SAAgB,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvD,SAAgB,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC/D,SAAgB,KAAK,EAAE,MAAM,CAAC;IACvB,cAAc,SAAK;gBAEd,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;IAYjC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,CAAC;IAQ1B,UAAU,CAAC,eAAe,EAAE,GAAG,GAAG,GAAG;CAGtC"}

package/dist/core/wizard/steps/compute.d.ts

@@ -1,9 +1,9 @@
-import { Step, ContextFunction, ContextType } from './base';
+import { Step, Context, ContextType } from './base';
 export interface ComputeStepConfig {
     id: string;
     instruction: string;
     update: (result: any, workflowContext: Record<string, any>, actions: import('./base').WizardActions) => Promise<import('./base').FlowControlSignal>;
-    contextFunction?: ContextFunction;
+    context?: Context;
     contextType?: ContextType;
     beforeRun?: () => Promise<void> | void;
     afterRun?: (result: any) => Promise<void> | void;

package/dist/core/wizard/steps/compute.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"compute.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/steps/compute.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAE5D,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,QAAQ,EAAE,aAAa,KAAK,OAAO,CAAC,OAAO,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACpJ,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAClD;AAED,qBAAa,WAAY,SAAQ,IAAI,CAAC,GAAG,CAAC;IACxC,SAAgB,aAAa,EAAE,OAAO,CAAQ;gBAElC,MAAM,EAAE,iBAAiB;IAUrC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,GAAG;CAI7B"}
+{"version":3,"file":"compute.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/steps/compute.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAEpD,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,QAAQ,EAAE,aAAa,KAAK,OAAO,CAAC,OAAO,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACpJ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAClD;AAED,qBAAa,WAAY,SAAQ,IAAI,CAAC,GAAG,CAAC;IACxC,SAAgB,aAAa,EAAE,OAAO,CAAQ;gBAElC,MAAM,EAAE,iBAAiB;IAUrC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,GAAG;CAI7B"}

package/dist/core/wizard/steps/text.d.ts

@@ -1,9 +1,9 @@
-import { Step, ContextFunction, ContextType } from './base';
+import { Step, Context, ContextType } from './base';
 export interface TextStepConfig {
     id: string;
     instruction: string;
     update: (result: string, workflowContext: Record<string, any>, actions: import('./base').WizardActions) => Promise<import('./base').FlowControlSignal>;
-    contextFunction?: ContextFunction;
+    context?: Context;
     contextType?: ContextType;
     beforeRun?: () => Promise<void> | void;
     afterRun?: (result: string) => Promise<void> | void;

package/dist/core/wizard/steps/text.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"text.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/steps/text.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAc,eAAe,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAExE,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,QAAQ,EAAE,aAAa,KAAK,OAAO,CAAC,OAAO,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACvJ,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACpD,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,QAAS,SAAQ,IAAI,CAAC,MAAM,CAAC;gBAC5B,MAAM,EAAE,cAAc;IASlC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM;CAIhC"}
+{"version":3,"file":"text.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/steps/text.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAc,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAC;AAEhE,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,QAAQ,EAAE,aAAa,KAAK,OAAO,CAAC,OAAO,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACvJ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACvC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACpD,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,QAAS,SAAQ,IAAI,CAAC,MAAM,CAAC;gBAC5B,MAAM,EAAE,cAAc;IASlC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM;CAIhC"}

package/dist/core/wizard/usage-tracker.d.ts

@@ -0,0 +1,25 @@
+export interface UsageCallback {
+    (usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }, provider: string): void;
+}
+export declare class UsageTracker {
+    private totalTokens;
+    private stepTokens;
+    private onUsage?;
+    private onUpdate?;
+    private startTime?;
+    constructor(onUsage?: UsageCallback, onUpdate?: (totalTokens: number, rate: number) => void);
+    setStartTime(time: number): void;
+    private notifyUpdate;
+    getTotalTokens(): number;
+    getStepTokens(): number;
+    updateUsage(usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }, provider: string): void;
+}
+//# sourceMappingURL=usage-tracker.d.ts.map

package/dist/core/wizard/usage-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracker.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/usage-tracker.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,aAAa;IAC5B,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CAC1G;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,WAAW,CAAa;IAChC,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,OAAO,CAAC,CAAgB;IAChC,OAAO,CAAC,QAAQ,CAAC,CAA8C;IAC/D,OAAO,CAAC,SAAS,CAAC,CAAS;gBAEf,OAAO,CAAC,EAAE,aAAa,EAAE,QAAQ,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI;IAc3F,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAIhC,OAAO,CAAC,YAAY;IAQpB,cAAc,IAAI,MAAM;IAIxB,aAAa,IAAI,MAAM;IAIvB,WAAW,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;CAUpH"}

package/dist/core/wizard/visualization-manager.d.ts

@@ -0,0 +1,34 @@
+import * as http from 'http';
+export interface WebSocketMessage {
+    type: string;
+    [key: string]: any;
+}
+export declare class VisualizationManager {
+    private wizard;
+    visualizationServer?: http.Server;
+    private wss?;
+    private visualizationPort?;
+    private connectedClients;
+    private readonly maxWebSocketConnections;
+    private wsIntervals;
+    private runResolver?;
+    private pauseResolver?;
+    constructor(wizard: any);
+    private getStepsInfo;
+    visualize(port?: number): Promise<{
+        server: http.Server;
+        url: string;
+    }>;
+    private setupWebSocketHandlers;
+    private handleWebSocketMessage;
+    sendToClients(message: WebSocketMessage): void;
+    waitForRunCommand(): Promise<void>;
+    waitForResume(): Promise<void>;
+    sendStatusUpdate(status: any): void;
+    sendWizardStart(steps: any[]): void;
+    sendStepUpdate(update: any): void;
+    sendContextUpdate(context: any): void;
+    sendTokenUpdate(totalTokens: number, rate: number): void;
+    private getVisualizationHtml;
+}
+//# sourceMappingURL=visualization-manager.d.ts.map

package/dist/core/wizard/visualization-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"visualization-manager.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/visualization-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAM7B,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAED,qBAAa,oBAAoB;IAWnB,OAAO,CAAC,MAAM;IAVnB,mBAAmB,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC;IACzC,OAAO,CAAC,GAAG,CAAC,CAAmB;IAC/B,OAAO,CAAC,iBAAiB,CAAC,CAAS;IACnC,OAAO,CAAC,gBAAgB,CAA6B;IACrD,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAM;IAC9C,OAAO,CAAC,WAAW,CAAqD;IAExE,OAAO,CAAC,WAAW,CAAC,CAAa;IACjC,OAAO,CAAC,aAAa,CAAC,CAAa;gBAEf,MAAM,EAAE,GAAG;IAE/B,OAAO,CAAC,YAAY;IAoBd,SAAS,CAAC,IAAI,GAAE,MAAa,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE,CAAC;IA+CnF,OAAO,CAAC,sBAAsB;IAuD9B,OAAO,CAAC,sBAAsB;IAqGvB,aAAa,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAS/C,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAMlC,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAMpC,gBAAgB,CAAC,MAAM,EAAE,GAAG,GAAG,IAAI;IAInC,eAAe,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,IAAI;IAInC,cAAc,CAAC,MAAM,EAAE,GAAG,GAAG,IAAI;IAIjC,iBAAiB,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAIrC,eAAe,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IAQxD,OAAO,CAAC,oBAAoB;CAQ7B"}

package/dist/core/wizard/wizard.d.ts

@@ -1,4 +1,3 @@
-import * as http from 'http';
 import { Step, StepConfig, FlowControlSignal } from './steps/base';
 import { TextStepConfig } from './steps/text';
 import { ComputeStepConfig } from './steps/compute';
@@ -13,45 +12,48 @@ export interface WizardConfig {
     }, provider: string) => void;
 }
 export interface WizardContext {
-    updateContext: (updates: any) => void;
+    updateContext: (updates: ContextData) => void;
     llmClient: LLMClient;
     goto: (stepId: string) => FlowControlSignal;
     next: () => FlowControlSignal;
     stop: () => FlowControlSignal;
     retry: () => FlowControlSignal;
 }
+type ContextData = Record<string, any>;
 export declare class Wizard {
     private static readonly TEMPLATE_REGEX;
     private static readonly WIZARD_TAG_PATTERN;
+    private static readonly NEXT;
+    private static readonly STOP;
+    private static readonly RETRY;
+    private static readonly WAIT;
     private id;
     private llmClient;
     private systemPrompt?;
     private steps;
-    private workflowContext;
-    private logFilePath;
-    private bungeeWorkers;
-    private pendingReentry;
     private stepIndexMap;
-    private schemaDescriptions;
-    private readonly maxCacheSize;
-    private visualizationServer?;
-    private wss?;
-    private visualizationPort?;
-    private connectedClients;
-    private readonly maxWebSocketConnections;
-    private wsIntervals;
-    private totalTokens;
-    private stepTokens;
     private currentStepIndex;
     private isPaused;
     private isRunning;
     private isStepMode;
+    private skipStartWait;
     private pauseResolver?;
     private userOverrideData?;
-    private runResolver?;
+    private logger;
+    private usageTracker;
+    private contextManager;
+    private visualizationManager;
+    private bungeeExecutor;
+    private events;
+    private isLoggingEnabled;
+    private debounce;
+    private debouncedSendContextUpdate;
+    private get workflowContext();
+    private set workflowContext(value);
+    private get log();
+    private get sendToClients();
+    private get visualizationServer();
     constructor(config: WizardConfig);
-    private log;
-    private appendToFile;
     addStep<T>(config: StepConfig<T>): this;
     addParallelSteps(callback: (addStep: <T>(config: StepConfig<T>) => void) => void): this;
     addTextStep(config: TextStepConfig): this;
@@ -61,50 +63,47 @@ export declare class Wizard {
     stop(): FlowControlSignal;
     retry(): FlowControlSignal;
     wait(): FlowControlSignal;
+    on(event: string, callback: Function): this;
     private clearStepError;
     private isStringSignal;
     private isBungeeJumpSignal;
-    private executeBungeePlan;
-    private launchBungeeWorker;
+    private handleFlowControlSignal;
+    private initializeRun;
+    private executeParallelSteps;
+    private executeSequentialStep;
+    private finalizeRun;
+    private createBaseActions;
     private createWizardActions;
     private createWorkerActions;
-    private createTelescopeContext;
-    private executeWorkerStep;
-    private mergeWorkerResults;
-    private retriggerAnchor;
-    private processReentries;
     findStep(stepId: string): Step | null;
     private findStepIndex;
     private executeStep;
+    private finalizeStepExecution;
     private processStepResult;
-    setContext(context: any): this;
-    getContext(): any;
-    updateContext(updates: any): this;
+    setContext(context: ContextData): this;
+    getContext(): ContextData;
+    updateContext(updates: ContextData): this;
     run(): Promise<void>;
-    private waitForRunCommand;
+    startFrom(stepId: string): void;
+    private executionLoop;
     generateStepData(step: Step, stepContext: any): Promise<any>;
     private repairSchemaData;
-    private describeSchema;
-    private getXmlExample;
-    private getSchemaType;
-    private extractSchemaFields;
+    private createStreamingXmlParser;
     private parseXmlToJson;
     private parseXmlElementWithTagCategory;
+    private parseValueByType;
     private parseNumber;
     private parseBoolean;
     private parseArray;
+    private parseArrayItem;
     private inferAndParseValue;
     private objectToXml;
     private escapeXml;
     private applyTemplate;
     visualize(port?: number): Promise<{
-        server: http.Server;
+        server: any;
         url: string;
     }>;
-    private setupWebSocketHandlers;
-    private handleWebSocketMessage;
-    private sendToClients;
-    private waitForResume;
-    private getVisualizationHtml;
 }
+export {};
 //# sourceMappingURL=wizard.d.ts.map

package/dist/core/wizard/wizard.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"wizard.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/wizard.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAG7B,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,iBAAiB,EAAiB,MAAM,cAAc,CAAC;AAClF,OAAO,EAAY,cAAc,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,EAAe,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACjE,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAKxD,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;CACtH;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,CAAC,OAAO,EAAE,GAAG,KAAK,IAAI,CAAC;IACtC,SAAS,EAAE,SAAS,CAAC;IACrB,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,iBAAiB,CAAC;IAC5C,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,KAAK,EAAE,MAAM,iBAAiB,CAAC;CAChC;AAED,qBAAa,MAAM;IACjB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAA8B;IACpE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAyD;IAEnG,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,KAAK,CAA4B;IACzC,OAAO,CAAC,eAAe,CAAW;IAClC,OAAO,CAAC,WAAW,CAAqB;IAGxC,OAAO,CAAC,aAAa,CAKL;IAChB,OAAO,CAAC,cAAc,CAA0B;IAGhD,OAAO,CAAC,YAAY,CAAkC;IACtD,OAAO,CAAC,kBAAkB,CAAkC;IAC5D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAO;IAGpC,OAAO,CAAC,mBAAmB,CAAC,CAAc;IAC1C,OAAO,CAAC,GAAG,CAAC,CAAmB;IAC/B,OAAO,CAAC,iBAAiB,CAAC,CAAS;IACnC,OAAO,CAAC,gBAAgB,CAA6B;IACrD,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAM;IAC9C,OAAO,CAAC,WAAW,CAAqD;IAKxE,OAAO,CAAC,WAAW,CAAa;IAChC,OAAO,CAAC,UAAU,CAAa;IAE/B,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,QAAQ,CAAkB;IAClC,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,aAAa,CAAC,CAAa;IACnC,OAAO,CAAC,gBAAgB,CAAC,CAAM;IAC/B,OAAO,CAAC,WAAW,CAAC,CAAa;gBAErB,MAAM,EAAE,YAAY;IA6BhC,OAAO,CAAC,GAAG;IAOX,OAAO,CAAC,YAAY;IASpB,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI;IAQvC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,KAAK,IAAI,KAAK,IAAI,GAAG,IAAI;IAevF,WAAW,CAAC,MAAM,EAAE,cAAc,GAAG,IAAI;IAQzC,cAAc,CAAC,MAAM,EAAE,iBAAiB,GAAG,IAAI;IAQ/C,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,iBAAiB;IACvC,IAAI,IAAI,iBAAiB;IACzB,IAAI,IAAI,iBAAiB;IACzB,KAAK,IAAI,iBAAiB;IAC1B,IAAI,IAAI,iBAAiB;IAEzB,OAAO,CAAC,cAAc;IAKtB,OAAO,CAAC,cAAc;IAItB,OAAO,CAAC,kBAAkB;YAIZ,iBAAiB;YA6BjB,kBAAkB;IAsChC,OAAO,CAAC,mBAAmB;IAe3B,OAAO,CAAC,mBAAmB;IAmB3B,OAAO,CAAC,sBAAsB;YAShB,iBAAiB;IAW/B,OAAO,CAAC,kBAAkB;YAMZ,eAAe;YAOf,gBAAgB;IASvB,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI;IAY5C,OAAO,CAAC,aAAa;YAIP,WAAW;YA0FX,iBAAiB;IAe/B,UAAU,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAK9B,UAAU,IAAI,GAAG;IAIjB,aAAa,CAAC,OAAO,EAAE,GAAG,GAAG,IAAI;IAK3B,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;YAuHZ,iBAAiB;IAMlB,gBAAgB,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;YA0G3D,gBAAgB;IAwC9B,OAAO,CAAC,cAAc;IA8BtB,OAAO,CAAC,aAAa;IAerB,OAAO,CAAC,aAAa;IAUrB,OAAO,CAAC,mBAAmB;IAgB3B,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,8BAA8B;IAiFtC,OAAO,CAAC,WAAW;IAQnB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,UAAU;IAYlB,OAAO,CAAC,kBAAkB;IA4B1B,OAAO,CAAC,WAAW;IAqBnB,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,aAAa;IAef,SAAS,CAAC,IAAI,GAAE,MAAa,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE,CAAC;IAiCnF,OAAO,CAAC,sBAAsB;IAuD9B,OAAO,CAAC,sBAAsB;IAmG9B,OAAO,CAAC,aAAa;YAUP,aAAa;IAM3B,OAAO,CAAC,oBAAoB;CAQ7B"}
+{"version":3,"file":"wizard.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/wizard.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,iBAAiB,EAAiB,MAAM,cAAc,CAAC;AAClF,OAAO,EAAY,cAAc,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,EAAe,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACjE,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AA8BxD,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,CAAC;CACtH;AAED,MAAM,WAAW,aAAa;IAC5B,aAAa,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,IAAI,CAAC;IAC9C,SAAS,EAAE,SAAS,CAAC;IACrB,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,iBAAiB,CAAC;IAC5C,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,iBAAiB,CAAC;IAC9B,KAAK,EAAE,MAAM,iBAAiB,CAAC;CAChC;AAED,KAAK,WAAW,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAEvC,qBAAa,MAAM;IACjB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAA8B;IACpE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAyD;IAGnG,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAU;IACtC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAU;IACtC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAW;IACxC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAU;IAEtC,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,KAAK,CAA4B;IACzC,OAAO,CAAC,YAAY,CAAkC;IAEtD,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,QAAQ,CAAkB;IAClC,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,aAAa,CAAC,CAAa;IACnC,OAAO,CAAC,gBAAgB,CAAC,CAAM;IAG/B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,oBAAoB,CAAuB;IACnD,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,MAAM,CAAe;IAE7B,OAAO,CAAC,gBAAgB,CAAmE;IAE3F,OAAO,CAAC,QAAQ;IAQhB,OAAO,CAAC,0BAA0B,CAE1B;IAGR,OAAO,KAAK,eAAe,GAE1B;IAED,OAAO,KAAK,eAAe,QAE1B;IAED,OAAO,KAAK,GAAG,GAEd;IAED,OAAO,KAAK,aAAa,GAExB;IAED,OAAO,KAAK,mBAAmB,GAE9B;gBAEW,MAAM,EAAE,YAAY;IAkBhC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI;IAQvC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,KAAK,IAAI,KAAK,IAAI,GAAG,IAAI;IAevF,WAAW,CAAC,MAAM,EAAE,cAAc,GAAG,IAAI;IAQzC,cAAc,CAAC,MAAM,EAAE,iBAAiB,GAAG,IAAI;IAQ/C,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,iBAAiB;IACvC,IAAI,IAAI,iBAAiB;IACzB,IAAI,IAAI,iBAAiB;IACzB,KAAK,IAAI,iBAAiB;IAC1B,IAAI,IAAI,iBAAiB;IAEzB,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,GAAG,IAAI;IAK3C,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,cAAc;IAItB,OAAO,CAAC,kBAAkB;YAIZ,uBAAuB;YA+BvB,aAAa;YAuDb,oBAAoB;YAwBpB,qBAAqB;IAKnC,OAAO,CAAC,WAAW;IAkBnB,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,mBAAmB;IAS3B,OAAO,CAAC,mBAAmB;IAwBpB,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI;IAW5C,OAAO,CAAC,aAAa;YAIP,WAAW;YAiHX,qBAAqB;YAYrB,iBAAiB;IAM/B,UAAU,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI;IAKtC,UAAU,IAAI,WAAW;IAIzB,aAAa,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI;IAKnC,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;IAQnB,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;YA8BxB,aAAa;IA6Bd,gBAAgB,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;YA6H3D,gBAAgB;IAyC9B,OAAO,CAAC,wBAAwB;IAiEhC,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,8BAA8B;IAiFtC,OAAO,CAAC,gBAAgB;IAYxB,OAAO,CAAC,WAAW;IAQnB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,UAAU;IAiIlB,OAAO,CAAC,cAAc;IA8BtB,OAAO,CAAC,kBAAkB;IA4B1B,OAAO,CAAC,WAAW;IAqBnB,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,aAAa;IAef,SAAS,CAAC,IAAI,GAAE,MAAa,GAAG,OAAO,CAAC;QAAE,MAAM,EAAE,GAAG,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE,CAAC;CAG5E"}

package/dist/index.d.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import * as http from 'http';
 
 interface Provider {
     complete(options: CompletionOptions): Promise<CompletionResult>;
@@ -66,6 +65,7 @@ interface CompletionOptions {
     maxTokens?: number;
     temperature?: number;
     stream?: boolean;
+    onChunk?: (chunk: string) => void;
     onUsage?: (usage: {
         promptTokens: number;
         completionTokens: number;
@@ -142,14 +142,14 @@ type FlowControlSignal = 'NEXT' | 'STOP' | 'RETRY' | 'WAIT' | string | {
     type: 'BUNGEE_JUMP';
     plan: BungeePlan;
 };
-type ContextFunction = (workflowContext: any) => any;
+type Context = (workflowContext: any) => any;
 type ContextType = 'xml' | 'template' | 'both';
 interface StepConfig<T = any> {
     id: string;
     instruction: string;
     schema: z.ZodType<T>;
     update: (result: T, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
-    contextFunction?: ContextFunction;
+    context?: Context;
     contextType?: ContextType;
     beforeRun?: () => Promise<void> | void;
     afterRun?: (result: T) => Promise<void> | void;
@@ -160,11 +160,12 @@ declare class Step<T = any> {
     readonly instruction: string;
     readonly schema: z.ZodType<T>;
     readonly update: (result: T, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
-    readonly contextFunction?: ContextFunction;
+    readonly context?: Context;
     readonly contextType?: ContextType;
     readonly beforeRun?: () => Promise<void> | void;
     readonly afterRun?: (result: T) => Promise<void> | void;
     readonly model: string;
+    executionCount: number;
     constructor(config: StepConfig<T>);
     validate(data: unknown): T;
     getContext(workflowContext: any): any;
@@ -174,7 +175,7 @@ interface TextStepConfig {
     id: string;
     instruction: string;
     update: (result: string, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
-    contextFunction?: ContextFunction;
+    context?: Context;
     contextType?: ContextType;
     beforeRun?: () => Promise<void> | void;
     afterRun?: (result: string) => Promise<void> | void;
@@ -189,7 +190,7 @@ interface ComputeStepConfig {
     id: string;
     instruction: string;
     update: (result: any, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
-    contextFunction?: ContextFunction;
+    context?: Context;
     contextType?: ContextType;
     beforeRun?: () => Promise<void> | void;
     afterRun?: (result: any) => Promise<void> | void;
@@ -210,45 +211,48 @@ interface WizardConfig {
     }, provider: string) => void;
 }
 interface WizardContext {
-    updateContext: (updates: any) => void;
+    updateContext: (updates: ContextData) => void;
     llmClient: LLMClient;
     goto: (stepId: string) => FlowControlSignal;
     next: () => FlowControlSignal;
     stop: () => FlowControlSignal;
     retry: () => FlowControlSignal;
 }
+type ContextData = Record<string, any>;
 declare class Wizard {
     private static readonly TEMPLATE_REGEX;
     private static readonly WIZARD_TAG_PATTERN;
+    private static readonly NEXT;
+    private static readonly STOP;
+    private static readonly RETRY;
+    private static readonly WAIT;
     private id;
     private llmClient;
     private systemPrompt?;
     private steps;
-    private workflowContext;
-    private logFilePath;
-    private bungeeWorkers;
-    private pendingReentry;
     private stepIndexMap;
-    private schemaDescriptions;
-    private readonly maxCacheSize;
-    private visualizationServer?;
-    private wss?;
-    private visualizationPort?;
-    private connectedClients;
-    private readonly maxWebSocketConnections;
-    private wsIntervals;
-    private totalTokens;
-    private stepTokens;
     private currentStepIndex;
     private isPaused;
     private isRunning;
     private isStepMode;
+    private skipStartWait;
     private pauseResolver?;
     private userOverrideData?;
-    private runResolver?;
+    private logger;
+    private usageTracker;
+    private contextManager;
+    private visualizationManager;
+    private bungeeExecutor;
+    private events;
+    private isLoggingEnabled;
+    private debounce;
+    private debouncedSendContextUpdate;
+    private get workflowContext();
+    private set workflowContext(value);
+    private get log();
+    private get sendToClients();
+    private get visualizationServer();
     constructor(config: WizardConfig);
-    private log;
-    private appendToFile;
     addStep<T>(config: StepConfig<T>): this;
     addParallelSteps(callback: (addStep: <T>(config: StepConfig<T>) => void) => void): this;
     addTextStep(config: TextStepConfig): this;
@@ -258,52 +262,48 @@ declare class Wizard {
     stop(): FlowControlSignal;
     retry(): FlowControlSignal;
     wait(): FlowControlSignal;
+    on(event: string, callback: Function): this;
     private clearStepError;
     private isStringSignal;
     private isBungeeJumpSignal;
-    private executeBungeePlan;
-    private launchBungeeWorker;
+    private handleFlowControlSignal;
+    private initializeRun;
+    private executeParallelSteps;
+    private executeSequentialStep;
+    private finalizeRun;
+    private createBaseActions;
     private createWizardActions;
     private createWorkerActions;
-    private createTelescopeContext;
-    private executeWorkerStep;
-    private mergeWorkerResults;
-    private retriggerAnchor;
-    private processReentries;
     findStep(stepId: string): Step | null;
     private findStepIndex;
     private executeStep;
+    private finalizeStepExecution;
     private processStepResult;
-    setContext(context: any): this;
-    getContext(): any;
-    updateContext(updates: any): this;
+    setContext(context: ContextData): this;
+    getContext(): ContextData;
+    updateContext(updates: ContextData): this;
     run(): Promise<void>;
-    private waitForRunCommand;
+    startFrom(stepId: string): void;
+    private executionLoop;
     generateStepData(step: Step, stepContext: any): Promise<any>;
     private repairSchemaData;
-    private describeSchema;
-    private getXmlExample;
-    private getSchemaType;
-    private extractSchemaFields;
+    private createStreamingXmlParser;
     private parseXmlToJson;
     private parseXmlElementWithTagCategory;
+    private parseValueByType;
     private parseNumber;
     private parseBoolean;
     private parseArray;
+    private parseArrayItem;
     private inferAndParseValue;
     private objectToXml;
     private escapeXml;
     private applyTemplate;
     visualize(port?: number): Promise<{
-        server: http.Server;
+        server: any;
         url: string;
     }>;
-    private setupWebSocketHandlers;
-    private handleWebSocketMessage;
-    private sendToClients;
-    private waitForResume;
-    private getVisualizationHtml;
 }
 
 export { BaseProvider, BungeeBuilder, ComputeStep, LLMClient, Model, Models, MultiProvider, ProviderRegistry, Step, SwizzyProvider, TextStep, Wizard, Wizard as default };
-export type { BungeeDestination, BungeePlan, CompletionOptions, CompletionResult, ComputeStepConfig, ContextFunction, ContextType, FlowControlSignal, LLMConfig, Provider, StepConfig, TextStepConfig, WizardActions, WizardConfig, WizardContext };
+export type { BungeeDestination, BungeePlan, CompletionOptions, CompletionResult, ComputeStepConfig, Context, ContextType, FlowControlSignal, LLMConfig, Provider, StepConfig, TextStepConfig, WizardActions, WizardConfig, WizardContext };