deepline 0.1.119 → 0.1.121

package/dist/bundling-sources/shared_libs/play-runtime/runtime-actions.ts

@@ -0,0 +1,178 @@
+import type { ComputeBillingItem } from './worker-api-types';
+import type { DbLogicalTable, DbSessionOperation } from './db-session';
+import type { PlayRowUpdate } from './ctx-types';
+import type { PlaySheetContract } from '../plays/static-pipeline';
+import type { PlayRunLedgerEvent } from './run-ledger';
+
+export type RuntimeAppendRunEventsAction = {
+  action: 'append_run_events';
+  playId: string;
+  events: PlayRunLedgerEvent[];
+};
+
+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: DbLogicalTable;
+      };
+      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 }>;
+    };
+
+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;
+      reclaimRunning?: boolean;
+    }
+  | {
+      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
+  | RuntimeAppendRunEventsAction
+  | RuntimeResultsAction
+  | RuntimeBillingAction
+  | RuntimeReceiptAction;
+
+export type RuntimeBatchAction = {
+  action: 'batch';
+  events: RuntimeApiAction[];
+};
+
+export const runtimeRunActions = {
+  appendEvents(
+    input: Omit<RuntimeAppendRunEventsAction, 'action'>,
+  ): RuntimeAppendRunEventsAction {
+    return { action: 'append_run_events', ...input };
+  },
+} 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;