@swizzy_ai/kit 1.0.0

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

@@ -0,0 +1,27 @@
+import { BungeePlan } from './types';
+export declare class BungeeBuilder {
+    private _plan;
+    constructor(currentStepId: string);
+    /**
+     * Add a single step execution.
+     */
+    readonly add: (stepId: string) => this;
+    /**
+     * Add multiple executions based on count with config function.
+     */
+    readonly batch: (stepId: string, count: number, configFn: (index: number) => Record<string, any>) => this;
+    /**
+     * Configure execution settings.
+     */
+    readonly config: (options: {
+        concurrency?: number;
+    }) => this;
+    /**
+     * Trigger the Jump.
+     */
+    readonly jump: () => {
+        type: "BUNGEE_JUMP";
+        plan: BungeePlan;
+    };
+}
+//# sourceMappingURL=builder.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"builder.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/bungee/builder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAqB,MAAM,SAAS,CAAC;AAExD,qBAAa,aAAa;IACxB,OAAO,CAAC,KAAK,CAAa;gBAEd,aAAa,EAAE,MAAM;IASjC;;OAEG;IACH,SAAgB,GAAG,GAAI,QAAQ,MAAM,UAGnC;IAEF;;OAEG;IACH,SAAgB,KAAK,GACnB,QAAQ,MAAM,EACd,OAAO,MAAM,EACb,UAAU,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,UAOhD;IAEF;;OAEG;IACH,SAAgB,MAAM,GAAI,SAAS;QAAE,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,UAKzD;IAEF;;OAEG;IACH,SAAgB,IAAI,QAAO;QAAE,IAAI,EAAE,aAAa,CAAC;QAAC,IAAI,EAAE,UAAU,CAAA;KAAE,CAElE;CACH"}

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

@@ -0,0 +1,12 @@
+export interface BungeeDestination {
+    type: 'step';
+    targetId: string;
+}
+export interface BungeePlan {
+    id: string;
+    anchorId: string;
+    destinations: BungeeDestination[];
+    concurrency: number;
+    configFn?: (index: number) => Record<string, any>;
+}
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/core/wizard/bungee/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,iBAAiB,EAAE,CAAC;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACnD"}

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

@@ -0,0 +1,7 @@
+export * from './steps/base';
+export * from './steps/text';
+export * from './steps/compute';
+export * from './wizard';
+export * from './bungee/types';
+export * from './bungee/builder';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/wizard/index.ts"],"names":[],"mappings":"AAAA,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,iBAAiB,CAAC;AAChC,cAAc,UAAU,CAAC;AACzB,cAAc,gBAAgB,CAAC;AAC/B,cAAc,kBAAkB,CAAC"}

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

@@ -0,0 +1,48 @@
+import { z } from 'zod';
+import { LLMClient } from '../../../services/client';
+import { BungeeBuilder } from '../bungee/builder';
+export interface WizardActions {
+    updateContext: (updates: Record<string, any>) => void;
+    llmClient: LLMClient;
+    goto: (stepId: string) => FlowControlSignal;
+    next: () => FlowControlSignal;
+    stop: () => FlowControlSignal;
+    retry: () => FlowControlSignal;
+    wait: () => FlowControlSignal;
+    bungee: {
+        init: () => BungeeBuilder;
+    };
+}
+import { BungeePlan } from '../bungee/types';
+export type FlowControlSignal = 'NEXT' | 'STOP' | 'RETRY' | 'WAIT' | string | {
+    type: 'BUNGEE_JUMP';
+    plan: BungeePlan;
+};
+export type ContextFunction = (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;
+    contextType?: ContextType;
+    beforeRun?: () => Promise<void> | void;
+    afterRun?: (result: T) => Promise<void> | void;
+    model: string;
+}
+export declare class Step<T = any> {
+    readonly id: string;
+    readonly instruction: string;
+    readonly schema: z.ZodType<T>;
+    readonly update: (result: T, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
+    readonly contextFunction?: ContextFunction;
+    readonly contextType?: ContextType;
+    readonly beforeRun?: () => Promise<void> | void;
+    readonly afterRun?: (result: T) => Promise<void> | void;
+    readonly model: string;
+    constructor(config: StepConfig<T>);
+    validate(data: unknown): T;
+    getContext(workflowContext: any): any;
+}
+//# sourceMappingURL=base.d.ts.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,16 @@
+import { Step, ContextFunction, 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;
+    contextType?: ContextType;
+    beforeRun?: () => Promise<void> | void;
+    afterRun?: (result: any) => Promise<void> | void;
+}
+export declare class ComputeStep extends Step<any> {
+    readonly isComputeStep: boolean;
+    constructor(config: ComputeStepConfig);
+    validate(data: unknown): any;
+}
+//# sourceMappingURL=compute.d.ts.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,16 @@
+import { Step, ContextFunction, 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;
+    contextType?: ContextType;
+    beforeRun?: () => Promise<void> | void;
+    afterRun?: (result: string) => Promise<void> | void;
+    model: string;
+}
+export declare class TextStep extends Step<string> {
+    constructor(config: TextStepConfig);
+    validate(data: unknown): string;
+}
+//# sourceMappingURL=text.d.ts.map

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

@@ -0,0 +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"}

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

@@ -0,0 +1,110 @@
+import * as http from 'http';
+import { Step, StepConfig, FlowControlSignal } from './steps/base';
+import { TextStepConfig } from './steps/text';
+import { ComputeStepConfig } from './steps/compute';
+import { LLMClient } from '../../services/client/index';
+export interface WizardConfig {
+    id: string;
+    systemPrompt?: string;
+    onUsage?: (usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }, provider: string) => void;
+}
+export interface WizardContext {
+    updateContext: (updates: any) => void;
+    llmClient: LLMClient;
+    goto: (stepId: string) => FlowControlSignal;
+    next: () => FlowControlSignal;
+    stop: () => FlowControlSignal;
+    retry: () => FlowControlSignal;
+}
+export declare class Wizard {
+    private static readonly TEMPLATE_REGEX;
+    private static readonly WIZARD_TAG_PATTERN;
+    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 pauseResolver?;
+    private userOverrideData?;
+    private runResolver?;
+    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;
+    addComputeStep(config: ComputeStepConfig): this;
+    goto(stepId: string): FlowControlSignal;
+    next(): FlowControlSignal;
+    stop(): FlowControlSignal;
+    retry(): FlowControlSignal;
+    wait(): FlowControlSignal;
+    private clearStepError;
+    private isStringSignal;
+    private isBungeeJumpSignal;
+    private executeBungeePlan;
+    private launchBungeeWorker;
+    private createWizardActions;
+    private createWorkerActions;
+    private createTelescopeContext;
+    private executeWorkerStep;
+    private mergeWorkerResults;
+    private retriggerAnchor;
+    private processReentries;
+    findStep(stepId: string): Step | null;
+    private findStepIndex;
+    private executeStep;
+    private processStepResult;
+    setContext(context: any): this;
+    getContext(): any;
+    updateContext(updates: any): this;
+    run(): Promise<void>;
+    private waitForRunCommand;
+    generateStepData(step: Step, stepContext: any): Promise<any>;
+    private repairSchemaData;
+    private describeSchema;
+    private getXmlExample;
+    private getSchemaType;
+    private extractSchemaFields;
+    private parseXmlToJson;
+    private parseXmlElementWithTagCategory;
+    private parseNumber;
+    private parseBoolean;
+    private parseArray;
+    private inferAndParseValue;
+    private objectToXml;
+    private escapeXml;
+    private applyTemplate;
+    visualize(port?: number): Promise<{
+        server: http.Server;
+        url: string;
+    }>;
+    private setupWebSocketHandlers;
+    private handleWebSocketMessage;
+    private sendToClients;
+    private waitForResume;
+    private getVisualizationHtml;
+}
+//# sourceMappingURL=wizard.d.ts.map

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

@@ -0,0 +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"}

package/dist/index.d.ts

@@ -0,0 +1,309 @@
+import { z } from 'zod';
+import * as http from 'http';
+
+interface Provider {
+    complete(options: CompletionOptions): Promise<CompletionResult>;
+    supportsModel(model: string): boolean;
+    getProviderName(): string;
+}
+declare abstract class BaseProvider implements Provider {
+    abstract complete(options: CompletionOptions): Promise<CompletionResult>;
+    abstract supportsModel(model: string): boolean;
+    abstract getProviderName(): string;
+}
+declare class SwizzyProvider extends BaseProvider {
+    private config;
+    constructor(config: LLMConfig);
+    complete(options: CompletionOptions): Promise<CompletionResult>;
+    supportsModel(model: string): boolean;
+    getProviderName(): string;
+}
+declare class MultiProvider extends BaseProvider {
+    private openaiKey?;
+    private anthropicKey?;
+    private googleKey?;
+    private xaiKey?;
+    constructor();
+    complete(options: CompletionOptions): Promise<CompletionResult>;
+    supportsModel(model: string): boolean;
+    getProviderName(): string;
+}
+
+declare class ProviderRegistry {
+    private providers;
+    constructor();
+    private registerDefaultProviders;
+    registerProvider(name: string, provider: Provider): void;
+    getProviderForModel(model: string): Provider;
+    getAllProviders(): Provider[];
+    hasProviderForModel(model: string): boolean;
+}
+
+declare enum Models {
+    GPT4 = "gpt-4",
+    GPT35_TURBO = "gpt-3.5-turbo",
+    CLAUDE3_SONNET = "claude-3-sonnet-20240229",
+    CLAUDE3_HAIKU = "claude-3-haiku-20240307",
+    GEMINI_PRO = "gemini-pro",
+    GEMINI_PRO_VISION = "gemini-pro-vision",
+    GROK_BETA = "grok-beta",
+    SWIZZY_DEFAULT = "swizzy-default"
+}
+declare class Model {
+    readonly provider: string;
+    readonly model: string;
+    constructor(modelEnum: Models);
+    private getProviderFromModel;
+}
+
+interface LLMConfig {
+    baseURL?: string;
+    apiKey?: string;
+}
+interface CompletionOptions {
+    prompt: string;
+    model?: string;
+    maxTokens?: number;
+    temperature?: number;
+    stream?: boolean;
+    onUsage?: (usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }, provider: string) => void;
+}
+interface CompletionResult {
+    text: string;
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}
+
+declare class LLMClient {
+    private registry;
+    constructor(registry: ProviderRegistry);
+    complete(options: CompletionOptions): Promise<CompletionResult>;
+}
+
+interface BungeeDestination {
+    type: 'step';
+    targetId: string;
+}
+interface BungeePlan {
+    id: string;
+    anchorId: string;
+    destinations: BungeeDestination[];
+    concurrency: number;
+    configFn?: (index: number) => Record<string, any>;
+}
+
+declare class BungeeBuilder {
+    private _plan;
+    constructor(currentStepId: string);
+    /**
+     * Add a single step execution.
+     */
+    readonly add: (stepId: string) => this;
+    /**
+     * Add multiple executions based on count with config function.
+     */
+    readonly batch: (stepId: string, count: number, configFn: (index: number) => Record<string, any>) => this;
+    /**
+     * Configure execution settings.
+     */
+    readonly config: (options: {
+        concurrency?: number;
+    }) => this;
+    /**
+     * Trigger the Jump.
+     */
+    readonly jump: () => {
+        type: "BUNGEE_JUMP";
+        plan: BungeePlan;
+    };
+}
+
+interface WizardActions {
+    updateContext: (updates: Record<string, any>) => void;
+    llmClient: LLMClient;
+    goto: (stepId: string) => FlowControlSignal;
+    next: () => FlowControlSignal;
+    stop: () => FlowControlSignal;
+    retry: () => FlowControlSignal;
+    wait: () => FlowControlSignal;
+    bungee: {
+        init: () => BungeeBuilder;
+    };
+}
+
+type FlowControlSignal = 'NEXT' | 'STOP' | 'RETRY' | 'WAIT' | string | {
+    type: 'BUNGEE_JUMP';
+    plan: BungeePlan;
+};
+type ContextFunction = (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;
+    contextType?: ContextType;
+    beforeRun?: () => Promise<void> | void;
+    afterRun?: (result: T) => Promise<void> | void;
+    model: string;
+}
+declare class Step<T = any> {
+    readonly id: string;
+    readonly instruction: string;
+    readonly schema: z.ZodType<T>;
+    readonly update: (result: T, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
+    readonly contextFunction?: ContextFunction;
+    readonly contextType?: ContextType;
+    readonly beforeRun?: () => Promise<void> | void;
+    readonly afterRun?: (result: T) => Promise<void> | void;
+    readonly model: string;
+    constructor(config: StepConfig<T>);
+    validate(data: unknown): T;
+    getContext(workflowContext: any): any;
+}
+
+interface TextStepConfig {
+    id: string;
+    instruction: string;
+    update: (result: string, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
+    contextFunction?: ContextFunction;
+    contextType?: ContextType;
+    beforeRun?: () => Promise<void> | void;
+    afterRun?: (result: string) => Promise<void> | void;
+    model: string;
+}
+declare class TextStep extends Step<string> {
+    constructor(config: TextStepConfig);
+    validate(data: unknown): string;
+}
+
+interface ComputeStepConfig {
+    id: string;
+    instruction: string;
+    update: (result: any, workflowContext: Record<string, any>, actions: WizardActions) => Promise<FlowControlSignal>;
+    contextFunction?: ContextFunction;
+    contextType?: ContextType;
+    beforeRun?: () => Promise<void> | void;
+    afterRun?: (result: any) => Promise<void> | void;
+}
+declare class ComputeStep extends Step<any> {
+    readonly isComputeStep: boolean;
+    constructor(config: ComputeStepConfig);
+    validate(data: unknown): any;
+}
+
+interface WizardConfig {
+    id: string;
+    systemPrompt?: string;
+    onUsage?: (usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    }, provider: string) => void;
+}
+interface WizardContext {
+    updateContext: (updates: any) => void;
+    llmClient: LLMClient;
+    goto: (stepId: string) => FlowControlSignal;
+    next: () => FlowControlSignal;
+    stop: () => FlowControlSignal;
+    retry: () => FlowControlSignal;
+}
+declare class Wizard {
+    private static readonly TEMPLATE_REGEX;
+    private static readonly WIZARD_TAG_PATTERN;
+    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 pauseResolver?;
+    private userOverrideData?;
+    private runResolver?;
+    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;
+    addComputeStep(config: ComputeStepConfig): this;
+    goto(stepId: string): FlowControlSignal;
+    next(): FlowControlSignal;
+    stop(): FlowControlSignal;
+    retry(): FlowControlSignal;
+    wait(): FlowControlSignal;
+    private clearStepError;
+    private isStringSignal;
+    private isBungeeJumpSignal;
+    private executeBungeePlan;
+    private launchBungeeWorker;
+    private createWizardActions;
+    private createWorkerActions;
+    private createTelescopeContext;
+    private executeWorkerStep;
+    private mergeWorkerResults;
+    private retriggerAnchor;
+    private processReentries;
+    findStep(stepId: string): Step | null;
+    private findStepIndex;
+    private executeStep;
+    private processStepResult;
+    setContext(context: any): this;
+    getContext(): any;
+    updateContext(updates: any): this;
+    run(): Promise<void>;
+    private waitForRunCommand;
+    generateStepData(step: Step, stepContext: any): Promise<any>;
+    private repairSchemaData;
+    private describeSchema;
+    private getXmlExample;
+    private getSchemaType;
+    private extractSchemaFields;
+    private parseXmlToJson;
+    private parseXmlElementWithTagCategory;
+    private parseNumber;
+    private parseBoolean;
+    private parseArray;
+    private inferAndParseValue;
+    private objectToXml;
+    private escapeXml;
+    private applyTemplate;
+    visualize(port?: number): Promise<{
+        server: http.Server;
+        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 };

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,IAAI,OAAO,EAAE,MAAM,eAAe,CAAC;AAGlD,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AACpE,cAAc,eAAe,CAAC;AAC9B,cAAc,mBAAmB,CAAC"}