@playdrop/playdrop-cli 0.16.12 → 0.17.0

package/dist/commands/worker/generation.d.ts

@@ -0,0 +1,94 @@
+import type { ApiClient } from "@playdrop/api-client";
+import type { CreateAiGenerationTaskRequest, WorkerGenerationTaskAssignmentV5, WorkerGenerationUploadManifestFile, WorkerScreeningTaskAssignmentV5 } from "@playdrop/types";
+type GeneratedFile = {
+    role: WorkerGenerationUploadManifestFile["role"];
+    fileName: string;
+    contentType: string;
+    filePath: string;
+};
+type FalFile = {
+    url: string;
+    contentType: string | null;
+};
+export declare function emitGenerationWorkerEvent(event: string, fields: Record<string, string | number | boolean | null>, write?: (line: string) => void): void;
+export declare function persistFalRequestId(input: {
+    client: Pick<ApiClient, "workerRecordGenerationProviderRequest">;
+    taskId: number;
+    workerKey: string;
+    leaseToken: string;
+    endpoint: string;
+    requestId: string;
+    signal: AbortSignal;
+    emit?: typeof emitGenerationWorkerEvent;
+}): Promise<void>;
+export declare function generationFailureDisposition(error: unknown): "cancel" | "requeue" | "withhold" | "fail";
+export declare function falSubmissionFailure(endpoint: string, error: unknown): Error;
+export declare function createGenerationLeaseMonitor(input: {
+    heartbeat: () => Promise<{
+        action: "continue" | "abort";
+    }>;
+    intervalMs?: number;
+    failureLimit?: number;
+}): {
+    signal: AbortSignal;
+    ready: Promise<void>;
+    stop: () => Promise<void>;
+};
+export declare function falCredentialPath(workerHomeDir: string): string;
+export declare function hasFalCredential(workerHomeDir: string): boolean;
+export declare function storeFalCredential(workerHomeDir: string, value: string): Promise<void>;
+export declare const GENERATION_WORKER_FAILPOINTS: {
+    readonly falRequestPersisted: {
+        readonly env: "PLAYDROP_TEST_WORKER_EXIT_AFTER_FAL_REQUEST_PERSIST";
+        readonly marker: "playdrop.test_failpoint.fal_request_persisted";
+    };
+    readonly uploadBeforeFinalize: {
+        readonly env: "PLAYDROP_TEST_WORKER_EXIT_AFTER_GENERATION_UPLOAD";
+        readonly marker: "playdrop.test_failpoint.generation_upload_complete";
+    };
+};
+export declare function triggerGenerationWorkerFailpoint(input: {
+    failpoint: keyof typeof GENERATION_WORKER_FAILPOINTS;
+    workerHomeDir: string;
+    taskId: number;
+    exit?: (code: number) => never | void;
+}): Promise<boolean>;
+export declare function configureFalCredential(workerHomeDir: string): Promise<void>;
+export declare function buildSandboxedCodexArgs(input: {
+    sandbox: "read-only" | "workspace-write";
+    outputPath: string;
+    prompt: string;
+    outputSchemaPath?: string;
+}): string[];
+export declare function falInput(request: CreateAiGenerationTaskRequest, resolvedInputs: Map<string, string>): Record<string, unknown>;
+export declare function selectFalOutputs(kind: CreateAiGenerationTaskRequest["kind"], value: Record<string, unknown>): {
+    primary: FalFile;
+    thumbnail: FalFile | null;
+};
+export declare function validatePrimary(kind: string, filePath: string, signal?: AbortSignal): Promise<Record<string, unknown>>;
+export declare function embedMetadata(kind: string, primaryPath: string, metadata: Record<string, unknown>, signal?: AbortSignal): Promise<void>;
+export declare function uploadFiles(input: {
+    client: ApiClient;
+    assignment: WorkerGenerationTaskAssignmentV5;
+    workerKey: string;
+    leaseToken: string;
+    files: GeneratedFile[];
+    metadata: Record<string, unknown>;
+    providerOutput?: Record<string, unknown>;
+    signal: AbortSignal;
+    workerHomeDir: string;
+}): Promise<void>;
+export declare function runWorkerGenerationTask(input: {
+    client: ApiClient;
+    assignment: WorkerGenerationTaskAssignmentV5;
+    workerKey: string;
+    leaseToken: string;
+    workerHomeDir: string;
+}): Promise<void>;
+export declare function runWorkerScreeningTask(input: {
+    client: ApiClient;
+    assignment: WorkerScreeningTaskAssignmentV5;
+    workerKey: string;
+    leaseToken: string;
+}): Promise<void>;
+export {};