@ai-sdk/prodia 2.0.0-beta.4 → 2.0.0-beta.53

package/src/prodia-image-model.ts

@@ -1,16 +1,24 @@
 import type { ImageModelV4, SharedV4Warning } from '@ai-sdk/provider';
-import type { InferSchema, Resolvable } from '@ai-sdk/provider-utils';
 import {
   combineHeaders,
-  createJsonErrorResponseHandler,
-  type FetchFunction,
-  lazySchema,
+  parseJSON,
   parseProviderOptions,
   postToApi,
   resolve,
+  serializeModelOptions,
+  WORKFLOW_SERIALIZE,
+  WORKFLOW_DESERIALIZE,
   zodSchema,
 } from '@ai-sdk/provider-utils';
-import { z } from 'zod/v4';
+import {
+  buildProdiaProviderMetadata,
+  parseMultipart,
+  prodiaFailedResponseHandler,
+  prodiaJobResultSchema,
+  type ProdiaJobResult,
+  type ProdiaModelConfig,
+} from './prodia-api';
+import { prodiaImageModelOptionsSchema } from './prodia-image-model-options';
 import type { ProdiaImageModelId } from './prodia-image-settings';
 
 export class ProdiaImageModel implements ImageModelV4 {
@@ -21,9 +29,23 @@ export class ProdiaImageModel implements ImageModelV4 {
     return this.config.provider;
   }
 
+  static [WORKFLOW_SERIALIZE](model: ProdiaImageModel) {
+    return serializeModelOptions({
+      modelId: model.modelId,
+      config: model.config,
+    });
+  }
+
+  static [WORKFLOW_DESERIALIZE](options: {
+    modelId: ProdiaImageModelId;
+    config: ProdiaModelConfig;
+  }) {
+    return new ProdiaImageModel(options.modelId, options.config);
+  }
+
   constructor(
     readonly modelId: ProdiaImageModelId,
-    private readonly config: ProdiaImageModelConfig,
+    private readonly config: ProdiaModelConfig,
   ) {}
 
   private async getArgs({
@@ -109,7 +131,7 @@ export class ProdiaImageModel implements ImageModelV4 {
 
     const currentDate = this.config._internal?.currentDate?.() ?? new Date();
     const combinedHeaders = combineHeaders(
-      await resolve(this.config.headers),
+      this.config.headers ? await resolve(this.config.headers) : undefined,
       options.headers,
     );
 
@@ -137,29 +159,7 @@ export class ProdiaImageModel implements ImageModelV4 {
       warnings,
       providerMetadata: {
         prodia: {
-          images: [
-            {
-              jobId: jobResult.id,
-              ...(jobResult.config?.seed != null && {
-                seed: jobResult.config.seed,
-              }),
-              ...(jobResult.metrics?.elapsed != null && {
-                elapsed: jobResult.metrics.elapsed,
-              }),
-              ...(jobResult.metrics?.ips != null && {
-                iterationsPerSecond: jobResult.metrics.ips,
-              }),
-              ...(jobResult.created_at != null && {
-                createdAt: jobResult.created_at,
-              }),
-              ...(jobResult.updated_at != null && {
-                updatedAt: jobResult.updated_at,
-              }),
-              ...(jobResult.price?.dollars != null && {
-                dollars: jobResult.price.dollars,
-              }),
-            },
-          ],
+          images: [buildProdiaProviderMetadata(jobResult)],
         },
       },
       response: {
@@ -171,103 +171,6 @@ export class ProdiaImageModel implements ImageModelV4 {
   }
 }
 
-const stylePresets = [
-  '3d-model',
-  'analog-film',
-  'anime',
-  'cinematic',
-  'comic-book',
-  'digital-art',
-  'enhance',
-  'fantasy-art',
-  'isometric',
-  'line-art',
-  'low-poly',
-  'neon-punk',
-  'origami',
-  'photographic',
-  'pixel-art',
-  'texture',
-  'craft-clay',
-] as const;
-
-export const prodiaImageModelOptionsSchema = lazySchema(() =>
-  zodSchema(
-    z.object({
-      /**
-       * Amount of computational iterations to run. More is typically higher quality.
-       */
-      steps: z.number().int().min(1).max(4).optional(),
-      /**
-       * Width of the output image in pixels.
-       */
-      width: z.number().int().min(256).max(1920).optional(),
-      /**
-       * Height of the output image in pixels.
-       */
-      height: z.number().int().min(256).max(1920).optional(),
-      /**
-       * Apply a visual theme to your output image.
-       */
-      stylePreset: z.enum(stylePresets).optional(),
-      /**
-       * Augment the output with a LoRa model.
-       */
-      loras: z.array(z.string()).max(3).optional(),
-      /**
-       * When using JPEG output, return a progressive JPEG.
-       */
-      progressive: z.boolean().optional(),
-    }),
-  ),
-);
-
-export type ProdiaImageModelOptions = InferSchema<
-  typeof prodiaImageModelOptionsSchema
->;
-
-interface ProdiaImageModelConfig {
-  provider: string;
-  baseURL: string;
-  headers?: Resolvable<Record<string, string | undefined>>;
-  fetch?: FetchFunction;
-  _internal?: {
-    currentDate?: () => Date;
-  };
-}
-
-const prodiaJobResultSchema = z.object({
-  id: z.string(),
-  created_at: z.string().optional(),
-  updated_at: z.string().optional(),
-  expires_at: z.string().optional(),
-  state: z
-    .object({
-      current: z.string(),
-    })
-    .optional(),
-  config: z
-    .object({
-      seed: z.number().optional(),
-    })
-    .passthrough()
-    .optional(),
-  metrics: z
-    .object({
-      elapsed: z.number().optional(),
-      ips: z.number().optional(),
-    })
-    .optional(),
-  price: z
-    .object({
-      product: z.string(),
-      dollars: z.number(),
-    })
-    .nullish(),
-});
-
-type ProdiaJobResult = z.infer<typeof prodiaJobResultSchema>;
-
 interface MultipartResult {
   jobResult: ProdiaJobResult;
   imageBytes: Uint8Array;
@@ -310,7 +213,10 @@ function createMultipartResponseHandler() {
 
       if (contentDisposition.includes('name="job"')) {
         const jsonStr = new TextDecoder().decode(part.body);
-        jobResult = prodiaJobResultSchema.parse(JSON.parse(jsonStr));
+        jobResult = await parseJSON({
+          text: jsonStr,
+          schema: zodSchema(prodiaJobResultSchema),
+        });
       } else if (contentDisposition.includes('name="output"')) {
         imageBytes = part.body;
       } else if (partContentType.startsWith('image/')) {
@@ -331,129 +237,3 @@ function createMultipartResponseHandler() {
     };
   };
 }
-
-interface MultipartPart {
-  headers: Record<string, string>;
-  body: Uint8Array;
-}
-
-function parseMultipart(data: Uint8Array, boundary: string): MultipartPart[] {
-  const parts: MultipartPart[] = [];
-  const boundaryBytes = new TextEncoder().encode(`--${boundary}`);
-  const endBoundaryBytes = new TextEncoder().encode(`--${boundary}--`);
-
-  const positions: number[] = [];
-  for (let i = 0; i <= data.length - boundaryBytes.length; i++) {
-    let match = true;
-    for (let j = 0; j < boundaryBytes.length; j++) {
-      if (data[i + j] !== boundaryBytes[j]) {
-        match = false;
-        break;
-      }
-    }
-    if (match) {
-      positions.push(i);
-    }
-  }
-
-  for (let i = 0; i < positions.length - 1; i++) {
-    const start = positions[i] + boundaryBytes.length;
-    const end = positions[i + 1];
-
-    let isEndBoundary = true;
-    for (let j = 0; j < endBoundaryBytes.length && isEndBoundary; j++) {
-      if (data[positions[i] + j] !== endBoundaryBytes[j]) {
-        isEndBoundary = false;
-      }
-    }
-    if (
-      isEndBoundary &&
-      positions[i] + endBoundaryBytes.length <= data.length
-    ) {
-      continue;
-    }
-
-    let partStart = start;
-    if (data[partStart] === 0x0d && data[partStart + 1] === 0x0a) {
-      partStart += 2;
-    } else if (data[partStart] === 0x0a) {
-      partStart += 1;
-    }
-
-    let partEnd = end;
-    if (data[partEnd - 2] === 0x0d && data[partEnd - 1] === 0x0a) {
-      partEnd -= 2;
-    } else if (data[partEnd - 1] === 0x0a) {
-      partEnd -= 1;
-    }
-
-    const partData = data.slice(partStart, partEnd);
-
-    let headerEnd = -1;
-    for (let j = 0; j < partData.length - 3; j++) {
-      if (
-        partData[j] === 0x0d &&
-        partData[j + 1] === 0x0a &&
-        partData[j + 2] === 0x0d &&
-        partData[j + 3] === 0x0a
-      ) {
-        headerEnd = j;
-        break;
-      }
-      if (partData[j] === 0x0a && partData[j + 1] === 0x0a) {
-        headerEnd = j;
-        break;
-      }
-    }
-
-    if (headerEnd === -1) {
-      continue;
-    }
-
-    const headerBytes = partData.slice(0, headerEnd);
-    const headerStr = new TextDecoder().decode(headerBytes);
-    const headers: Record<string, string> = {};
-    for (const line of headerStr.split(/\r?\n/)) {
-      const colonIdx = line.indexOf(':');
-      if (colonIdx > 0) {
-        const key = line.slice(0, colonIdx).trim().toLowerCase();
-        const value = line.slice(colonIdx + 1).trim();
-        headers[key] = value;
-      }
-    }
-
-    let bodyStart = headerEnd + 2;
-    if (partData[headerEnd] === 0x0d) {
-      bodyStart = headerEnd + 4;
-    }
-    const body = partData.slice(bodyStart);
-
-    parts.push({ headers, body });
-  }
-
-  return parts;
-}
-
-const prodiaErrorSchema = z.object({
-  message: z.string().optional(),
-  detail: z.unknown().optional(),
-  error: z.string().optional(),
-});
-
-const prodiaFailedResponseHandler = createJsonErrorResponseHandler({
-  errorSchema: prodiaErrorSchema,
-  errorToMessage: error => {
-    const parsed = prodiaErrorSchema.safeParse(error);
-    if (!parsed.success) return 'Unknown Prodia error';
-    const { message, detail, error: errorField } = parsed.data;
-    if (typeof detail === 'string') return detail;
-    if (detail != null) {
-      try {
-        return JSON.stringify(detail);
-      } catch {
-        // ignore
-      }
-    }
-    return errorField ?? message ?? 'Unknown Prodia error';
-  },
-});

package/src/prodia-language-model-options.ts

@@ -0,0 +1,35 @@
+import {
+  lazySchema,
+  zodSchema,
+  type InferSchema,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+export const prodiaLanguageModelOptionsSchema = lazySchema(() =>
+  zodSchema(
+    z.object({
+      /**
+       * Aspect ratio for the output image.
+       */
+      aspectRatio: z
+        .enum([
+          '1:1',
+          '2:3',
+          '3:2',
+          '4:5',
+          '5:4',
+          '4:7',
+          '7:4',
+          '9:16',
+          '16:9',
+          '9:21',
+          '21:9',
+        ])
+        .optional(),
+    }),
+  ),
+);
+
+export type ProdiaLanguageModelOptions = InferSchema<
+  typeof prodiaLanguageModelOptionsSchema
+>;

package/src/prodia-language-model-settings.ts

@@ -0,0 +1,6 @@
+/**
+ * Prodia job types for language model (multimodal img2img).
+ */
+export type ProdiaLanguageModelId =
+  | 'inference.nano-banana.img2img.v2'
+  | (string & {});