deepline 0.1.0 → 0.1.2

package/dist/repo/shared_libs/play-runtime/runtime-actions.ts

@@ -0,0 +1,208 @@
+import type { ComputeBillingItem } from './worker-api-types';
+import type { DbSessionOperation } from './db-session';
+import type { PlayRowUpdate } from './ctx-types';
+import type { PlaySheetContract } from '../plays/static-pipeline';
+
+export type RuntimeRunStatusAction = {
+  action: 'update_run_status';
+  playId: string;
+  status: string;
+  error?: string;
+  runtimeBackend?: string | null;
+  artifactHash?: string | null;
+  graphHash?: string | null;
+  waitKind?: string | null;
+  waitUntil?: number | null;
+  activeBoundaryId?: string | null;
+  lastCheckpointAt?: number | null;
+  liveLogs?: string[];
+  liveTimeline?: unknown;
+  liveNodeProgress?: unknown;
+  result?: Record<string, unknown> | null;
+};
+
+export type RuntimeSheetAction =
+  | {
+      action: 'ensure_sheet';
+      playName: string;
+      tableNamespace: string;
+      sheetContract?: PlaySheetContract | null;
+      userEmail?: string | null;
+    }
+  | {
+      action: 'create_db_session';
+      playName: string;
+      target: {
+        tableNamespace: string;
+        logicalTable: 'sheet_rows' | 'dataset_rows' | 'play_output';
+      };
+      operations: DbSessionOperation[];
+      limits?: {
+        maxRows?: number;
+        maxBytes?: number;
+        maxRequests?: number;
+      };
+      sheetContract?: PlaySheetContract | null;
+      ttlSeconds?: number;
+      userEmail?: string | null;
+    }
+  | {
+      action: 'apply_row_updates';
+      playName: string;
+      tableNamespace: string;
+      sheetContract?: PlaySheetContract | null;
+      contractSnapshot?: unknown;
+      runId: string;
+      userEmail?: string | null;
+      updates: Array<Omit<PlayRowUpdate, 'rowId'> & { runId?: string }>;
+    }
+  | {
+      action: 'start_sheet_dataset';
+      playName: string;
+      tableNamespace: string;
+      sheetContract?: PlaySheetContract | null;
+      contractSnapshot?: unknown;
+      rows: unknown;
+      runId: string;
+      userEmail?: string | null;
+    }
+  | {
+      action: 'persist_completed_sheet_rows';
+      playName: string;
+      tableNamespace: string;
+      sheetContract?: PlaySheetContract | null;
+      contractSnapshot?: unknown;
+      rows: unknown;
+      outputFields: string[];
+      runId: string;
+      userEmail?: string | null;
+    };
+
+export type RuntimeArtifactAction =
+  | {
+      action: 'create_signed_artifact_url';
+      storageKey: string;
+    }
+  | {
+      action: 'create_signed_staged_file_url';
+      file: { storageKey: string };
+    };
+
+export type RuntimeBillingAction =
+  | {
+      action: 'compute_billing_upsert';
+      sessionId: string;
+      orgId: string;
+      userId?: string | null;
+      operation: string;
+      workflowId?: string;
+      runId?: string;
+    }
+  | {
+      action: 'compute_billing_record_item';
+      sessionId: string;
+      orgId: string;
+      userId?: string | null;
+      operation: string;
+      item: ComputeBillingItem;
+    }
+  | {
+      action: 'compute_billing_finalize';
+      sessionId: string;
+      orgId: string;
+      userId?: string | null;
+      operation: string;
+      status: 'completed' | 'error';
+      workflowId?: string;
+      runId?: string;
+      maxCreditsPerRun?: number | null;
+      finalItem: ComputeBillingItem;
+    };
+
+export type RuntimeReceiptAction =
+  | {
+      action: 'get_runtime_step_receipt';
+      playName: string;
+      runId: string;
+      key: string;
+    }
+  | {
+      action: 'claim_runtime_step_receipt';
+      playName: string;
+      runId: string;
+      key: string;
+    }
+  | {
+      action: 'complete_runtime_step_receipt';
+      playName: string;
+      runId: string;
+      key: string;
+      output: unknown;
+    }
+  | {
+      action: 'fail_runtime_step_receipt';
+      playName: string;
+      runId: string;
+      key: string;
+      error: string;
+    }
+  | {
+      action: 'skip_runtime_step_receipt';
+      playName: string;
+      runId: string;
+      key: string;
+      output?: unknown;
+    };
+
+export type RuntimePlayResolutionAction = {
+  action: 'resolve_play';
+  playRef: string;
+};
+
+export type RuntimeResultsAction = {
+  action: 'save_results';
+  playId: string;
+  userId?: string | null;
+  result: {
+    success: boolean;
+    error?: string;
+    publicResult?: Record<string, unknown> | null;
+    maxCreditsPerRun?: number | null;
+    temporalActionEstimate?: Record<string, unknown> | null;
+  };
+};
+
+export type RuntimeApiAction =
+  | RuntimePlayResolutionAction
+  | RuntimeSheetAction
+  | RuntimeArtifactAction
+  | RuntimeRunStatusAction
+  | RuntimeResultsAction
+  | RuntimeBillingAction
+  | RuntimeReceiptAction;
+
+export type RuntimeBatchAction = {
+  action: 'batch';
+  events: RuntimeApiAction[];
+};
+
+export const runtimeRunActions = {
+  updateStatus(input: Omit<RuntimeRunStatusAction, 'action'>): RuntimeRunStatusAction {
+    return { action: 'update_run_status', ...input };
+  },
+} as const;
+
+export const runtimeSheetActions = {
+  startDataset(input: Omit<Extract<RuntimeSheetAction, { action: 'start_sheet_dataset' }>, 'action'>) {
+    return { action: 'start_sheet_dataset', ...input } satisfies RuntimeSheetAction;
+  },
+  persistCompletedRows(input: Omit<Extract<RuntimeSheetAction, { action: 'persist_completed_sheet_rows' }>, 'action'>) {
+    return { action: 'persist_completed_sheet_rows', ...input } satisfies RuntimeSheetAction;
+  },
+} as const;
+
+export const runtimeBillingActions = {
+  finalize(input: Omit<Extract<RuntimeBillingAction, { action: 'compute_billing_finalize' }>, 'action'>) {
+    return { action: 'compute_billing_finalize', ...input } satisfies RuntimeBillingAction;
+  },
+} as const;