import type { SourceBundle } from "../generate/bundle";
import type { PlainValue } from "./plain";
import type { BackendInfo, GpuStages, RunResult } from "./types";

export type ToWorker =
  | {
      type: "run";
      id: number;
      bundle: SourceBundle;
      inputs?: Record<string, PlainValue>;
      gpu?: GpuStages;
    }
  | { type: "info"; id: number };

export type FromWorker =
  | { type: "output"; id: number; text: string }
  | { type: "done"; id: number; result: RunResult }
  | { type: "info"; id: number; backend: BackendInfo };