@ai-sdk/prodia 0.0.4 → 0.0.6

package/dist/index.mjs

@@ -11,14 +11,162 @@ import {
 // src/prodia-image-model.ts
 import {
   combineHeaders,
-  createJsonErrorResponseHandler,
   lazySchema,
+  parseJSON,
   parseProviderOptions,
   postToApi,
   resolve,
   zodSchema
 } from "@ai-sdk/provider-utils";
+import { z as z2 } from "zod/v4";
+
+// src/prodia-api.ts
+import { createJsonErrorResponseHandler } from "@ai-sdk/provider-utils";
 import { z } from "zod/v4";
+var 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()
+});
+function buildProdiaProviderMetadata(jobResult) {
+  var _a, _b, _c, _d;
+  return {
+    jobId: jobResult.id,
+    ...((_a = jobResult.config) == null ? void 0 : _a.seed) != null && {
+      seed: jobResult.config.seed
+    },
+    ...((_b = jobResult.metrics) == null ? void 0 : _b.elapsed) != null && {
+      elapsed: jobResult.metrics.elapsed
+    },
+    ...((_c = jobResult.metrics) == null ? void 0 : _c.ips) != null && {
+      iterationsPerSecond: jobResult.metrics.ips
+    },
+    ...jobResult.created_at != null && {
+      createdAt: jobResult.created_at
+    },
+    ...jobResult.updated_at != null && {
+      updatedAt: jobResult.updated_at
+    },
+    ...((_d = jobResult.price) == null ? void 0 : _d.dollars) != null && {
+      dollars: jobResult.price.dollars
+    }
+  };
+}
+function parseMultipart(data, boundary) {
+  const parts = [];
+  const boundaryBytes = new TextEncoder().encode(`--${boundary}`);
+  const endBoundaryBytes = new TextEncoder().encode(`--${boundary}--`);
+  const positions = [];
+  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] === 13 && data[partStart + 1] === 10) {
+      partStart += 2;
+    } else if (data[partStart] === 10) {
+      partStart += 1;
+    }
+    let partEnd = end;
+    if (data[partEnd - 2] === 13 && data[partEnd - 1] === 10) {
+      partEnd -= 2;
+    } else if (data[partEnd - 1] === 10) {
+      partEnd -= 1;
+    }
+    const partData = data.slice(partStart, partEnd);
+    let headerEnd = -1;
+    for (let j = 0; j < partData.length - 3; j++) {
+      if (partData[j] === 13 && partData[j + 1] === 10 && partData[j + 2] === 13 && partData[j + 3] === 10) {
+        headerEnd = j;
+        break;
+      }
+      if (partData[j] === 10 && partData[j + 1] === 10) {
+        headerEnd = j;
+        break;
+      }
+    }
+    if (headerEnd === -1) {
+      continue;
+    }
+    const headerBytes = partData.slice(0, headerEnd);
+    const headerStr = new TextDecoder().decode(headerBytes);
+    const headers = {};
+    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] === 13) {
+      bodyStart = headerEnd + 4;
+    }
+    const body = partData.slice(bodyStart);
+    parts.push({ headers, body });
+  }
+  return parts;
+}
+var prodiaErrorSchema = z.object({
+  message: z.string().optional(),
+  detail: z.unknown().optional(),
+  error: z.string().optional()
+});
+var prodiaFailedResponseHandler = createJsonErrorResponseHandler({
+  errorSchema: prodiaErrorSchema,
+  errorToMessage: (error) => {
+    var _a;
+    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 (e) {
+      }
+    }
+    return (_a = errorField != null ? errorField : message) != null ? _a : "Unknown Prodia error";
+  }
+});
+
+// src/prodia-image-model.ts
 var ProdiaImageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -92,7 +240,7 @@ var ProdiaImageModel = class {
     return { body, warnings };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g;
+    var _a, _b, _c;
     const { body, warnings } = await this.getArgs(options);
     const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
     const combinedHeaders = combineHeaders(
@@ -121,29 +269,7 @@ var ProdiaImageModel = class {
       warnings,
       providerMetadata: {
         prodia: {
-          images: [
-            {
-              jobId: jobResult.id,
-              ...((_d = jobResult.config) == null ? void 0 : _d.seed) != null && {
-                seed: jobResult.config.seed
-              },
-              ...((_e = jobResult.metrics) == null ? void 0 : _e.elapsed) != null && {
-                elapsed: jobResult.metrics.elapsed
-              },
-              ...((_f = jobResult.metrics) == null ? void 0 : _f.ips) != null && {
-                iterationsPerSecond: jobResult.metrics.ips
-              },
-              ...jobResult.created_at != null && {
-                createdAt: jobResult.created_at
-              },
-              ...jobResult.updated_at != null && {
-                updatedAt: jobResult.updated_at
-              },
-              ...((_g = jobResult.price) == null ? void 0 : _g.dollars) != null && {
-                dollars: jobResult.price.dollars
-              }
-            }
-          ]
+          images: [buildProdiaProviderMetadata(jobResult)]
         }
       },
       response: {
@@ -175,54 +301,34 @@ var stylePresets = [
 ];
 var prodiaImageProviderOptionsSchema = lazySchema(
   () => zodSchema(
-    z.object({
+    z2.object({
       /**
        * Amount of computational iterations to run. More is typically higher quality.
        */
-      steps: z.number().int().min(1).max(4).optional(),
+      steps: z2.number().int().min(1).max(4).optional(),
       /**
        * Width of the output image in pixels.
        */
-      width: z.number().int().min(256).max(1920).optional(),
+      width: z2.number().int().min(256).max(1920).optional(),
       /**
        * Height of the output image in pixels.
        */
-      height: z.number().int().min(256).max(1920).optional(),
+      height: z2.number().int().min(256).max(1920).optional(),
       /**
        * Apply a visual theme to your output image.
        */
-      stylePreset: z.enum(stylePresets).optional(),
+      stylePreset: z2.enum(stylePresets).optional(),
       /**
        * Augment the output with a LoRa model.
        */
-      loras: z.array(z.string()).max(3).optional(),
+      loras: z2.array(z2.string()).max(3).optional(),
       /**
        * When using JPEG output, return a progressive JPEG.
        */
-      progressive: z.boolean().optional()
+      progressive: z2.boolean().optional()
     })
   )
 );
-var 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()
-});
 function createMultipartResponseHandler() {
   return async ({
     response
@@ -250,7 +356,10 @@ function createMultipartResponseHandler() {
       const partContentType = (_c = part.headers["content-type"]) != null ? _c : "";
       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/")) {
@@ -269,107 +378,321 @@ function createMultipartResponseHandler() {
     };
   };
 }
-function parseMultipart(data, boundary) {
-  const parts = [];
-  const boundaryBytes = new TextEncoder().encode(`--${boundary}`);
-  const endBoundaryBytes = new TextEncoder().encode(`--${boundary}--`);
-  const positions = [];
-  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;
-      }
+
+// src/prodia-language-model.ts
+import {
+  combineHeaders as combineHeaders2,
+  convertBase64ToUint8Array,
+  generateId,
+  lazySchema as lazySchema2,
+  parseJSON as parseJSON2,
+  parseProviderOptions as parseProviderOptions2,
+  postFormDataToApi,
+  resolve as resolve2,
+  zodSchema as zodSchema2
+} from "@ai-sdk/provider-utils";
+import { z as z3 } from "zod/v4";
+var ProdiaLanguageModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v2";
+    this.supportedUrls = {};
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate(options) {
+    var _a, _b, _c, _d;
+    const warnings = [];
+    if (options.temperature !== void 0) {
+      warnings.push({ type: "unsupported-setting", setting: "temperature" });
     }
-    if (match) {
-      positions.push(i);
+    if (options.topP !== void 0) {
+      warnings.push({ type: "unsupported-setting", setting: "topP" });
     }
-  }
-  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 (options.topK !== void 0) {
+      warnings.push({ type: "unsupported-setting", setting: "topK" });
     }
-    if (isEndBoundary && positions[i] + endBoundaryBytes.length <= data.length) {
-      continue;
+    if (options.maxOutputTokens !== void 0) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "maxOutputTokens"
+      });
     }
-    let partStart = start;
-    if (data[partStart] === 13 && data[partStart + 1] === 10) {
-      partStart += 2;
-    } else if (data[partStart] === 10) {
-      partStart += 1;
+    if (options.stopSequences !== void 0) {
+      warnings.push({ type: "unsupported-setting", setting: "stopSequences" });
     }
-    let partEnd = end;
-    if (data[partEnd - 2] === 13 && data[partEnd - 1] === 10) {
-      partEnd -= 2;
-    } else if (data[partEnd - 1] === 10) {
-      partEnd -= 1;
+    if (options.presencePenalty !== void 0) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "presencePenalty"
+      });
     }
-    const partData = data.slice(partStart, partEnd);
-    let headerEnd = -1;
-    for (let j = 0; j < partData.length - 3; j++) {
-      if (partData[j] === 13 && partData[j + 1] === 10 && partData[j + 2] === 13 && partData[j + 3] === 10) {
-        headerEnd = j;
+    if (options.frequencyPenalty !== void 0) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "frequencyPenalty"
+      });
+    }
+    if (options.tools !== void 0 && options.tools.length > 0) {
+      warnings.push({ type: "unsupported-setting", setting: "tools" });
+    }
+    if (options.toolChoice !== void 0) {
+      warnings.push({ type: "unsupported-setting", setting: "toolChoice" });
+    }
+    if (options.responseFormat !== void 0 && options.responseFormat.type !== "text") {
+      warnings.push({ type: "unsupported-setting", setting: "responseFormat" });
+    }
+    const prodiaOptions = await parseProviderOptions2({
+      provider: "prodia",
+      providerOptions: options.providerOptions,
+      schema: prodiaLanguageModelOptionsSchema
+    });
+    let prompt = "";
+    let systemMessage = "";
+    for (const message of options.prompt) {
+      if (message.role === "system") {
+        systemMessage = message.content;
+      }
+    }
+    for (let i = options.prompt.length - 1; i >= 0; i--) {
+      const message = options.prompt[i];
+      if (message.role === "user") {
+        for (const part of message.content) {
+          if (part.type === "text") {
+            prompt += (prompt ? "\n" : "") + part.text;
+          }
+        }
         break;
       }
-      if (partData[j] === 10 && partData[j + 1] === 10) {
-        headerEnd = j;
+    }
+    if (systemMessage) {
+      prompt = systemMessage + "\n" + prompt;
+    }
+    let imageBytes;
+    let imageMediaType = "image/png";
+    for (let i = options.prompt.length - 1; i >= 0; i--) {
+      const message = options.prompt[i];
+      if (message.role === "user") {
+        for (const part of message.content) {
+          if (part.type === "file" && part.mediaType.startsWith("image/")) {
+            if (part.data instanceof Uint8Array) {
+              imageBytes = part.data;
+            } else if (typeof part.data === "string") {
+              imageBytes = convertBase64ToUint8Array(part.data);
+            } else if (part.data instanceof URL) {
+              const fetchFn = (_a = this.config.fetch) != null ? _a : globalThis.fetch;
+              const response = await fetchFn(part.data.toString());
+              const arrayBuffer = await response.arrayBuffer();
+              imageBytes = new Uint8Array(arrayBuffer);
+            }
+            imageMediaType = part.mediaType;
+            break;
+          }
+        }
         break;
       }
     }
-    if (headerEnd === -1) {
-      continue;
+    const jobConfig = {
+      prompt,
+      include_messages: true
+    };
+    if ((prodiaOptions == null ? void 0 : prodiaOptions.aspectRatio) !== void 0) {
+      jobConfig.aspect_ratio = prodiaOptions.aspectRatio;
     }
-    const headerBytes = partData.slice(0, headerEnd);
-    const headerStr = new TextDecoder().decode(headerBytes);
-    const headers = {};
-    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;
+    const body = {
+      type: this.modelId,
+      config: jobConfig
+    };
+    const currentDate = (_d = (_c = (_b = this.config._internal) == null ? void 0 : _b.currentDate) == null ? void 0 : _c.call(_b)) != null ? _d : /* @__PURE__ */ new Date();
+    const combinedHeaders = combineHeaders2(
+      await resolve2(this.config.headers),
+      options.headers
+    );
+    const formData = new FormData();
+    formData.append(
+      "job",
+      new Blob([JSON.stringify(body)], { type: "application/json" }),
+      "job.json"
+    );
+    if (imageBytes) {
+      const ext = imageMediaType === "image/png" ? ".png" : imageMediaType === "image/jpeg" ? ".jpg" : imageMediaType === "image/webp" ? ".webp" : "";
+      formData.append(
+        "input",
+        new Blob([imageBytes], { type: imageMediaType }),
+        "input" + ext
+      );
+    }
+    const { value: multipartResult, responseHeaders } = await postFormDataToApi(
+      {
+        url: `${this.config.baseURL}/job?price=true`,
+        headers: {
+          ...combinedHeaders,
+          Accept: "multipart/form-data"
+        },
+        formData,
+        failedResponseHandler: prodiaFailedResponseHandler,
+        successfulResponseHandler: createLanguageMultipartResponseHandler(),
+        abortSignal: options.abortSignal,
+        fetch: this.config.fetch
       }
+    );
+    const { jobResult, textContent, fileContent } = multipartResult;
+    const content = [];
+    if (textContent !== void 0) {
+      content.push({ type: "text", text: textContent });
     }
-    let bodyStart = headerEnd + 2;
-    if (partData[headerEnd] === 13) {
-      bodyStart = headerEnd + 4;
+    for (const file of fileContent) {
+      content.push({
+        type: "file",
+        mediaType: file.mediaType,
+        data: file.data
+      });
     }
-    const body = partData.slice(bodyStart);
-    parts.push({ headers, body });
+    return {
+      content,
+      finishReason: "stop",
+      usage: {
+        inputTokens: void 0,
+        outputTokens: void 0,
+        totalTokens: void 0
+      },
+      warnings,
+      providerMetadata: {
+        prodia: buildProdiaProviderMetadata(jobResult)
+      },
+      response: {
+        modelId: this.modelId,
+        timestamp: currentDate,
+        headers: responseHeaders
+      }
+    };
   }
-  return parts;
-}
-var prodiaErrorSchema = z.object({
-  message: z.string().optional(),
-  detail: z.unknown().optional(),
-  error: z.string().optional()
-});
-var prodiaFailedResponseHandler = createJsonErrorResponseHandler({
-  errorSchema: prodiaErrorSchema,
-  errorToMessage: (error) => {
+  async doStream(options) {
     var _a;
-    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 (e) {
+    const result = await this.doGenerate(options);
+    const stream = new ReadableStream({
+      start(controller) {
+        var _a2, _b;
+        controller.enqueue({
+          type: "stream-start",
+          warnings: result.warnings
+        });
+        controller.enqueue({
+          type: "response-metadata",
+          modelId: (_a2 = result.response) == null ? void 0 : _a2.modelId,
+          timestamp: (_b = result.response) == null ? void 0 : _b.timestamp
+        });
+        for (const part of result.content) {
+          if (part.type === "text") {
+            const id = generateId();
+            controller.enqueue({ type: "text-start", id });
+            controller.enqueue({
+              type: "text-delta",
+              id,
+              delta: part.text
+            });
+            controller.enqueue({ type: "text-end", id });
+          } else if (part.type === "file") {
+            controller.enqueue({
+              type: "file",
+              mediaType: part.mediaType,
+              data: part.data
+            });
+          }
+        }
+        controller.enqueue({
+          type: "finish",
+          usage: result.usage,
+          finishReason: result.finishReason,
+          providerMetadata: result.providerMetadata
+        });
+        controller.close();
       }
-    }
-    return (_a = errorField != null ? errorField : message) != null ? _a : "Unknown Prodia error";
+    });
+    return {
+      stream,
+      response: {
+        headers: (_a = result.response) == null ? void 0 : _a.headers
+      }
+    };
   }
-});
+};
+var prodiaLanguageModelOptionsSchema = lazySchema2(
+  () => zodSchema2(
+    z3.object({
+      aspectRatio: z3.enum([
+        "1:1",
+        "2:3",
+        "3:2",
+        "4:5",
+        "5:4",
+        "4:7",
+        "7:4",
+        "9:16",
+        "16:9",
+        "9:21",
+        "21:9"
+      ]).optional()
+    })
+  )
+);
+function createLanguageMultipartResponseHandler() {
+  return async ({
+    response
+  }) => {
+    var _a, _b, _c;
+    const contentType = (_a = response.headers.get("content-type")) != null ? _a : "";
+    const responseHeaders = {};
+    response.headers.forEach((value, key) => {
+      responseHeaders[key] = value;
+    });
+    const boundaryMatch = contentType.match(/boundary=([^\s;]+)/);
+    if (!boundaryMatch) {
+      throw new Error(
+        `Prodia response missing multipart boundary in content-type: ${contentType}`
+      );
+    }
+    const boundary = boundaryMatch[1];
+    const arrayBuffer = await response.arrayBuffer();
+    const bytes = new Uint8Array(arrayBuffer);
+    const parts = parseMultipart(bytes, boundary);
+    let jobResult;
+    let textContent;
+    const fileContent = [];
+    for (const part of parts) {
+      const contentDisposition = (_b = part.headers["content-disposition"]) != null ? _b : "";
+      const partContentType = (_c = part.headers["content-type"]) != null ? _c : "";
+      if (contentDisposition.includes('name="job"')) {
+        const jsonStr = new TextDecoder().decode(part.body);
+        jobResult = await parseJSON2({
+          text: jsonStr,
+          schema: zodSchema2(prodiaJobResultSchema)
+        });
+      } else if (contentDisposition.includes('name="output"')) {
+        if (partContentType.startsWith("text/") || contentDisposition.includes(".txt")) {
+          textContent = new TextDecoder().decode(part.body);
+        } else if (partContentType.startsWith("image/")) {
+          fileContent.push({
+            mediaType: partContentType,
+            data: part.body
+          });
+        }
+      }
+    }
+    if (!jobResult) {
+      throw new Error("Prodia multipart response missing job part");
+    }
+    return {
+      value: { jobResult, textContent, fileContent },
+      responseHeaders
+    };
+  };
+}
 
 // src/version.ts
-var VERSION = true ? "0.0.4" : "0.0.0-test";
+var VERSION = true ? "0.0.6" : "0.0.0-test";
 
 // src/prodia-provider.ts
 var defaultBaseURL = "https://inference.prodia.com/v2";
@@ -393,22 +716,22 @@ function createProdia(options = {}) {
     headers: getHeaders,
     fetch: options.fetch
   });
+  const createLanguageModel = (modelId) => new ProdiaLanguageModel(modelId, {
+    provider: "prodia.language",
+    baseURL: baseURL != null ? baseURL : defaultBaseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const textEmbeddingModel = (modelId) => {
     throw new NoSuchModelError({
       modelId,
       modelType: "textEmbeddingModel"
     });
   };
-  const languageModel = (modelId) => {
-    throw new NoSuchModelError({
-      modelId,
-      modelType: "languageModel"
-    });
-  };
   return {
+    languageModel: createLanguageModel,
     imageModel: createImageModel,
     image: createImageModel,
-    languageModel,
     textEmbeddingModel
   };
 }