@ai-sdk/black-forest-labs 0.0.0-70e0935a-20260114150030 → 0.0.0-98261322-20260122142521

package/src/black-forest-labs-provider.ts

@@ -0,0 +1,113 @@
+import { ImageModelV3, NoSuchModelError, ProviderV3 } from '@ai-sdk/provider';
+import type { FetchFunction } from '@ai-sdk/provider-utils';
+import {
+  loadApiKey,
+  withoutTrailingSlash,
+  withUserAgentSuffix,
+} from '@ai-sdk/provider-utils';
+import { BlackForestLabsImageModel } from './black-forest-labs-image-model';
+import { BlackForestLabsImageModelId } from './black-forest-labs-image-settings';
+import { VERSION } from './version';
+
+export interface BlackForestLabsProviderSettings {
+  /**
+Black Forest Labs API key. Default value is taken from the `BFL_API_KEY` environment variable.
+   */
+  apiKey?: string;
+
+  /**
+Base URL for the API calls. Defaults to `https://api.bfl.ai/v1`.
+   */
+  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;
+
+  /**
+ Poll interval in milliseconds between status checks. Defaults to 500ms.
+   */
+  pollIntervalMillis?: number;
+
+  /**
+ Overall timeout in milliseconds for polling before giving up. Defaults to 60s.
+   */
+  pollTimeoutMillis?: number;
+}
+
+export interface BlackForestLabsProvider extends ProviderV3 {
+  /**
+Creates a model for image generation.
+   */
+  image(modelId: BlackForestLabsImageModelId): ImageModelV3;
+
+  /**
+Creates a model for image generation.
+   */
+  imageModel(modelId: BlackForestLabsImageModelId): ImageModelV3;
+
+  /**
+   * @deprecated Use `embeddingModel` instead.
+   */
+  textEmbeddingModel(modelId: string): never;
+}
+
+const defaultBaseURL = 'https://api.bfl.ai/v1';
+
+export function createBlackForestLabs(
+  options: BlackForestLabsProviderSettings = {},
+): BlackForestLabsProvider {
+  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);
+  const getHeaders = () =>
+    withUserAgentSuffix(
+      {
+        'x-key': loadApiKey({
+          apiKey: options.apiKey,
+          environmentVariableName: 'BFL_API_KEY',
+          description: 'Black Forest Labs',
+        }),
+        ...options.headers,
+      },
+      `ai-sdk/black-forest-labs/${VERSION}`,
+    );
+
+  const createImageModel = (modelId: BlackForestLabsImageModelId) =>
+    new BlackForestLabsImageModel(modelId, {
+      provider: 'black-forest-labs.image',
+      baseURL: baseURL ?? defaultBaseURL,
+      headers: getHeaders,
+      fetch: options.fetch,
+      pollIntervalMillis: options.pollIntervalMillis,
+      pollTimeoutMillis: options.pollTimeoutMillis,
+    });
+
+  const embeddingModel = (modelId: string) => {
+    throw new NoSuchModelError({
+      modelId,
+      modelType: 'embeddingModel',
+    });
+  };
+
+  return {
+    specificationVersion: 'v3',
+    imageModel: createImageModel,
+    image: createImageModel,
+    languageModel: (modelId: string) => {
+      throw new NoSuchModelError({
+        modelId,
+        modelType: 'languageModel',
+      });
+    },
+    embeddingModel,
+    textEmbeddingModel: embeddingModel,
+  };
+}
+
+export const blackForestLabs = createBlackForestLabs();

package/src/index.ts

@@ -0,0 +1,14 @@
+export {
+  createBlackForestLabs,
+  blackForestLabs,
+} from './black-forest-labs-provider';
+export type {
+  BlackForestLabsProvider,
+  BlackForestLabsProviderSettings,
+} from './black-forest-labs-provider';
+export type {
+  BlackForestLabsImageModelId,
+  BlackForestLabsAspectRatio,
+} from './black-forest-labs-image-settings';
+export type { BlackForestLabsImageProviderOptions } from './black-forest-labs-image-model';
+export { VERSION } from './version';

package/src/version.ts

@@ -0,0 +1,6 @@
+// Version string of this package injected at build time.
+declare const __PACKAGE_VERSION__: string | undefined;
+export const VERSION: string =
+  typeof __PACKAGE_VERSION__ !== 'undefined'
+    ? __PACKAGE_VERSION__
+    : '0.0.0-test';