@civitai/blocks-react 0.76.1 → 0.78.0

package/dist/hooks/useRunTraining.d.ts

@@ -1,5 +1,5 @@
-import type { BlockRunTrainingHostError, BlockTrainingQuote, BlockWorkflowSnapshot, WorkflowBodyTraining } from '@civitai/app-sdk/blocks';
-export type { BlockRunTrainingHostError, BlockTrainingQuote, WorkflowBodyTraining };
+import type { BlockRunTrainingHostError, BlockTrainingDatasetItem, BlockTrainingQuote, BlockWorkflowSnapshot, WorkflowBodyTraining } from '@civitai/app-sdk/blocks';
+export type { BlockRunTrainingHostError, BlockTrainingDatasetItem, BlockTrainingQuote, WorkflowBodyTraining, };
 /**
  * A training body ready to RUN: {@link WorkflowBodyTraining} with the `quoteId`
  * its estimate returned. Required here so a body without one is a compile error
@@ -16,7 +16,7 @@ export type QuotedTrainingBody = WorkflowBodyTraining & {
  * a body that differs from the quoted one) arrive in the same field — see
  * {@link isRunTrainingErrorCode}.
  */
-export declare const RUN_TRAINING_ERROR_CODES: readonly ["review-mode", "block is not ready", "sign in to train", "invalid training request", "no block token", "declined", "submission-unconfirmed"];
+export declare const RUN_TRAINING_ERROR_CODES: readonly ["review-mode", "block is not ready", "sign in to train", "invalid training request", "no block token", "training data not uploaded here", "declined", "submission-unconfirmed"];
 /**
  * The `code` a {@link RunTrainingError} can carry: one of the host's closed
  * refusal codes, or `'refused'` — assigned by THIS SDK, never sent by the host,
@@ -49,9 +49,15 @@ export declare function isRunTrainingErrorCode(error: string): error is BlockRun
  *    `.snapshot` the refusal it arrived in. Retrying needs a new estimate and
  *    usually a later time; buying Buzz does not lift these caps.
  *  - `.signInRequired` — no session. Route into `useRequestSignIn()`.
+ *  - `.trainingDataNotUploaded` (`code: 'training data not uploaded here'`) —
+ *    the quote names an image THIS host did not upload for this block. The host
+ *    remembers only the blobs its own `useImageUpload({ purpose: 'trainingData' })`
+ *    uploaded, and forgets them on a page reload. Refused before any dialog:
+ *    NO RUN, nothing charged. Ask the viewer to upload the images again, then
+ *    estimate and run the new quote.
  *  - anything else — a refusal before any submit (an expired or used quote, a
- *    body that no longer matches its quote, an ineligible image). Estimate again
- *    before retrying; `.message` is the host's or server's text.
+ *    body that no longer matches its quote). Estimate again before retrying;
+ *    `.message` is the host's or server's text.
  */
 export declare class RunTrainingError extends Error {
     /**
@@ -78,6 +84,12 @@ export declare class RunTrainingError extends Error {
     readonly timedOut: boolean;
     /** There is no session (`sign in to train`, or a bare `UNAUTHORIZED`). */
     readonly signInRequired: boolean;
+    /**
+     * The quote names an image this host did not upload for this block
+     * (`training data not uploaded here`). No run, nothing charged: upload the
+     * images again, estimate again, and run that quote.
+     */
+    readonly trainingDataNotUploaded: boolean;
     constructor(error: string, opts?: {
         timedOut?: boolean;
         refusedSnapshot?: BlockWorkflowSnapshot;
@@ -112,15 +124,19 @@ export interface UseRunTraining {
  * viewer. Refused from `dev:live` and from review sessions — build the flow
  * against the mock host. See `WorkflowBodyTraining` in `@civitai/app-sdk`.
  *
- * The flow: `usePrepareTrainingDataset()` → `useBuzzWorkflow().estimate(body)`
- * (its `trainingQuote.quoteId`) → `runTraining({ ...body, quoteId })` →
- * `useBuzzWorkflow().watch(workflowId)`. Change nothing in the body between the
- * estimate and the run: the server checks it against the quoted one.
+ * The flow: `useImageUpload({ purpose: 'trainingData' }).open()` (the viewer
+ * uploads the images; you caption them into `items: [{ air, caption }]`) →
+ * `useBuzzWorkflow().estimate(body)` (its `trainingQuote.quoteId`) →
+ * `runTraining({ ...body, quoteId })` → `useBuzzWorkflow().watch(workflowId)`.
+ * Change nothing in the body between the estimate and the run: the server checks
+ * it against the quoted one. Upload and run in the same page session: the host
+ * refuses a quote naming an image it did not upload for this block.
  *
  * 🔴 THE CONSENT DIALOG IS CIVITAI'S, AND IT IS THE CHARGE. Every number on it is
  * read back from the server's stored quote — never from your body. A run may cost
  * more than the token's per-call budget (the viewer confirms the exact price), up
- * to `BLOCK_TRAINING_MAX_BUZZ_PER_RUN` (5,000 Buzz). Images are the viewer's own.
+ * to `BLOCK_TRAINING_MAX_BUZZ_PER_RUN` (5,000 Buzz). Images are the ones the
+ * viewer uploaded through the host.
  *
  * 🔴 NO AUTOMATIC CONSENT RETRY, unlike the other consent-gated hooks — on
  * purpose. `RUN_TRAINING` carries no idempotency key and has an outcome
@@ -132,8 +148,12 @@ export interface UseRunTraining {
  * the viewer clicks or dismisses its dialog.
  *
  * @example
+ * const { open } = useImageUpload({ purpose: 'trainingData' });
  * const { estimate, watch } = useBuzzWorkflow();
  * const { runTraining } = useRunTraining();
+ * const images = await open();
+ * if (!images) return;                              // cancelled
+ * const body = { ...settings, items: images.map((i) => ({ air: i.air, caption: captionFor(i) })) };
  * const quote = (await estimate(body)).trainingQuote!;
  * try {
  *   const started = await runTraining({ ...body, quoteId: quote.quoteId });
@@ -142,6 +162,7 @@ export interface UseRunTraining {
  *   if (!(e instanceof RunTrainingError)) throw e;
  *   if (e.declined) return;                          // no run — say nothing
  *   if (e.signInRequired) return requestSignIn();
+ *   if (e.trainingDataNotUploaded) return askToUploadAgain(); // no run — re-upload, re-estimate
  *   if (e.unconfirmed) return showCheckYourTrainings(); // may be running — never auto-retry
  *   if (e.refused) return showError(e.message); // a cap / availability refusal; no run, nothing charged
  *   showError('Could not start training. Get a new price and try again.');

package/dist/hooks/useRunTraining.js

@@ -17,9 +17,12 @@ export const RUN_TRAINING_ERROR_CODES = [
     'sign in to train',
     'invalid training request',
     'no block token',
+    'training data not uploaded here',
     'declined',
     'submission-unconfirmed',
 ];
+/** The host code for a quote naming an image this host did not upload — see {@link RunTrainingError.trainingDataNotUploaded}. */
+const NOT_UPLOADED_HERE = 'training data not uploaded here';
 const CODE_SET = new Set(RUN_TRAINING_ERROR_CODES);
 /**
  * The `workflowId` the host's training submit stamps on a snapshot that is NOT a
@@ -55,9 +58,15 @@ export function isRunTrainingErrorCode(error) {
  *    `.snapshot` the refusal it arrived in. Retrying needs a new estimate and
  *    usually a later time; buying Buzz does not lift these caps.
  *  - `.signInRequired` — no session. Route into `useRequestSignIn()`.
+ *  - `.trainingDataNotUploaded` (`code: 'training data not uploaded here'`) —
+ *    the quote names an image THIS host did not upload for this block. The host
+ *    remembers only the blobs its own `useImageUpload({ purpose: 'trainingData' })`
+ *    uploaded, and forgets them on a page reload. Refused before any dialog:
+ *    NO RUN, nothing charged. Ask the viewer to upload the images again, then
+ *    estimate and run the new quote.
  *  - anything else — a refusal before any submit (an expired or used quote, a
- *    body that no longer matches its quote, an ineligible image). Estimate again
- *    before retrying; `.message` is the host's or server's text.
+ *    body that no longer matches its quote). Estimate again before retrying;
+ *    `.message` is the host's or server's text.
  */
 export class RunTrainingError extends Error {
     /**
@@ -84,6 +93,12 @@ export class RunTrainingError extends Error {
     timedOut;
     /** There is no session (`sign in to train`, or a bare `UNAUTHORIZED`). */
     signInRequired;
+    /**
+     * The quote names an image this host did not upload for this block
+     * (`training data not uploaded here`). No run, nothing charged: upload the
+     * images again, estimate again, and run that quote.
+     */
+    trainingDataNotUploaded;
     constructor(error, opts) {
         super(error);
         this.name = 'RunTrainingError';
@@ -99,6 +114,7 @@ export class RunTrainingError extends Error {
         this.declined = error === 'declined';
         this.unconfirmed = this.timedOut || error === 'submission-unconfirmed';
         this.signInRequired = error === 'sign in to train' || error === 'UNAUTHORIZED';
+        this.trainingDataNotUploaded = error === NOT_UPLOADED_HERE;
     }
 }
 /**
@@ -110,15 +126,19 @@ export class RunTrainingError extends Error {
  * viewer. Refused from `dev:live` and from review sessions — build the flow
  * against the mock host. See `WorkflowBodyTraining` in `@civitai/app-sdk`.
  *
- * The flow: `usePrepareTrainingDataset()` → `useBuzzWorkflow().estimate(body)`
- * (its `trainingQuote.quoteId`) → `runTraining({ ...body, quoteId })` →
- * `useBuzzWorkflow().watch(workflowId)`. Change nothing in the body between the
- * estimate and the run: the server checks it against the quoted one.
+ * The flow: `useImageUpload({ purpose: 'trainingData' }).open()` (the viewer
+ * uploads the images; you caption them into `items: [{ air, caption }]`) →
+ * `useBuzzWorkflow().estimate(body)` (its `trainingQuote.quoteId`) →
+ * `runTraining({ ...body, quoteId })` → `useBuzzWorkflow().watch(workflowId)`.
+ * Change nothing in the body between the estimate and the run: the server checks
+ * it against the quoted one. Upload and run in the same page session: the host
+ * refuses a quote naming an image it did not upload for this block.
  *
  * 🔴 THE CONSENT DIALOG IS CIVITAI'S, AND IT IS THE CHARGE. Every number on it is
  * read back from the server's stored quote — never from your body. A run may cost
  * more than the token's per-call budget (the viewer confirms the exact price), up
- * to `BLOCK_TRAINING_MAX_BUZZ_PER_RUN` (5,000 Buzz). Images are the viewer's own.
+ * to `BLOCK_TRAINING_MAX_BUZZ_PER_RUN` (5,000 Buzz). Images are the ones the
+ * viewer uploaded through the host.
  *
  * 🔴 NO AUTOMATIC CONSENT RETRY, unlike the other consent-gated hooks — on
  * purpose. `RUN_TRAINING` carries no idempotency key and has an outcome
@@ -130,8 +150,12 @@ export class RunTrainingError extends Error {
  * the viewer clicks or dismisses its dialog.
  *
  * @example
+ * const { open } = useImageUpload({ purpose: 'trainingData' });
  * const { estimate, watch } = useBuzzWorkflow();
  * const { runTraining } = useRunTraining();
+ * const images = await open();
+ * if (!images) return;                              // cancelled
+ * const body = { ...settings, items: images.map((i) => ({ air: i.air, caption: captionFor(i) })) };
  * const quote = (await estimate(body)).trainingQuote!;
  * try {
  *   const started = await runTraining({ ...body, quoteId: quote.quoteId });
@@ -140,6 +164,7 @@ export class RunTrainingError extends Error {
  *   if (!(e instanceof RunTrainingError)) throw e;
  *   if (e.declined) return;                          // no run — say nothing
  *   if (e.signInRequired) return requestSignIn();
+ *   if (e.trainingDataNotUploaded) return askToUploadAgain(); // no run — re-upload, re-estimate
  *   if (e.unconfirmed) return showCheckYourTrainings(); // may be running — never auto-retry
  *   if (e.refused) return showError(e.message); // a cap / availability refusal; no run, nothing charged
  *   showError('Could not start training. Get a new price and try again.');

package/dist/index.d.ts

@@ -57,6 +57,8 @@ export { useDailyCompensation } from './hooks/useDailyCompensation.js';
 export type { UseDailyCompensation } from './hooks/useDailyCompensation.js';
 export { useWildcardPack, WildcardPackError } from './hooks/useWildcardPack.js';
 export type { UseWildcardPack } from './hooks/useWildcardPack.js';
+export { useResourceFile, ResourceFileError } from './hooks/useResourceFile.js';
+export type { UseResourceFile } from './hooks/useResourceFile.js';
 export { useAppWorkflows } from './hooks/useAppWorkflows.js';
 export type { UseAppWorkflows } from './hooks/useAppWorkflows.js';
 export { usePublishGenerationOutputs } from './hooks/usePublishGenerationOutputs.js';
@@ -73,19 +75,17 @@ export { useReactions, ReactionError, REACTIONS, REACTION_ERROR_CODES, isReactio
 export type { UseReactions, UseReactionsOptions } from './hooks/useReactions.js';
 export { useCreatePostFromApp, CreatePostError, CREATE_POST_ERROR_CODES, isCreatePostErrorCode, } from './hooks/useCreatePostFromApp.js';
 export type { UseCreatePostFromApp, BlockPostSource, BlockCreatePostRequest, BlockCreatePostResult, BlockCreatePostHostError, } from './hooks/useCreatePostFromApp.js';
-export { usePrepareTrainingDataset, PrepareTrainingDatasetError, PREPARE_TRAINING_DATASET_ERROR_CODES, isPrepareTrainingDatasetErrorCode, } from './hooks/usePrepareTrainingDataset.js';
-export type { UsePrepareTrainingDataset, BlockPrepareTrainingDatasetHostError, BlockTrainingDatasetItem, BlockTrainingDatasetResult, BlockTrainingRejectionReason, } from './hooks/usePrepareTrainingDataset.js';
 export { useRunTraining, RunTrainingError, RUN_TRAINING_ERROR_CODES, isRunTrainingErrorCode, } from './hooks/useRunTraining.js';
-export type { UseRunTraining, QuotedTrainingBody, RunTrainingErrorCode, BlockRunTrainingHostError, BlockTrainingQuote, WorkflowBodyTraining, } from './hooks/useRunTraining.js';
+export type { UseRunTraining, QuotedTrainingBody, RunTrainingErrorCode, BlockRunTrainingHostError, BlockTrainingQuote, BlockTrainingDatasetItem, WorkflowBodyTraining, } from './hooks/useRunTraining.js';
 export { useRunConfirmedStep, RunConfirmedStepError, RUN_CONFIRMED_STEP_ERROR_CODES, isRunConfirmedStepErrorCode, isStepQuoteId, } from './hooks/useRunConfirmedStep.js';
 export type { UseRunConfirmedStep, QuotedStepBody, RunConfirmedStepErrorCode, BlockRunConfirmedStepHostError, BlockStepQuote, WorkflowBodyPassThroughStep, } from './hooks/useRunConfirmedStep.js';
-export type { BlockBuzzTransaction, BlockBuzzAccount, BlockDailyCompensationResource, BlockWildcardPack, BlockWildcardPackErrorCode, AppWorkflow, AppWorkflowImage, AppWorkflowsParams, BlockGatedImage, BlockCollectionFollowErrorCode, BlockCollectionFollowResult, BlockAppReaction, BlockStoredReaction, BlockReactionEntityType, BlockEntityReactionErrorCode, BlockReactionCounts, BlockEntityReactionState, BlockEntityReactionResult, BlockEntityReactionsResult, } from '@civitai/app-sdk/blocks';
+export type { BlockBuzzTransaction, BlockBuzzAccount, BlockDailyCompensationResource, BlockWildcardPack, BlockWildcardPackErrorCode, BlockResourceFile, BlockResourceFileEntry, BlockResourceFileEntryFormat, BlockResourceFileErrorCode, BlockResourceFileMeta, AppWorkflow, AppWorkflowImage, AppWorkflowsParams, BlockGatedImage, BlockCollectionFollowErrorCode, BlockCollectionFollowResult, BlockAppReaction, BlockStoredReaction, BlockReactionEntityType, BlockEntityReactionErrorCode, BlockReactionCounts, BlockEntityReactionState, BlockEntityReactionResult, BlockEntityReactionsResult, } from '@civitai/app-sdk/blocks';
 export { useCheckpointPicker } from './hooks/useCheckpointPicker.js';
 export type { UseCheckpointPicker } from './hooks/useCheckpointPicker.js';
 export { useResourcePicker, RESOURCE_PICKER_MULTIPLE_MAX } from './hooks/useResourcePicker.js';
 export type { UseResourcePicker, ResourcePickerOpenOptions, ResourcePickerMultiple, } from './hooks/useResourcePicker.js';
 export { useImageUpload } from './hooks/useImageUpload.js';
-export type { UseImageUpload, UseImageUploadAsyncScan, UseImageUploadGenerationSource, UseImageUploadOptions, } from './hooks/useImageUpload.js';
+export type { UseImageUpload, UseImageUploadAsyncScan, UseImageUploadGenerationSource, UseImageUploadTrainingData, UseImageUploadOptions, BlockTrainingDataImage, } from './hooks/useImageUpload.js';
 export { useUploadImageBytes } from './hooks/useUploadImageBytes.js';
 export type { UploadImageBytesOptions, UseUploadImageBytes, } from './hooks/useUploadImageBytes.js';
 export { useMediaUpload } from './hooks/useMediaUpload.js';

package/dist/index.js

@@ -46,6 +46,7 @@ export { useBuzzTransactions } from './hooks/useBuzzTransactions.js';
 export { useBuzzAccounts } from './hooks/useBuzzAccounts.js';
 export { useDailyCompensation } from './hooks/useDailyCompensation.js';
 export { useWildcardPack, WildcardPackError } from './hooks/useWildcardPack.js';
+export { useResourceFile, ResourceFileError } from './hooks/useResourceFile.js';
 export { useAppWorkflows } from './hooks/useAppWorkflows.js';
 export { usePublishGenerationOutputs } from './hooks/usePublishGenerationOutputs.js';
 export { useGatedImages } from './hooks/useGatedImages.js';
@@ -62,7 +63,6 @@ export { InvalidIdempotencyKeyError } from './transport/transport.js';
 export { useCollectionFollow, CollectionFollowError, COLLECTION_FOLLOW_ERROR_CODES, isCollectionFollowErrorCode, } from './hooks/useCollectionFollow.js';
 export { useReactions, ReactionError, REACTIONS, REACTION_ERROR_CODES, isReactionErrorCode, } from './hooks/useReactions.js';
 export { useCreatePostFromApp, CreatePostError, CREATE_POST_ERROR_CODES, isCreatePostErrorCode, } from './hooks/useCreatePostFromApp.js';
-export { usePrepareTrainingDataset, PrepareTrainingDatasetError, PREPARE_TRAINING_DATASET_ERROR_CODES, isPrepareTrainingDatasetErrorCode, } from './hooks/usePrepareTrainingDataset.js';
 export { useRunTraining, RunTrainingError, RUN_TRAINING_ERROR_CODES, isRunTrainingErrorCode, } from './hooks/useRunTraining.js';
 export { useRunConfirmedStep, RunConfirmedStepError, RUN_CONFIRMED_STEP_ERROR_CODES, isRunConfirmedStepErrorCode, isStepQuoteId, } from './hooks/useRunConfirmedStep.js';
 export { useCheckpointPicker } from './hooks/useCheckpointPicker.js';

package/dist/internal/liveHost.d.ts

@@ -113,12 +113,9 @@
  *     `entity-unavailable`.
  *   • CREATE_POST_FROM_APP — needs the server-resolved preview plus host-chrome
  *     confirm before a PUBLIC post is written. Replies with a refusal.
- *   • PREPARE_TRAINING_DATASET — the server refuses every `kind: 'training'`
- *     request from a dev token, so there is nothing to forward. Replies with a
- *     refusal.
- *   • RUN_TRAINING — the same dev-token refusal, plus the host-chrome consent
- *     dialog a run is confirmed in, which this harness does not have. Replies
- *     with a refusal.
+ *   • RUN_TRAINING — the server refuses every `kind: 'training'` request from a
+ *     dev token, and a run is confirmed in a host-chrome consent dialog this
+ *     harness does not have. Replies with a refusal.
  *   • RUN_CONFIRMED_STEP — the server refuses a confirmed step from a dev token
  *     and the estimate never returns a `stepQuote` for one, and the price is
  *     confirmed in host chrome this harness does not have. Replies with a
@@ -130,6 +127,9 @@
  *     `publishError` options. Replies with a refusal.
  *   • GET_WILDCARD_PACK — needs the session-authed resolve plus the in-tab
  *     zip/yaml parse that lives in civitai, not this SDK. Replies `parse-failed`.
+ *   • GET_RESOURCE_FILE — the same session-authed resolve plus in-tab
+ *     fetch/unzip/parse, and only full-page app hosts answer it. Replies
+ *     `parse-failed`.
  *   • OPEN_IMAGE_UPLOAD — needs the host's native modal + session-authed byte
  *     pipeline. Replies DISMISSED (the hook resolves `null`).
  *   • OPEN_IMAGE_PICKER — needs the host's picker over the viewer's own posted
@@ -239,8 +239,7 @@ export interface LiveHostOptions {
     manifest?: unknown;
 }
 /**
- * The reply both training bridges get in dev:live — see their handlers. One
- * string so the two cannot drift.
+ * The reply `RUN_TRAINING` gets in dev:live — see its handler.
  */
 export declare const TRAINING_UNSUPPORTED_ERROR = "training is not supported in dev:live \u2014 use dev:mock";
 /**

package/dist/internal/liveHost.js

@@ -113,12 +113,9 @@
  *     `entity-unavailable`.
  *   • CREATE_POST_FROM_APP — needs the server-resolved preview plus host-chrome
  *     confirm before a PUBLIC post is written. Replies with a refusal.
- *   • PREPARE_TRAINING_DATASET — the server refuses every `kind: 'training'`
- *     request from a dev token, so there is nothing to forward. Replies with a
- *     refusal.
- *   • RUN_TRAINING — the same dev-token refusal, plus the host-chrome consent
- *     dialog a run is confirmed in, which this harness does not have. Replies
- *     with a refusal.
+ *   • RUN_TRAINING — the server refuses every `kind: 'training'` request from a
+ *     dev token, and a run is confirmed in a host-chrome consent dialog this
+ *     harness does not have. Replies with a refusal.
  *   • RUN_CONFIRMED_STEP — the server refuses a confirmed step from a dev token
  *     and the estimate never returns a `stepQuote` for one, and the price is
  *     confirmed in host chrome this harness does not have. Replies with a
@@ -130,6 +127,9 @@
  *     `publishError` options. Replies with a refusal.
  *   • GET_WILDCARD_PACK — needs the session-authed resolve plus the in-tab
  *     zip/yaml parse that lives in civitai, not this SDK. Replies `parse-failed`.
+ *   • GET_RESOURCE_FILE — the same session-authed resolve plus in-tab
+ *     fetch/unzip/parse, and only full-page app hosts answer it. Replies
+ *     `parse-failed`.
  *   • OPEN_IMAGE_UPLOAD — needs the host's native modal + session-authed byte
  *     pipeline. Replies DISMISSED (the hook resolves `null`).
  *   • OPEN_IMAGE_PICKER — needs the host's picker over the viewer's own posted
@@ -174,8 +174,7 @@ const DEFAULT_BACKEND_BASE_URL = 'https://civitai.com';
 /** The budgeted-spend scope the workflow procedures require. */
 const BUDGETED_SCOPE = 'ai:write:budgeted';
 /**
- * The reply both training bridges get in dev:live — see their handlers. One
- * string so the two cannot drift.
+ * The reply `RUN_TRAINING` gets in dev:live — see its handler.
  */
 export const TRAINING_UNSUPPORTED_ERROR = 'training is not supported in dev:live — use dev:mock';
 /** The once-per-session console line for a refused training bridge. */
@@ -1139,35 +1138,25 @@ export function createLiveHost(options) {
                         });
                         return;
                     }
-                    case 'PREPARE_TRAINING_DATASET': {
+                    case 'RUN_TRAINING': {
                         // UNSUPPORTED in live, and here the SERVER decides it, not this
                         // harness: every `kind: 'training'` procedure refuses a dev token
                         // ("training is not available from a development or review session"
                         // — `assertTrainingRequestAllowed` in civitai/civitai), because there
                         // is no host page whose chrome can show the viewer a price. A
                         // training ESTIMATE is still forwarded (ESTIMATE_WORKFLOW above) and
-                        // comes back carrying that same server refusal.
+                        // comes back carrying that same server refusal. A run is also
+                        // CHARGED and confirmed in host chrome this harness does not have, so
+                        // faking either half would let dev prove out a flow production does
+                        // not have. The upload half (`OPEN_IMAGE_UPLOAD { purpose:
+                        // 'trainingData' }`) is answered as a dismissal (`null`), like every
+                        // other upload mode — see OPEN_IMAGE_UPLOAD below.
                         //
                         // FREE TEXT on the message's own reply channel; no host code is
                         // honest about "this harness cannot do it". Use dev:mock — its
-                        // `trainingDatasetError` / `trainingDatasetRejected` knobs cover the
-                        // refusal set.
-                        if (!isRoutableRequestId(requestId))
-                            return;
-                        logOnce('training-dataset', trainingUnsupportedLog('PREPARE_TRAINING_DATASET'));
-                        dispatchToBlock({
-                            type: 'TRAINING_DATASET_RESULT',
-                            payload: { requestId, error: TRAINING_UNSUPPORTED_ERROR },
-                        });
-                        return;
-                    }
-                    case 'RUN_TRAINING': {
-                        // UNSUPPORTED in live for the same server-side reason as
-                        // PREPARE_TRAINING_DATASET directly above — and doubly so: a run is
-                        // CHARGED and confirmed in host chrome this harness does not have, so
-                        // faking either half would let dev prove out a flow production does
-                        // not have. Use dev:mock — its `runTrainingError` knob covers
-                        // `declined`, `submission-unconfirmed` and the rest.
+                        // `cannedTrainingDataUpload`, `trainingDataNotUploadedHere` and
+                        // `runTrainingError` knobs cover the upload, the not-uploaded-here
+                        // refusal, `declined`, `submission-unconfirmed` and the rest.
                         if (!isRoutableRequestId(requestId))
                             return;
                         logOnce('training-run', trainingUnsupportedLog('RUN_TRAINING'));
@@ -1221,6 +1210,26 @@ export function createLiveHost(options) {
                         });
                         return;
                     }
+                    case 'GET_RESOURCE_FILE': {
+                        // UNSUPPORTED in live v1, for the same reason as GET_WILDCARD_PACK
+                        // directly above: the real host resolves the file through a
+                        // SESSION-authed procedure and fetches + unzips + parses it in the
+                        // viewer's page tab. A liveHost holds a BLOCK token, not a session,
+                        // so it cannot resolve the file. Reply with an honest `parse-failed`
+                        // (never a fabricated file) so the hook surfaces a typed error rather
+                        // than hanging. The closed error set has no "unsupported" code, and
+                        // `parse-failed` is the one the wildcard refusal already uses.
+                        if (!isRoutableRequestId(requestId))
+                            return;
+                        logOnce('resource-file', 'GET_RESOURCE_FILE is not supported in dev:live (it needs the session-authed ' +
+                            'resolve + in-tab fetch and unzip, on a full-page app host). Replying ' +
+                            '`parse-failed`. Use dev:mock to exercise the resource-file path.');
+                        dispatchToBlock({
+                            type: 'RESOURCE_FILE_RESULT',
+                            payload: { requestId, error: 'parse-failed' },
+                        });
+                        return;
+                    }
                     case 'QUERY_APP_WORKFLOWS': {
                         // App generator SUBQUEUE read via the token-bound
                         // `blocks.queryAppWorkflows` MUTATION (POST). params (cursor/limit) are
@@ -1391,6 +1400,12 @@ export function createLiveHost(options) {
                                 'Test the bytes upload against the real site, or use ' +
                                 'createMockHost({ uploadImageBytesResult }) in dev:mock.');
                         }
+                        else if (typed.payload?.purpose === 'trainingData') {
+                            logOnce('open-image-upload-training-data', "OPEN_IMAGE_UPLOAD { purpose: 'trainingData' } received but live dev mode " +
+                                'cannot serve the training-data upload modal — resolving the upload as cancelled (null). ' +
+                                'Test the training flow against the real site, or use ' +
+                                'createMockHost({ cannedTrainingDataUpload }) in dev:mock.');
+                        }
                         else {
                             logOnce('open-image-upload', 'OPEN_IMAGE_UPLOAD received but live dev mode cannot serve the native upload ' +
                                 'modal — resolving the upload as dismissed (null). Test the image-upload flow ' +

package/dist/internal/mockHost.d.ts

@@ -37,7 +37,7 @@
  * called). Only the postMessage bridge round-trips are exercised. Never import
  * this from production code.
  */
-import { type BlockContext, type BlockResourceInfo, type BlockResourcePickerType, type BlockUploadedImageInfo, type BlockGenerationSourceImageInfo, type BuzzAccountType, type BlockBuzzTransaction, type BlockBuzzAccount, type BlockDailyCompensationResource, type BlockViewer, type BlockWildcardPack, type BlockWildcardPackErrorCode, type AppWorkflow, type BlockModel3dOutput, type BlockPublishedModel, type BlockRunConfirmedStepHostError, type BlockGatedImage, type BlockPickedImageInfo, type BlockCollectionFollowErrorCode, type BlockEntityReactionErrorCode, type BlockCreatePostHostError, type BlockCreatePostResult, type BlockPrepareTrainingDatasetHostError, type BlockRunTrainingHostError, type BlockTrainingRejectionReason, type ColorDomain, type SharedStorageValue, type Theme, type ViewerInfo, type WorkflowBody } from '@civitai/app-sdk/blocks';
+import { type BlockContext, type BlockResourceInfo, type BlockResourcePickerType, type BlockUploadedImageInfo, type BlockGenerationSourceImageInfo, type BuzzAccountType, type BlockBuzzTransaction, type BlockBuzzAccount, type BlockDailyCompensationResource, type BlockViewer, type BlockWildcardPack, type BlockWildcardPackErrorCode, type BlockResourceFile, type BlockResourceFileErrorCode, type AppWorkflow, type BlockModel3dOutput, type BlockPublishedModel, type BlockRunConfirmedStepHostError, type BlockGatedImage, type BlockPickedImageInfo, type BlockCollectionFollowErrorCode, type BlockEntityReactionErrorCode, type BlockCreatePostHostError, type BlockCreatePostResult, type BlockRunTrainingHostError, type BlockTrainingDataImage, type ColorDomain, type SharedStorageValue, type Theme, type ViewerInfo, type WorkflowBody } from '@civitai/app-sdk/blocks';
 import { type MockMediaUploads } from './mediaUploadBytes.js';
 import { type MockWorkflowQuoteFailure } from './mockWorkflowGraph.js';
 import { type MockReactionSeed } from './mockHostReactions.js';
@@ -412,7 +412,7 @@ export interface MockGenerationScenario {
  */
 export declare const MOCK_TRAINING_BOUNDS: {
     readonly datasetMaxItems: 50;
-    readonly captionMaxChars: 1000;
+    readonly captionMaxChars: 8192;
     readonly maxBuzzPerRun: 5000;
 };
 /**
@@ -425,16 +425,15 @@ export declare const MOCK_TRAINING_BOUNDS: {
 export declare const MOCK_SPEND_SCOPE_ERROR = "block lacks ai:write:budgeted scope";
 /** The real server's refusal for an unknown, expired or spent quote on `RUN_TRAINING`. */
 export declare const MOCK_TRAINING_QUOTE_GONE_ERROR = "training quote not found, expired or already used \u2014 estimate again";
-/** The real server's refusal for a training estimate naming a dataset it does not hold. */
-export declare const MOCK_TRAINING_DATASET_GONE_ERROR = "training dataset not found or expired";
 /**
- * The real server's two refusals for a dataset with NOTHING admitted
- * (`prepareBlockTrainingDataset`): every image failed eligibility, or every
- * eligible image failed its import. The server THROWS rather than returning
- * `count: 0`, so a successful reply always admits at least one image.
+ * The estimate's refusals for `items` outside the server's schema. The first two
+ * are the server's own custom messages (`blockTrainingDatasetItemsSchema`); the
+ * third stands for every other zod issue (count, caption length, missing or extra
+ * keys), whose real text is zod's formatting and not a contract.
  */
-export declare const MOCK_TRAINING_NONE_ELIGIBLE_ERROR = "none of the requested images can be used for training";
-export declare const MOCK_TRAINING_NONE_IMPORTED_ERROR = "none of the requested images could be prepared for training";
+export declare const MOCK_TRAINING_AIR_ERROR = "air must be a consumer blob key or its Civitai-hosted blob url";
+export declare const MOCK_TRAINING_DUPLICATE_ERROR = "a training dataset names each image once";
+export declare const MOCK_TRAINING_ITEMS_ERROR = "training items must be 1 to 50 { air, caption } entries with captions of at most 8192 characters";
 /**
  * The per-job ceiling of a confirmed step — a COPY of `@civitai/app-sdk/blocks`'s
  * `BLOCK_CONFIRMED_STEP_MAX_BUZZ_PER_JOB`, not value-imported for the reason
@@ -932,6 +931,21 @@ export interface MockHostOptions {
      * returned. Live-tunable via {@link MockHost.setScenario}.
      */
     wildcardPackError?: BlockWildcardPackErrorCode;
+    /**
+     * The parsed file returned from `GET_RESOURCE_FILE` (what `useResourceFile`
+     * reads). Absent → {@link DEFAULT_RESOURCE_FILE}, one UI-format graph. Ignored
+     * when {@link resourceFileError} is set. The mock returns it whatever
+     * `modelVersionId` / `fileId` the block asks for.
+     */
+    resourceFile?: BlockResourceFile;
+    /**
+     * Force `GET_RESOURCE_FILE` to reply with this code from the CLOSED set
+     * (`not-found` | `forbidden` | `unauthorized` | `too-large` | `parse-failed` |
+     * `busy` | `rate-limited`) instead of a file — exercises `useResourceFile`'s
+     * error UI. Absent → a file is returned. Live-tunable via
+     * {@link MockHost.setScenario}, where passing the key as `undefined` clears it.
+     */
+    resourceFileError?: BlockResourceFileErrorCode;
     /**
      * Force `SET_COLLECTION_FOLLOW` to reply with an `error` instead of writing —
      * exercises `useCollectionFollow`'s refusal handling. Pass a
@@ -1014,41 +1028,34 @@ export interface MockHostOptions {
      */
     createPostError?: BlockCreatePostHostError | string;
     /**
-     * Images `PREPARE_TRAINING_DATASET` leaves out, with their reasons — what
-     * `usePrepareTrainingDataset()` reports on `rejected`. Only entries whose
-     * `imageId` the request actually named are reported, and `count` is the rest,
-     * exactly as the server derives it. Absent → every image is admitted.
+     * The images an `OPEN_IMAGE_UPLOAD { purpose: 'trainingData' }` returns — what
+     * `useImageUpload({ purpose: 'trainingData' }).open()` resolves with. Absent →
+     * {@link DEFAULT_TRAINING_DATA_UPLOAD_COUNT} images with FRESH keys on every
+     * upload, so two uploads never collide. `null` (or `[]`) → the viewer cancelled.
      * Live-tunable via {@link MockHost.setScenario}.
      *
-     * When this rejects EVERY named image the reply is an `error`, as on the real
-     * server — {@link MOCK_TRAINING_NONE_ELIGIBLE_ERROR}, or
-     * {@link MOCK_TRAINING_NONE_IMPORTED_ERROR} when every image that passed
-     * eligibility was rejected with an `import-*` reason. A success never has
-     * `count: 0`.
-     *
-     * KIND-FAITHFUL, like the rest of the training path: the mock holds the
-     * datasets it prepared, an estimate naming any other `datasetId` fails as the
-     * server's does, a quote is run at most once, and every training call needs
-     * `ai:write:budgeted` on the token ({@link MockHostOptions.consentGranted}) and
-     * a signed-in viewer. What it cannot model: image ownership and moderation (it
-     * has no images), the flag, and the page-only / dev-token refusals.
-     *
-     * ⚠️ The quote↔body check is NARROWER than the server's: the mock compares only
-     * `datasetId`, while the server hashes the whole body (minus `quoteId`) and
-     * refuses any change. A body edited between estimate and run passes here and
-     * fails in production.
-     */
-    trainingDatasetRejected?: Array<{
-        imageId: number;
-        reason: BlockTrainingRejectionReason;
-    }>;
-    /**
-     * Force `PREPARE_TRAINING_DATASET` to reply with an `error` — a
-     * {@link BlockPrepareTrainingDatasetHostError} for a host refusal, or any other
-     * string for the free-text server-message variant. Absent → it prepares.
-     * Live-tunable via {@link MockHost.setScenario}.
+     * The mock REMEMBERS every key it hands out, per install, as the real host
+     * does per block instance: `RUN_TRAINING` refuses a quote naming any other key
+     * with `training data not uploaded here`, before any other check, and keeps the
+     * quote. Keys compare case-insensitively; an item may name its blob by the
+     * `url` too.
+     *
+     * KIND-FAITHFUL, like the rest of the training path: the estimate refuses items
+     * outside the server's schema, a quote is run at most once and is checked
+     * against the whole body it was quoted for (minus `quoteId`), and every
+     * training call needs `ai:write:budgeted` on the token
+     * ({@link MockHostOptions.consentGranted}) and a signed-in viewer. What it
+     * cannot model: blob storage, the Civitai-hosted-url rule on `air`, the flag,
+     * and the page-only / dev-token refusals.
+     */
+    cannedTrainingDataUpload?: BlockTrainingDataImage[] | null;
+    /**
+     * Make `RUN_TRAINING` refuse EVERY quote with `training data not uploaded
+     * here`, as the real host does after a page reload forgot the uploads. Settles
+     * before the server claims the quote, so the quote survives. Live-tunable via
+     * {@link MockHost.setScenario}; `undefined` clears it.
      */
-    trainingDatasetError?: BlockPrepareTrainingDatasetHostError | string;
+    trainingDataNotUploadedHere?: boolean;
     /**
      * The Buzz total a `kind: 'training'` estimate quotes. Default `500`. Above
      * `BLOCK_TRAINING_MAX_BUZZ_PER_RUN` (5,000) the estimate fails exactly as the
@@ -1064,7 +1071,8 @@ export interface MockHostOptions {
      *
      * Whether the QUOTE survives follows the real flow: the host's own gate and
      * dialog codes (`declined`, `review-mode`, `block is not ready`, `sign in to
-     * train`, `invalid training request`, `no block token`) settle before the
+     * train`, `invalid training request`, `no block token`, `training data not
+     * uploaded here`) settle before the
      * server claims the quote, so it stays runnable; `submission-unconfirmed` and
      * any server message settle after the claim, so it is used up — re-running it
      * gets the quote-gone error. Live-tunable via {@link MockHost.setScenario};
@@ -1431,7 +1439,7 @@ export interface MockHostOptions {
  * {@link MockHost.setTheme}, which pushes a real `THEME_CHANGE` — the mock of
  * the viewer toggling dark mode with the block already mounted.)
  */
-export type MockHostScenarioPatch = Pick<MockHostOptions, 'consentGrantable' | 'failMode' | 'cost' | 'pollsUntilDone' | 'cannedPicks' | 'cannedMultiPicks' | 'cannedImagePicks' | 'cannedImageUpload' | 'cannedGenerationSourceUpload' | 'cannedImageScan' | 'generation' | 'buzz' | 'storage' | 'shared' | 'buzzBalanceError' | 'viewerResult' | 'viewerError' | 'buzzTransactions' | 'buzzAccounts' | 'dailyCompensation' | 'buzzReadError' | 'wildcardPack' | 'wildcardPackError' | 'collectionFollowError' | 'reactionStates' | 'reactionError' | 'createPostResult' | 'createPostError' | 'trainingDatasetRejected' | 'trainingDatasetError' | 'trainingQuoteTotal' | 'runTrainingError' | 'runTrainingCapRefusal' | 'runConfirmedStepError' | 'runConfirmedStepCapRefusal' | 'appWorkflows' | 'appWorkflowsError' | 'publishImageIds' | 'publishError' | 'gatedImages' | 'gatedImagesError' | 'disallowedAccountTypes' | 'saveImageError' | 'uploadImageBytesResult' | 'uploadImageBytesError' | 'cannedMediaUploads' | 'mediaUploadError'>;
+export type MockHostScenarioPatch = Pick<MockHostOptions, 'consentGrantable' | 'failMode' | 'cost' | 'pollsUntilDone' | 'cannedPicks' | 'cannedMultiPicks' | 'cannedImagePicks' | 'cannedImageUpload' | 'cannedGenerationSourceUpload' | 'cannedImageScan' | 'generation' | 'buzz' | 'storage' | 'shared' | 'buzzBalanceError' | 'viewerResult' | 'viewerError' | 'buzzTransactions' | 'buzzAccounts' | 'dailyCompensation' | 'buzzReadError' | 'wildcardPack' | 'wildcardPackError' | 'resourceFile' | 'resourceFileError' | 'collectionFollowError' | 'reactionStates' | 'reactionError' | 'createPostResult' | 'createPostError' | 'cannedTrainingDataUpload' | 'trainingDataNotUploadedHere' | 'trainingQuoteTotal' | 'runTrainingError' | 'runTrainingCapRefusal' | 'runConfirmedStepError' | 'runConfirmedStepCapRefusal' | 'appWorkflows' | 'appWorkflowsError' | 'publishImageIds' | 'publishError' | 'gatedImages' | 'gatedImagesError' | 'disallowedAccountTypes' | 'saveImageError' | 'uploadImageBytesResult' | 'uploadImageBytesError' | 'cannedMediaUploads' | 'mediaUploadError'>;
 /** Runtime Buzz-balance handle exposed on {@link MockHost.buzz}. */
 export interface MockBuzzHandle {
     /** Current simulated balance, or `undefined` when balance isn't simulated. */
@@ -1447,6 +1455,7 @@ export interface MockBuzzHandle {
  *
  * After install, a harness UI can drive scenarios live:
  *  - `setScenario(patch)` — merge new generation/buzz/storage/failMode controls.
+ *    `setScenario({ resourceFileError: undefined })` clears a forced resource-file error.
  *  - `setTheme('light' | 'dark')` — flip the site theme and push `THEME_CHANGE`.
  *  - `buzz.setBalance(n)` / `buzz.getBalance()` — flip the simulated wallet.
  *