@ai-sdk/luma 0.0.1

package/CHANGELOG.md

@@ -0,0 +1,9 @@
+# @ai-sdk/luma
+
+## 0.0.1
+
+### Patch Changes
+
+- 39e5c1f: feat (provider/luma): add Luma provider
+- Updated dependencies [39e5c1f]
+  - @ai-sdk/provider-utils@2.1.3

package/LICENSE

@@ -0,0 +1,13 @@
+Copyright 2023 Vercel, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

package/README.md

@@ -0,0 +1,50 @@
+# AI SDK - Luma Provider
+
+The **Luma provider** for the [AI SDK](https://sdk.vercel.ai/docs) contains support for Luma AI's state-of-the-art image generation models - Photon and Photon Flash.
+
+## About Luma Photon Models
+
+Luma Photon and Photon Flash are groundbreaking image generation models that deliver:
+
+- Ultra-high quality image generation
+- 10x higher cost efficiency compared to similar models
+- Superior prompt understanding and adherence
+- Unique character consistency capabilities from single reference images
+- Multi-image reference support for precise style matching
+
+## Setup
+
+The Luma provider is available in the `@ai-sdk/luma` module. You can install it with:
+
+```bash
+npm i @ai-sdk/luma
+```
+
+## Provider Instance
+
+You can import the default provider instance `luma` from `@ai-sdk/luma`:
+
+```ts
+import { luma } from '@ai-sdk/luma';
+```
+
+## Image Generation Example
+
+```ts
+import { luma } from '@ai-sdk/luma';
+import { experimental_generateImage as generateImage } from 'ai';
+import fs from 'fs';
+
+const { image } = await generateImage({
+  model: luma.image('photon'),
+  prompt: 'A serene mountain landscape at sunset',
+});
+
+const filename = `image-${Date.now()}.png`;
+fs.writeFileSync(filename, image.uint8Array);
+console.log(`Image saved to ${filename}`);
+```
+
+## Documentation
+
+For more detailed information about the Luma models and their capabilities, please visit [Luma AI](https://lumalabs.ai/).

package/dist/index.d.mts

@@ -0,0 +1,114 @@
+import { ImageModelV1 } from '@ai-sdk/provider';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { z } from 'zod';
+
+type LumaImageModelId = 'photon-1' | 'photon-flash-1' | (string & {});
+/**
+Configuration settings for Luma image generation.
+
+Since the Luma API processes images through an asynchronous queue system, these
+settings allow you to tune the polling behavior when waiting for image
+generation to complete.
+ */
+interface LumaImageSettings {
+    /**
+  Override the maximum number of images per call (default 1)
+     */
+    maxImagesPerCall?: number;
+    /**
+  Override the polling interval in milliseconds (default 500). This controls how
+  frequently the API is checked for completed images while they are being
+  processed in Luma's queue.
+     */
+    pollIntervalMillis?: number;
+    /**
+  Override the maximum number of polling attempts (default 120). Since image
+  generation is queued and processed asynchronously, this limits how long to wait
+  for results before timing out.
+     */
+    maxPollAttempts?: number;
+}
+
+interface LumaProviderSettings {
+    /**
+  Luma API key. Default value is taken from the `LUMA_API_KEY` environment
+  variable.
+    */
+    apiKey?: string;
+    /**
+  Base URL for the API calls.
+    */
+    baseURL?: string;
+    /**
+  Custom headers to include in the requests.
+    */
+    headers?: Record<string, string>;
+    /**
+  Custom fetch implementation. You can use it as a middleware to intercept requests,
+  or to provide a custom fetch implementation for e.g. testing.
+    */
+    fetch?: FetchFunction;
+}
+interface LumaProvider {
+    /**
+  Creates a model for image generation.
+    */
+    image(modelId: LumaImageModelId, settings?: LumaImageSettings): ImageModelV1;
+}
+declare function createLuma(options?: LumaProviderSettings): LumaProvider;
+declare const luma: LumaProvider;
+
+declare const lumaErrorSchema: z.ZodObject<{
+    detail: z.ZodArray<z.ZodObject<{
+        type: z.ZodString;
+        loc: z.ZodArray<z.ZodString, "many">;
+        msg: z.ZodString;
+        input: z.ZodString;
+        ctx: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+            expected: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            expected: string;
+        }, {
+            expected: string;
+        }>>>;
+    }, "strip", z.ZodTypeAny, {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }, {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }>, "many">;
+}, "strip", z.ZodTypeAny, {
+    detail: {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }[];
+}, {
+    detail: {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }[];
+}>;
+type LumaErrorData = z.infer<typeof lumaErrorSchema>;
+
+export { type LumaErrorData, type LumaProvider, type LumaProviderSettings, createLuma, luma };

package/dist/index.d.ts

@@ -0,0 +1,114 @@
+import { ImageModelV1 } from '@ai-sdk/provider';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { z } from 'zod';
+
+type LumaImageModelId = 'photon-1' | 'photon-flash-1' | (string & {});
+/**
+Configuration settings for Luma image generation.
+
+Since the Luma API processes images through an asynchronous queue system, these
+settings allow you to tune the polling behavior when waiting for image
+generation to complete.
+ */
+interface LumaImageSettings {
+    /**
+  Override the maximum number of images per call (default 1)
+     */
+    maxImagesPerCall?: number;
+    /**
+  Override the polling interval in milliseconds (default 500). This controls how
+  frequently the API is checked for completed images while they are being
+  processed in Luma's queue.
+     */
+    pollIntervalMillis?: number;
+    /**
+  Override the maximum number of polling attempts (default 120). Since image
+  generation is queued and processed asynchronously, this limits how long to wait
+  for results before timing out.
+     */
+    maxPollAttempts?: number;
+}
+
+interface LumaProviderSettings {
+    /**
+  Luma API key. Default value is taken from the `LUMA_API_KEY` environment
+  variable.
+    */
+    apiKey?: string;
+    /**
+  Base URL for the API calls.
+    */
+    baseURL?: string;
+    /**
+  Custom headers to include in the requests.
+    */
+    headers?: Record<string, string>;
+    /**
+  Custom fetch implementation. You can use it as a middleware to intercept requests,
+  or to provide a custom fetch implementation for e.g. testing.
+    */
+    fetch?: FetchFunction;
+}
+interface LumaProvider {
+    /**
+  Creates a model for image generation.
+    */
+    image(modelId: LumaImageModelId, settings?: LumaImageSettings): ImageModelV1;
+}
+declare function createLuma(options?: LumaProviderSettings): LumaProvider;
+declare const luma: LumaProvider;
+
+declare const lumaErrorSchema: z.ZodObject<{
+    detail: z.ZodArray<z.ZodObject<{
+        type: z.ZodString;
+        loc: z.ZodArray<z.ZodString, "many">;
+        msg: z.ZodString;
+        input: z.ZodString;
+        ctx: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+            expected: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            expected: string;
+        }, {
+            expected: string;
+        }>>>;
+    }, "strip", z.ZodTypeAny, {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }, {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }>, "many">;
+}, "strip", z.ZodTypeAny, {
+    detail: {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }[];
+}, {
+    detail: {
+        type: string;
+        loc: string[];
+        msg: string;
+        input: string;
+        ctx?: {
+            expected: string;
+        } | null | undefined;
+    }[];
+}>;
+type LumaErrorData = z.infer<typeof lumaErrorSchema>;
+
+export { type LumaErrorData, type LumaProvider, type LumaProviderSettings, createLuma, luma };

package/dist/index.js

@@ -0,0 +1,230 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  createLuma: () => createLuma,
+  luma: () => luma
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/luma-provider.ts
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+
+// src/luma-image-model.ts
+var import_provider = require("@ai-sdk/provider");
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_zod = require("zod");
+var DEFAULT_POLL_INTERVAL_MILLIS = 500;
+var DEFAULT_MAX_POLL_ATTEMPTS = 6e4 / DEFAULT_POLL_INTERVAL_MILLIS;
+async function delay(delayInMs) {
+  return delayInMs == null ? Promise.resolve() : new Promise((resolve) => setTimeout(resolve, delayInMs));
+}
+var LumaImageModel = class {
+  constructor(modelId, settings, config) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+    this.specificationVersion = "v1";
+    var _a, _b;
+    this.pollIntervalMillis = (_a = settings.pollIntervalMillis) != null ? _a : DEFAULT_POLL_INTERVAL_MILLIS;
+    this.maxPollAttempts = (_b = settings.maxPollAttempts) != null ? _b : DEFAULT_MAX_POLL_ATTEMPTS;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  get maxImagesPerCall() {
+    var _a;
+    return (_a = this.settings.maxImagesPerCall) != null ? _a : 1;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    var _a, _b, _c, _d;
+    const warnings = [];
+    if (seed != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "seed",
+        details: "This model does not support the `seed` option."
+      });
+    }
+    if (size != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "size",
+        details: "This model does not support the `size` option. Use `aspectRatio` instead."
+      });
+    }
+    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 fullHeaders = (0, import_provider_utils.combineHeaders)(this.config.headers(), headers);
+    const { value: generationResponse, responseHeaders } = await (0, import_provider_utils.postJsonToApi)({
+      url: this.getLumaGenerationsUrl(),
+      headers: fullHeaders,
+      body: {
+        prompt,
+        ...aspectRatio ? { aspect_ratio: aspectRatio } : {},
+        model: this.modelId,
+        ...(_d = providerOptions.luma) != null ? _d : {}
+      },
+      abortSignal,
+      fetch: this.config.fetch,
+      failedResponseHandler: this.createLumaErrorHandler(),
+      successfulResponseHandler: (0, import_provider_utils.createJsonResponseHandler)(
+        lumaGenerationResponseSchema
+      )
+    });
+    const imageUrl = await this.pollForImageUrl(
+      generationResponse.id,
+      fullHeaders,
+      abortSignal
+    );
+    const downloadedImage = await this.downloadImage(imageUrl, abortSignal);
+    return {
+      images: [downloadedImage],
+      warnings,
+      response: {
+        modelId: this.modelId,
+        timestamp: currentDate,
+        headers: responseHeaders
+      }
+    };
+  }
+  async pollForImageUrl(generationId, headers, abortSignal) {
+    var _a;
+    let attemptCount = 0;
+    const url = this.getLumaGenerationsUrl(generationId);
+    for (let i = 0; i < this.maxPollAttempts; i++) {
+      const { value: statusResponse } = await (0, import_provider_utils.getFromApi)({
+        url,
+        headers,
+        abortSignal,
+        fetch: this.config.fetch,
+        failedResponseHandler: this.createLumaErrorHandler(),
+        successfulResponseHandler: (0, import_provider_utils.createJsonResponseHandler)(
+          lumaGenerationResponseSchema
+        )
+      });
+      switch (statusResponse.state) {
+        case "completed":
+          if (!((_a = statusResponse.assets) == null ? void 0 : _a.image)) {
+            throw new import_provider.InvalidResponseDataError({
+              data: statusResponse,
+              message: `Image generation completed but no image was found.`
+            });
+          }
+          return statusResponse.assets.image;
+        case "failed":
+          throw new import_provider.InvalidResponseDataError({
+            data: statusResponse,
+            message: `Image generation failed.`
+          });
+      }
+      await delay(this.pollIntervalMillis);
+    }
+    throw new Error(
+      `Image generation timed out after ${this.maxPollAttempts} attempts.`
+    );
+  }
+  createLumaErrorHandler() {
+    return (0, import_provider_utils.createJsonErrorResponseHandler)({
+      errorSchema: lumaErrorSchema,
+      errorToMessage: (error) => {
+        var _a;
+        return (_a = error.detail[0].msg) != null ? _a : "Unknown error";
+      }
+    });
+  }
+  getLumaGenerationsUrl(generationId) {
+    return `${this.config.baseURL}/dream-machine/v1/generations/${generationId != null ? generationId : "image"}`;
+  }
+  async downloadImage(url, abortSignal) {
+    const { value: response } = await (0, import_provider_utils.getFromApi)({
+      url,
+      // No specific headers should be needed for this request as it's a
+      // generated image provided by Luma.
+      abortSignal,
+      failedResponseHandler: (0, import_provider_utils.createStatusCodeErrorResponseHandler)(),
+      successfulResponseHandler: (0, import_provider_utils.createBinaryResponseHandler)(),
+      fetch: this.config.fetch
+    });
+    return response;
+  }
+};
+var lumaGenerationResponseSchema = import_zod.z.object({
+  id: import_zod.z.string(),
+  state: import_zod.z.enum(["queued", "dreaming", "completed", "failed"]),
+  failure_reason: import_zod.z.string().nullish(),
+  assets: import_zod.z.object({
+    image: import_zod.z.string()
+    // URL of the generated image
+  }).nullish()
+});
+var lumaErrorSchema = import_zod.z.object({
+  detail: import_zod.z.array(
+    import_zod.z.object({
+      type: import_zod.z.string(),
+      loc: import_zod.z.array(import_zod.z.string()),
+      msg: import_zod.z.string(),
+      input: import_zod.z.string(),
+      ctx: import_zod.z.object({
+        expected: import_zod.z.string()
+      }).nullish()
+    })
+  )
+});
+
+// src/luma-provider.ts
+var defaultBaseURL = "https://api.lumalabs.ai";
+function createLuma(options = {}) {
+  var _a;
+  const baseURL = (0, import_provider_utils2.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : defaultBaseURL);
+  const getHeaders = () => ({
+    Authorization: `Bearer ${(0, import_provider_utils2.loadApiKey)({
+      apiKey: options.apiKey,
+      environmentVariableName: "LUMA_API_KEY",
+      description: "Luma"
+    })}`,
+    ...options.headers
+  });
+  const createImageModel = (modelId, settings = {}) => new LumaImageModel(modelId, settings, {
+    provider: "luma.image",
+    baseURL: baseURL != null ? baseURL : defaultBaseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
+  const provider = (modelId, settings) => createImageModel(modelId, settings);
+  provider.image = createImageModel;
+  return provider;
+}
+var luma = createLuma();
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createLuma,
+  luma
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/luma-provider.ts","../src/luma-image-model.ts"],"sourcesContent":["export { createLuma, luma } from './luma-provider';\nexport type { LumaProvider, LumaProviderSettings } from './luma-provider';\nexport type { LumaErrorData } from './luma-image-model';\n","import { ImageModelV1 } from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { LumaImageModel } from './luma-image-model';\nimport { LumaImageModelId, LumaImageSettings } from './luma-image-settings';\n\nexport interface LumaProviderSettings {\n  /**\nLuma API key. Default value is taken from the `LUMA_API_KEY` environment\nvariable.\n  */\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n  */\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n  */\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n  */\n  fetch?: FetchFunction;\n}\n\nexport interface LumaProvider {\n  /**\nCreates a model for image generation.\n  */\n  image(modelId: LumaImageModelId, settings?: LumaImageSettings): ImageModelV1;\n}\n\nconst defaultBaseURL = 'https://api.lumalabs.ai';\n\nexport function createLuma(options: LumaProviderSettings = {}): LumaProvider {\n  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'LUMA_API_KEY',\n      description: 'Luma',\n    })}`,\n    ...options.headers,\n  });\n\n  const createImageModel = (\n    modelId: LumaImageModelId,\n    settings: LumaImageSettings = {},\n  ) =>\n    new LumaImageModel(modelId, settings, {\n      provider: 'luma.image',\n      baseURL: baseURL ?? defaultBaseURL,\n      headers: getHeaders,\n      fetch: options.fetch,\n    });\n\n  const provider = (modelId: LumaImageModelId, settings?: LumaImageSettings) =>\n    createImageModel(modelId, settings);\n\n  provider.image = createImageModel;\n\n  return provider as LumaProvider;\n}\n\nexport const luma = createLuma();\n","import {\n  ImageModelV1,\n  ImageModelV1CallWarning,\n  InvalidResponseDataError,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  combineHeaders,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createJsonErrorResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  getFromApi,\n  postJsonToApi,\n} from '@ai-sdk/provider-utils';\nimport { LumaImageSettings } from './luma-image-settings';\nimport { z } from 'zod';\n\nconst DEFAULT_POLL_INTERVAL_MILLIS = 500;\nconst DEFAULT_MAX_POLL_ATTEMPTS = 60000 / DEFAULT_POLL_INTERVAL_MILLIS;\n\ninterface LumaImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers: () => Record<string, string>;\n  fetch?: FetchFunction;\n  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\nasync function delay(delayInMs?: number | null): Promise<void> {\n  return delayInMs == null\n    ? Promise.resolve()\n    : new Promise(resolve => setTimeout(resolve, delayInMs));\n}\n\nexport class LumaImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\n\n  private readonly pollIntervalMillis: number;\n  private readonly maxPollAttempts: number;\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  get maxImagesPerCall(): number {\n    return this.settings.maxImagesPerCall ?? 1;\n  }\n\n  constructor(\n    readonly modelId: string,\n    private readonly settings: LumaImageSettings,\n    private readonly config: LumaImageModelConfig,\n  ) {\n    this.pollIntervalMillis =\n      settings.pollIntervalMillis ?? DEFAULT_POLL_INTERVAL_MILLIS;\n    this.maxPollAttempts =\n      settings.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS;\n  }\n\n  async doGenerate({\n    prompt,\n    n,\n    size,\n    aspectRatio,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    if (seed != null) {\n      warnings.push({\n        type: 'unsupported-setting',\n        setting: 'seed',\n        details: 'This model does not support the `seed` option.',\n      });\n    }\n\n    if (size != null) {\n      warnings.push({\n        type: 'unsupported-setting',\n        setting: 'size',\n        details:\n          'This model does not support the `size` option. Use `aspectRatio` instead.',\n      });\n    }\n\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n    const fullHeaders = combineHeaders(this.config.headers(), headers);\n    const { value: generationResponse, responseHeaders } = await postJsonToApi({\n      url: this.getLumaGenerationsUrl(),\n      headers: fullHeaders,\n      body: {\n        prompt,\n        ...(aspectRatio ? { aspect_ratio: aspectRatio } : {}),\n        model: this.modelId,\n        ...(providerOptions.luma ?? {}),\n      },\n      abortSignal,\n      fetch: this.config.fetch,\n      failedResponseHandler: this.createLumaErrorHandler(),\n      successfulResponseHandler: createJsonResponseHandler(\n        lumaGenerationResponseSchema,\n      ),\n    });\n\n    const imageUrl = await this.pollForImageUrl(\n      generationResponse.id,\n      fullHeaders,\n      abortSignal,\n    );\n\n    const downloadedImage = await this.downloadImage(imageUrl, abortSignal);\n\n    return {\n      images: [downloadedImage],\n      warnings,\n      response: {\n        modelId: this.modelId,\n        timestamp: currentDate,\n        headers: responseHeaders,\n      },\n    };\n  }\n\n  private async pollForImageUrl(\n    generationId: string,\n    headers: Record<string, string | undefined>,\n    abortSignal: AbortSignal | undefined,\n  ): Promise<string> {\n    let attemptCount = 0;\n    const url = this.getLumaGenerationsUrl(generationId);\n    for (let i = 0; i < this.maxPollAttempts; i++) {\n      const { value: statusResponse } = await getFromApi({\n        url,\n        headers,\n        abortSignal,\n        fetch: this.config.fetch,\n        failedResponseHandler: this.createLumaErrorHandler(),\n        successfulResponseHandler: createJsonResponseHandler(\n          lumaGenerationResponseSchema,\n        ),\n      });\n\n      switch (statusResponse.state) {\n        case 'completed':\n          if (!statusResponse.assets?.image) {\n            throw new InvalidResponseDataError({\n              data: statusResponse,\n              message: `Image generation completed but no image was found.`,\n            });\n          }\n          return statusResponse.assets.image;\n        case 'failed':\n          throw new InvalidResponseDataError({\n            data: statusResponse,\n            message: `Image generation failed.`,\n          });\n      }\n      await delay(this.pollIntervalMillis);\n    }\n\n    throw new Error(\n      `Image generation timed out after ${this.maxPollAttempts} attempts.`,\n    );\n  }\n\n  private createLumaErrorHandler() {\n    return createJsonErrorResponseHandler({\n      errorSchema: lumaErrorSchema,\n      errorToMessage: (error: LumaErrorData) =>\n        error.detail[0].msg ?? 'Unknown error',\n    });\n  }\n\n  private getLumaGenerationsUrl(generationId?: string) {\n    return `${this.config.baseURL}/dream-machine/v1/generations/${\n      generationId ?? 'image'\n    }`;\n  }\n\n  private async downloadImage(\n    url: string,\n    abortSignal: AbortSignal | undefined,\n  ): Promise<Uint8Array> {\n    const { value: response } = await getFromApi({\n      url,\n      // No specific headers should be needed for this request as it's a\n      // generated image provided by Luma.\n      abortSignal,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createBinaryResponseHandler(),\n      fetch: this.config.fetch,\n    });\n    return response;\n  }\n}\n\n// limited version of the schema, focussed on what is needed for the implementation\n// this approach limits breakages when the API changes and increases efficiency\nconst lumaGenerationResponseSchema = z.object({\n  id: z.string(),\n  state: z.enum(['queued', 'dreaming', 'completed', 'failed']),\n  failure_reason: z.string().nullish(),\n  assets: z\n    .object({\n      image: z.string(), // URL of the generated image\n    })\n    .nullish(),\n});\n\nconst lumaErrorSchema = z.object({\n  detail: z.array(\n    z.object({\n      type: z.string(),\n      loc: z.array(z.string()),\n      msg: z.string(),\n      input: z.string(),\n      ctx: z\n        .object({\n          expected: z.string(),\n        })\n        .nullish(),\n    }),\n  ),\n});\n\nexport type LumaErrorData = z.infer<typeof lumaErrorSchema>;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,IAAAA,yBAIO;;;ACLP,sBAIO;AACP,4BASO;AAEP,iBAAkB;AAElB,IAAM,+BAA+B;AACrC,IAAM,4BAA4B,MAAQ;AAY1C,eAAe,MAAM,WAA0C;AAC7D,SAAO,aAAa,OAChB,QAAQ,QAAQ,IAChB,IAAI,QAAQ,aAAW,WAAW,SAAS,SAAS,CAAC;AAC3D;AAEO,IAAM,iBAAN,MAA6C;AAAA,EAclD,YACW,SACQ,UACA,QACjB;AAHS;AACQ;AACA;AAhBnB,SAAS,uBAAuB;AAtClC;AAwDI,SAAK,sBACH,cAAS,uBAAT,YAA+B;AACjC,SAAK,mBACH,cAAS,oBAAT,YAA4B;AAAA,EAChC;AAAA,EAjBA,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AA/CjC;AAgDI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAaA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAzEJ;AA0EI,UAAM,WAA2C,CAAC;AAElD,QAAI,QAAQ,MAAM;AAChB,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AAEA,QAAI,QAAQ,MAAM;AAChB,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAEA,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AACvE,UAAM,kBAAc,sCAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AACjE,UAAM,EAAE,OAAO,oBAAoB,gBAAgB,IAAI,UAAM,qCAAc;AAAA,MACzE,KAAK,KAAK,sBAAsB;AAAA,MAChC,SAAS;AAAA,MACT,MAAM;AAAA,QACJ;AAAA,QACA,GAAI,cAAc,EAAE,cAAc,YAAY,IAAI,CAAC;AAAA,QACnD,OAAO,KAAK;AAAA,QACZ,IAAI,qBAAgB,SAAhB,YAAwB,CAAC;AAAA,MAC/B;AAAA,MACA;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,MACnB,uBAAuB,KAAK,uBAAuB;AAAA,MACnD,+BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,IACF,CAAC;AAED,UAAM,WAAW,MAAM,KAAK;AAAA,MAC1B,mBAAmB;AAAA,MACnB;AAAA,MACA;AAAA,IACF;AAEA,UAAM,kBAAkB,MAAM,KAAK,cAAc,UAAU,WAAW;AAEtE,WAAO;AAAA,MACL,QAAQ,CAAC,eAAe;AAAA,MACxB;AAAA,MACA,UAAU;AAAA,QACR,SAAS,KAAK;AAAA,QACd,WAAW;AAAA,QACX,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBACZ,cACA,SACA,aACiB;AAvIrB;AAwII,QAAI,eAAe;AACnB,UAAM,MAAM,KAAK,sBAAsB,YAAY;AACnD,aAAS,IAAI,GAAG,IAAI,KAAK,iBAAiB,KAAK;AAC7C,YAAM,EAAE,OAAO,eAAe,IAAI,UAAM,kCAAW;AAAA,QACjD;AAAA,QACA;AAAA,QACA;AAAA,QACA,OAAO,KAAK,OAAO;AAAA,QACnB,uBAAuB,KAAK,uBAAuB;AAAA,QACnD,+BAA2B;AAAA,UACzB;AAAA,QACF;AAAA,MACF,CAAC;AAED,cAAQ,eAAe,OAAO;AAAA,QAC5B,KAAK;AACH,cAAI,GAAC,oBAAe,WAAf,mBAAuB,QAAO;AACjC,kBAAM,IAAI,yCAAyB;AAAA,cACjC,MAAM;AAAA,cACN,SAAS;AAAA,YACX,CAAC;AAAA,UACH;AACA,iBAAO,eAAe,OAAO;AAAA,QAC/B,KAAK;AACH,gBAAM,IAAI,yCAAyB;AAAA,YACjC,MAAM;AAAA,YACN,SAAS;AAAA,UACX,CAAC;AAAA,MACL;AACA,YAAM,MAAM,KAAK,kBAAkB;AAAA,IACrC;AAEA,UAAM,IAAI;AAAA,MACR,oCAAoC,KAAK,eAAe;AAAA,IAC1D;AAAA,EACF;AAAA,EAEQ,yBAAyB;AAC/B,eAAO,sDAA+B;AAAA,MACpC,aAAa;AAAA,MACb,gBAAgB,CAAC,UAAsB;AAhL7C;AAiLQ,2BAAM,OAAO,CAAC,EAAE,QAAhB,YAAuB;AAAA;AAAA,IAC3B,CAAC;AAAA,EACH;AAAA,EAEQ,sBAAsB,cAAuB;AACnD,WAAO,GAAG,KAAK,OAAO,OAAO,iCAC3B,sCAAgB,OAClB;AAAA,EACF;AAAA,EAEA,MAAc,cACZ,KACA,aACqB;AACrB,UAAM,EAAE,OAAO,SAAS,IAAI,UAAM,kCAAW;AAAA,MAC3C;AAAA;AAAA;AAAA,MAGA;AAAA,MACA,2BAAuB,4DAAqC;AAAA,MAC5D,+BAA2B,mDAA4B;AAAA,MACvD,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AACD,WAAO;AAAA,EACT;AACF;AAIA,IAAM,+BAA+B,aAAE,OAAO;AAAA,EAC5C,IAAI,aAAE,OAAO;AAAA,EACb,OAAO,aAAE,KAAK,CAAC,UAAU,YAAY,aAAa,QAAQ,CAAC;AAAA,EAC3D,gBAAgB,aAAE,OAAO,EAAE,QAAQ;AAAA,EACnC,QAAQ,aACL,OAAO;AAAA,IACN,OAAO,aAAE,OAAO;AAAA;AAAA,EAClB,CAAC,EACA,QAAQ;AACb,CAAC;AAED,IAAM,kBAAkB,aAAE,OAAO;AAAA,EAC/B,QAAQ,aAAE;AAAA,IACR,aAAE,OAAO;AAAA,MACP,MAAM,aAAE,OAAO;AAAA,MACf,KAAK,aAAE,MAAM,aAAE,OAAO,CAAC;AAAA,MACvB,KAAK,aAAE,OAAO;AAAA,MACd,OAAO,aAAE,OAAO;AAAA,MAChB,KAAK,aACF,OAAO;AAAA,QACN,UAAU,aAAE,OAAO;AAAA,MACrB,CAAC,EACA,QAAQ;AAAA,IACb,CAAC;AAAA,EACH;AACF,CAAC;;;ADlMD,IAAM,iBAAiB;AAEhB,SAAS,WAAW,UAAgC,CAAC,GAAiB;AAvC7E;AAwCE,QAAM,cAAU,8CAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,mCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,mBAAmB,CACvB,SACA,WAA8B,CAAC,MAE/B,IAAI,eAAe,SAAS,UAAU;AAAA,IACpC,UAAU;AAAA,IACV,SAAS,4BAAW;AAAA,IACpB,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB,CAAC;AAEH,QAAM,WAAW,CAAC,SAA2B,aAC3C,iBAAiB,SAAS,QAAQ;AAEpC,WAAS,QAAQ;AAEjB,SAAO;AACT;AAEO,IAAM,OAAO,WAAW;","names":["import_provider_utils"]}

package/dist/index.mjs

@@ -0,0 +1,215 @@
+// src/luma-provider.ts
+import {
+  loadApiKey,
+  withoutTrailingSlash
+} from "@ai-sdk/provider-utils";
+
+// src/luma-image-model.ts
+import {
+  InvalidResponseDataError
+} from "@ai-sdk/provider";
+import {
+  combineHeaders,
+  createBinaryResponseHandler,
+  createJsonResponseHandler,
+  createJsonErrorResponseHandler,
+  createStatusCodeErrorResponseHandler,
+  getFromApi,
+  postJsonToApi
+} from "@ai-sdk/provider-utils";
+import { z } from "zod";
+var DEFAULT_POLL_INTERVAL_MILLIS = 500;
+var DEFAULT_MAX_POLL_ATTEMPTS = 6e4 / DEFAULT_POLL_INTERVAL_MILLIS;
+async function delay(delayInMs) {
+  return delayInMs == null ? Promise.resolve() : new Promise((resolve) => setTimeout(resolve, delayInMs));
+}
+var LumaImageModel = class {
+  constructor(modelId, settings, config) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+    this.specificationVersion = "v1";
+    var _a, _b;
+    this.pollIntervalMillis = (_a = settings.pollIntervalMillis) != null ? _a : DEFAULT_POLL_INTERVAL_MILLIS;
+    this.maxPollAttempts = (_b = settings.maxPollAttempts) != null ? _b : DEFAULT_MAX_POLL_ATTEMPTS;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  get maxImagesPerCall() {
+    var _a;
+    return (_a = this.settings.maxImagesPerCall) != null ? _a : 1;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    var _a, _b, _c, _d;
+    const warnings = [];
+    if (seed != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "seed",
+        details: "This model does not support the `seed` option."
+      });
+    }
+    if (size != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "size",
+        details: "This model does not support the `size` option. Use `aspectRatio` instead."
+      });
+    }
+    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 fullHeaders = combineHeaders(this.config.headers(), headers);
+    const { value: generationResponse, responseHeaders } = await postJsonToApi({
+      url: this.getLumaGenerationsUrl(),
+      headers: fullHeaders,
+      body: {
+        prompt,
+        ...aspectRatio ? { aspect_ratio: aspectRatio } : {},
+        model: this.modelId,
+        ...(_d = providerOptions.luma) != null ? _d : {}
+      },
+      abortSignal,
+      fetch: this.config.fetch,
+      failedResponseHandler: this.createLumaErrorHandler(),
+      successfulResponseHandler: createJsonResponseHandler(
+        lumaGenerationResponseSchema
+      )
+    });
+    const imageUrl = await this.pollForImageUrl(
+      generationResponse.id,
+      fullHeaders,
+      abortSignal
+    );
+    const downloadedImage = await this.downloadImage(imageUrl, abortSignal);
+    return {
+      images: [downloadedImage],
+      warnings,
+      response: {
+        modelId: this.modelId,
+        timestamp: currentDate,
+        headers: responseHeaders
+      }
+    };
+  }
+  async pollForImageUrl(generationId, headers, abortSignal) {
+    var _a;
+    let attemptCount = 0;
+    const url = this.getLumaGenerationsUrl(generationId);
+    for (let i = 0; i < this.maxPollAttempts; i++) {
+      const { value: statusResponse } = await getFromApi({
+        url,
+        headers,
+        abortSignal,
+        fetch: this.config.fetch,
+        failedResponseHandler: this.createLumaErrorHandler(),
+        successfulResponseHandler: createJsonResponseHandler(
+          lumaGenerationResponseSchema
+        )
+      });
+      switch (statusResponse.state) {
+        case "completed":
+          if (!((_a = statusResponse.assets) == null ? void 0 : _a.image)) {
+            throw new InvalidResponseDataError({
+              data: statusResponse,
+              message: `Image generation completed but no image was found.`
+            });
+          }
+          return statusResponse.assets.image;
+        case "failed":
+          throw new InvalidResponseDataError({
+            data: statusResponse,
+            message: `Image generation failed.`
+          });
+      }
+      await delay(this.pollIntervalMillis);
+    }
+    throw new Error(
+      `Image generation timed out after ${this.maxPollAttempts} attempts.`
+    );
+  }
+  createLumaErrorHandler() {
+    return createJsonErrorResponseHandler({
+      errorSchema: lumaErrorSchema,
+      errorToMessage: (error) => {
+        var _a;
+        return (_a = error.detail[0].msg) != null ? _a : "Unknown error";
+      }
+    });
+  }
+  getLumaGenerationsUrl(generationId) {
+    return `${this.config.baseURL}/dream-machine/v1/generations/${generationId != null ? generationId : "image"}`;
+  }
+  async downloadImage(url, abortSignal) {
+    const { value: response } = await getFromApi({
+      url,
+      // No specific headers should be needed for this request as it's a
+      // generated image provided by Luma.
+      abortSignal,
+      failedResponseHandler: createStatusCodeErrorResponseHandler(),
+      successfulResponseHandler: createBinaryResponseHandler(),
+      fetch: this.config.fetch
+    });
+    return response;
+  }
+};
+var lumaGenerationResponseSchema = z.object({
+  id: z.string(),
+  state: z.enum(["queued", "dreaming", "completed", "failed"]),
+  failure_reason: z.string().nullish(),
+  assets: z.object({
+    image: z.string()
+    // URL of the generated image
+  }).nullish()
+});
+var lumaErrorSchema = z.object({
+  detail: z.array(
+    z.object({
+      type: z.string(),
+      loc: z.array(z.string()),
+      msg: z.string(),
+      input: z.string(),
+      ctx: z.object({
+        expected: z.string()
+      }).nullish()
+    })
+  )
+});
+
+// src/luma-provider.ts
+var defaultBaseURL = "https://api.lumalabs.ai";
+function createLuma(options = {}) {
+  var _a;
+  const baseURL = withoutTrailingSlash((_a = options.baseURL) != null ? _a : defaultBaseURL);
+  const getHeaders = () => ({
+    Authorization: `Bearer ${loadApiKey({
+      apiKey: options.apiKey,
+      environmentVariableName: "LUMA_API_KEY",
+      description: "Luma"
+    })}`,
+    ...options.headers
+  });
+  const createImageModel = (modelId, settings = {}) => new LumaImageModel(modelId, settings, {
+    provider: "luma.image",
+    baseURL: baseURL != null ? baseURL : defaultBaseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
+  const provider = (modelId, settings) => createImageModel(modelId, settings);
+  provider.image = createImageModel;
+  return provider;
+}
+var luma = createLuma();
+export {
+  createLuma,
+  luma
+};
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/luma-provider.ts","../src/luma-image-model.ts"],"sourcesContent":["import { ImageModelV1 } from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { LumaImageModel } from './luma-image-model';\nimport { LumaImageModelId, LumaImageSettings } from './luma-image-settings';\n\nexport interface LumaProviderSettings {\n  /**\nLuma API key. Default value is taken from the `LUMA_API_KEY` environment\nvariable.\n  */\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n  */\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n  */\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n  */\n  fetch?: FetchFunction;\n}\n\nexport interface LumaProvider {\n  /**\nCreates a model for image generation.\n  */\n  image(modelId: LumaImageModelId, settings?: LumaImageSettings): ImageModelV1;\n}\n\nconst defaultBaseURL = 'https://api.lumalabs.ai';\n\nexport function createLuma(options: LumaProviderSettings = {}): LumaProvider {\n  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'LUMA_API_KEY',\n      description: 'Luma',\n    })}`,\n    ...options.headers,\n  });\n\n  const createImageModel = (\n    modelId: LumaImageModelId,\n    settings: LumaImageSettings = {},\n  ) =>\n    new LumaImageModel(modelId, settings, {\n      provider: 'luma.image',\n      baseURL: baseURL ?? defaultBaseURL,\n      headers: getHeaders,\n      fetch: options.fetch,\n    });\n\n  const provider = (modelId: LumaImageModelId, settings?: LumaImageSettings) =>\n    createImageModel(modelId, settings);\n\n  provider.image = createImageModel;\n\n  return provider as LumaProvider;\n}\n\nexport const luma = createLuma();\n","import {\n  ImageModelV1,\n  ImageModelV1CallWarning,\n  InvalidResponseDataError,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  combineHeaders,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createJsonErrorResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  getFromApi,\n  postJsonToApi,\n} from '@ai-sdk/provider-utils';\nimport { LumaImageSettings } from './luma-image-settings';\nimport { z } from 'zod';\n\nconst DEFAULT_POLL_INTERVAL_MILLIS = 500;\nconst DEFAULT_MAX_POLL_ATTEMPTS = 60000 / DEFAULT_POLL_INTERVAL_MILLIS;\n\ninterface LumaImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers: () => Record<string, string>;\n  fetch?: FetchFunction;\n  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\nasync function delay(delayInMs?: number | null): Promise<void> {\n  return delayInMs == null\n    ? Promise.resolve()\n    : new Promise(resolve => setTimeout(resolve, delayInMs));\n}\n\nexport class LumaImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\n\n  private readonly pollIntervalMillis: number;\n  private readonly maxPollAttempts: number;\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  get maxImagesPerCall(): number {\n    return this.settings.maxImagesPerCall ?? 1;\n  }\n\n  constructor(\n    readonly modelId: string,\n    private readonly settings: LumaImageSettings,\n    private readonly config: LumaImageModelConfig,\n  ) {\n    this.pollIntervalMillis =\n      settings.pollIntervalMillis ?? DEFAULT_POLL_INTERVAL_MILLIS;\n    this.maxPollAttempts =\n      settings.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS;\n  }\n\n  async doGenerate({\n    prompt,\n    n,\n    size,\n    aspectRatio,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    if (seed != null) {\n      warnings.push({\n        type: 'unsupported-setting',\n        setting: 'seed',\n        details: 'This model does not support the `seed` option.',\n      });\n    }\n\n    if (size != null) {\n      warnings.push({\n        type: 'unsupported-setting',\n        setting: 'size',\n        details:\n          'This model does not support the `size` option. Use `aspectRatio` instead.',\n      });\n    }\n\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n    const fullHeaders = combineHeaders(this.config.headers(), headers);\n    const { value: generationResponse, responseHeaders } = await postJsonToApi({\n      url: this.getLumaGenerationsUrl(),\n      headers: fullHeaders,\n      body: {\n        prompt,\n        ...(aspectRatio ? { aspect_ratio: aspectRatio } : {}),\n        model: this.modelId,\n        ...(providerOptions.luma ?? {}),\n      },\n      abortSignal,\n      fetch: this.config.fetch,\n      failedResponseHandler: this.createLumaErrorHandler(),\n      successfulResponseHandler: createJsonResponseHandler(\n        lumaGenerationResponseSchema,\n      ),\n    });\n\n    const imageUrl = await this.pollForImageUrl(\n      generationResponse.id,\n      fullHeaders,\n      abortSignal,\n    );\n\n    const downloadedImage = await this.downloadImage(imageUrl, abortSignal);\n\n    return {\n      images: [downloadedImage],\n      warnings,\n      response: {\n        modelId: this.modelId,\n        timestamp: currentDate,\n        headers: responseHeaders,\n      },\n    };\n  }\n\n  private async pollForImageUrl(\n    generationId: string,\n    headers: Record<string, string | undefined>,\n    abortSignal: AbortSignal | undefined,\n  ): Promise<string> {\n    let attemptCount = 0;\n    const url = this.getLumaGenerationsUrl(generationId);\n    for (let i = 0; i < this.maxPollAttempts; i++) {\n      const { value: statusResponse } = await getFromApi({\n        url,\n        headers,\n        abortSignal,\n        fetch: this.config.fetch,\n        failedResponseHandler: this.createLumaErrorHandler(),\n        successfulResponseHandler: createJsonResponseHandler(\n          lumaGenerationResponseSchema,\n        ),\n      });\n\n      switch (statusResponse.state) {\n        case 'completed':\n          if (!statusResponse.assets?.image) {\n            throw new InvalidResponseDataError({\n              data: statusResponse,\n              message: `Image generation completed but no image was found.`,\n            });\n          }\n          return statusResponse.assets.image;\n        case 'failed':\n          throw new InvalidResponseDataError({\n            data: statusResponse,\n            message: `Image generation failed.`,\n          });\n      }\n      await delay(this.pollIntervalMillis);\n    }\n\n    throw new Error(\n      `Image generation timed out after ${this.maxPollAttempts} attempts.`,\n    );\n  }\n\n  private createLumaErrorHandler() {\n    return createJsonErrorResponseHandler({\n      errorSchema: lumaErrorSchema,\n      errorToMessage: (error: LumaErrorData) =>\n        error.detail[0].msg ?? 'Unknown error',\n    });\n  }\n\n  private getLumaGenerationsUrl(generationId?: string) {\n    return `${this.config.baseURL}/dream-machine/v1/generations/${\n      generationId ?? 'image'\n    }`;\n  }\n\n  private async downloadImage(\n    url: string,\n    abortSignal: AbortSignal | undefined,\n  ): Promise<Uint8Array> {\n    const { value: response } = await getFromApi({\n      url,\n      // No specific headers should be needed for this request as it's a\n      // generated image provided by Luma.\n      abortSignal,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createBinaryResponseHandler(),\n      fetch: this.config.fetch,\n    });\n    return response;\n  }\n}\n\n// limited version of the schema, focussed on what is needed for the implementation\n// this approach limits breakages when the API changes and increases efficiency\nconst lumaGenerationResponseSchema = z.object({\n  id: z.string(),\n  state: z.enum(['queued', 'dreaming', 'completed', 'failed']),\n  failure_reason: z.string().nullish(),\n  assets: z\n    .object({\n      image: z.string(), // URL of the generated image\n    })\n    .nullish(),\n});\n\nconst lumaErrorSchema = z.object({\n  detail: z.array(\n    z.object({\n      type: z.string(),\n      loc: z.array(z.string()),\n      msg: z.string(),\n      input: z.string(),\n      ctx: z\n        .object({\n          expected: z.string(),\n        })\n        .nullish(),\n    }),\n  ),\n});\n\nexport type LumaErrorData = z.infer<typeof lumaErrorSchema>;\n"],"mappings":";AACA;AAAA,EAEE;AAAA,EACA;AAAA,OACK;;;ACLP;AAAA,EAGE;AAAA,OACK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP,SAAS,SAAS;AAElB,IAAM,+BAA+B;AACrC,IAAM,4BAA4B,MAAQ;AAY1C,eAAe,MAAM,WAA0C;AAC7D,SAAO,aAAa,OAChB,QAAQ,QAAQ,IAChB,IAAI,QAAQ,aAAW,WAAW,SAAS,SAAS,CAAC;AAC3D;AAEO,IAAM,iBAAN,MAA6C;AAAA,EAclD,YACW,SACQ,UACA,QACjB;AAHS;AACQ;AACA;AAhBnB,SAAS,uBAAuB;AAtClC;AAwDI,SAAK,sBACH,cAAS,uBAAT,YAA+B;AACjC,SAAK,mBACH,cAAS,oBAAT,YAA4B;AAAA,EAChC;AAAA,EAjBA,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,IAAI,mBAA2B;AA/CjC;AAgDI,YAAO,UAAK,SAAS,qBAAd,YAAkC;AAAA,EAC3C;AAAA,EAaA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAzEJ;AA0EI,UAAM,WAA2C,CAAC;AAElD,QAAI,QAAQ,MAAM;AAChB,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AAEA,QAAI,QAAQ,MAAM;AAChB,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAEA,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AACvE,UAAM,cAAc,eAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AACjE,UAAM,EAAE,OAAO,oBAAoB,gBAAgB,IAAI,MAAM,cAAc;AAAA,MACzE,KAAK,KAAK,sBAAsB;AAAA,MAChC,SAAS;AAAA,MACT,MAAM;AAAA,QACJ;AAAA,QACA,GAAI,cAAc,EAAE,cAAc,YAAY,IAAI,CAAC;AAAA,QACnD,OAAO,KAAK;AAAA,QACZ,IAAI,qBAAgB,SAAhB,YAAwB,CAAC;AAAA,MAC/B;AAAA,MACA;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,MACnB,uBAAuB,KAAK,uBAAuB;AAAA,MACnD,2BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,IACF,CAAC;AAED,UAAM,WAAW,MAAM,KAAK;AAAA,MAC1B,mBAAmB;AAAA,MACnB;AAAA,MACA;AAAA,IACF;AAEA,UAAM,kBAAkB,MAAM,KAAK,cAAc,UAAU,WAAW;AAEtE,WAAO;AAAA,MACL,QAAQ,CAAC,eAAe;AAAA,MACxB;AAAA,MACA,UAAU;AAAA,QACR,SAAS,KAAK;AAAA,QACd,WAAW;AAAA,QACX,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,gBACZ,cACA,SACA,aACiB;AAvIrB;AAwII,QAAI,eAAe;AACnB,UAAM,MAAM,KAAK,sBAAsB,YAAY;AACnD,aAAS,IAAI,GAAG,IAAI,KAAK,iBAAiB,KAAK;AAC7C,YAAM,EAAE,OAAO,eAAe,IAAI,MAAM,WAAW;AAAA,QACjD;AAAA,QACA;AAAA,QACA;AAAA,QACA,OAAO,KAAK,OAAO;AAAA,QACnB,uBAAuB,KAAK,uBAAuB;AAAA,QACnD,2BAA2B;AAAA,UACzB;AAAA,QACF;AAAA,MACF,CAAC;AAED,cAAQ,eAAe,OAAO;AAAA,QAC5B,KAAK;AACH,cAAI,GAAC,oBAAe,WAAf,mBAAuB,QAAO;AACjC,kBAAM,IAAI,yBAAyB;AAAA,cACjC,MAAM;AAAA,cACN,SAAS;AAAA,YACX,CAAC;AAAA,UACH;AACA,iBAAO,eAAe,OAAO;AAAA,QAC/B,KAAK;AACH,gBAAM,IAAI,yBAAyB;AAAA,YACjC,MAAM;AAAA,YACN,SAAS;AAAA,UACX,CAAC;AAAA,MACL;AACA,YAAM,MAAM,KAAK,kBAAkB;AAAA,IACrC;AAEA,UAAM,IAAI;AAAA,MACR,oCAAoC,KAAK,eAAe;AAAA,IAC1D;AAAA,EACF;AAAA,EAEQ,yBAAyB;AAC/B,WAAO,+BAA+B;AAAA,MACpC,aAAa;AAAA,MACb,gBAAgB,CAAC,UAAsB;AAhL7C;AAiLQ,2BAAM,OAAO,CAAC,EAAE,QAAhB,YAAuB;AAAA;AAAA,IAC3B,CAAC;AAAA,EACH;AAAA,EAEQ,sBAAsB,cAAuB;AACnD,WAAO,GAAG,KAAK,OAAO,OAAO,iCAC3B,sCAAgB,OAClB;AAAA,EACF;AAAA,EAEA,MAAc,cACZ,KACA,aACqB;AACrB,UAAM,EAAE,OAAO,SAAS,IAAI,MAAM,WAAW;AAAA,MAC3C;AAAA;AAAA;AAAA,MAGA;AAAA,MACA,uBAAuB,qCAAqC;AAAA,MAC5D,2BAA2B,4BAA4B;AAAA,MACvD,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AACD,WAAO;AAAA,EACT;AACF;AAIA,IAAM,+BAA+B,EAAE,OAAO;AAAA,EAC5C,IAAI,EAAE,OAAO;AAAA,EACb,OAAO,EAAE,KAAK,CAAC,UAAU,YAAY,aAAa,QAAQ,CAAC;AAAA,EAC3D,gBAAgB,EAAE,OAAO,EAAE,QAAQ;AAAA,EACnC,QAAQ,EACL,OAAO;AAAA,IACN,OAAO,EAAE,OAAO;AAAA;AAAA,EAClB,CAAC,EACA,QAAQ;AACb,CAAC;AAED,IAAM,kBAAkB,EAAE,OAAO;AAAA,EAC/B,QAAQ,EAAE;AAAA,IACR,EAAE,OAAO;AAAA,MACP,MAAM,EAAE,OAAO;AAAA,MACf,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC;AAAA,MACvB,KAAK,EAAE,OAAO;AAAA,MACd,OAAO,EAAE,OAAO;AAAA,MAChB,KAAK,EACF,OAAO;AAAA,QACN,UAAU,EAAE,OAAO;AAAA,MACrB,CAAC,EACA,QAAQ;AAAA,IACb,CAAC;AAAA,EACH;AACF,CAAC;;;ADlMD,IAAM,iBAAiB;AAEhB,SAAS,WAAW,UAAgC,CAAC,GAAiB;AAvC7E;AAwCE,QAAM,UAAU,sBAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,mBAAmB,CACvB,SACA,WAA8B,CAAC,MAE/B,IAAI,eAAe,SAAS,UAAU;AAAA,IACpC,UAAU;AAAA,IACV,SAAS,4BAAW;AAAA,IACpB,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB,CAAC;AAEH,QAAM,WAAW,CAAC,SAA2B,aAC3C,iBAAiB,SAAS,QAAQ;AAEpC,WAAS,QAAQ;AAEjB,SAAO;AACT;AAEO,IAAM,OAAO,WAAW;","names":[]}

package/package.json

@@ -0,0 +1,63 @@
+{
+  "name": "@ai-sdk/luma",
+  "version": "0.0.1",
+  "license": "Apache-2.0",
+  "sideEffects": false,
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist/**/*",
+    "CHANGELOG.md"
+  ],
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "@ai-sdk/provider": "1.0.6",
+    "@ai-sdk/provider-utils": "2.1.3"
+  },
+  "devDependencies": {
+    "@types/node": "^18",
+    "tsup": "^8",
+    "typescript": "5.6.3",
+    "zod": "3.23.8",
+    "@vercel/ai-tsconfig": "0.0.0"
+  },
+  "peerDependencies": {
+    "zod": "^3.0.0"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "homepage": "https://sdk.vercel.ai/docs",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/vercel/ai.git"
+  },
+  "bugs": {
+    "url": "https://github.com/vercel/ai/issues"
+  },
+  "keywords": [
+    "ai"
+  ],
+  "scripts": {
+    "build": "tsup",
+    "build:watch": "tsup --watch",
+    "clean": "rm -rf dist",
+    "lint": "eslint \"./**/*.ts*\"",
+    "type-check": "tsc --noEmit",
+    "prettier-check": "prettier --check \"./**/*.ts*\"",
+    "test": "pnpm test:node && pnpm test:edge",
+    "test:edge": "vitest --config vitest.edge.config.js --run",
+    "test:node": "vitest --config vitest.node.config.js --run"
+  }
+}