action-queue

action-queue.ts

interface IActionQueue<T> {
  jobId: string; // primary key
  currentStatus: 'pending' | 'running' | 'completed' | 'cancelled'; // GSI PK
  priorityScore: number; // timestamp, GSI SK (default value = createdAt) can be overriden.
  createdAt: number; // the real createdAt (although not really)
  resource: T;
}

// IPromptParam = { seed, sampler, prompt, negative, n_iter, ... }
interface IGenerateQueue extends IActionQueue<IPromptParam[]> {
  progressDone: number;
  totalWork: number;
}
 
// IFetchParam = { source, protocol: 's3' | 'http', type: 'checkpoint' | 'lora' }
interface IFetchQueue extends IActionQueue<IFetchParam> { }