@ai-sdk/prodia 1.0.6 → 1.0.8

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @ai-sdk/prodia
 
+## 1.0.8
+
+### Patch Changes
+
+- Updated dependencies [462ad00]
+  - @ai-sdk/provider-utils@4.0.10
+
+## 1.0.7
+
+### Patch Changes
+
+- 4de5a1d: chore: excluded tests from src folder in npm package
+- Updated dependencies [4de5a1d]
+  - @ai-sdk/provider@3.0.5
+  - @ai-sdk/provider-utils@4.0.9
+
 ## 1.0.6
 
 ### Patch Changes

package/dist/index.js

@@ -376,7 +376,7 @@ var prodiaFailedResponseHandler = (0, import_provider_utils.createJsonErrorRespo
 });
 
 // src/version.ts
-var VERSION = true ? "1.0.6" : "0.0.0-test";
+var VERSION = true ? "1.0.8" : "0.0.0-test";
 
 // src/prodia-provider.ts
 var defaultBaseURL = "https://inference.prodia.com/v2";

package/dist/index.mjs

@@ -362,7 +362,7 @@ var prodiaFailedResponseHandler = createJsonErrorResponseHandler({
 });
 
 // src/version.ts
-var VERSION = true ? "1.0.6" : "0.0.0-test";
+var VERSION = true ? "1.0.8" : "0.0.0-test";
 
 // src/prodia-provider.ts
 var defaultBaseURL = "https://inference.prodia.com/v2";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/prodia",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -9,6 +9,10 @@
   "files": [
     "dist/**/*",
     "src",
+    "!src/**/*.test.ts",
+    "!src/**/*.test-d.ts",
+    "!src/**/__snapshots__",
+    "!src/**/__fixtures__",
     "CHANGELOG.md"
   ],
   "exports": {
@@ -20,15 +24,15 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "3.0.4",
-    "@ai-sdk/provider-utils": "4.0.8"
+    "@ai-sdk/provider-utils": "4.0.10",
+    "@ai-sdk/provider": "3.0.5"
   },
   "devDependencies": {
     "@types/node": "20.17.24",
     "tsup": "^8",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "1.0.2",
+    "@ai-sdk/test-server": "1.0.3",
     "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {

package/src/prodia-image-model.test.ts

@@ -1,502 +0,0 @@
-import type { FetchFunction } from '@ai-sdk/provider-utils';
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import { describe, expect, it } from 'vitest';
-import { ProdiaImageModel } from './prodia-image-model';
-
-const prompt = 'A cute baby sea otter';
-
-function createBasicModel({
-  headers,
-  fetch,
-  currentDate,
-}: {
-  headers?: () => Record<string, string | undefined>;
-  fetch?: FetchFunction;
-  currentDate?: () => Date;
-} = {}) {
-  return new ProdiaImageModel('inference.flux-fast.schnell.txt2img.v2', {
-    provider: 'prodia.image',
-    baseURL: 'https://api.example.com/v2',
-    headers: headers ?? (() => ({ Authorization: 'Bearer test-key' })),
-    fetch,
-    _internal: {
-      currentDate,
-    },
-  });
-}
-
-function createMultipartResponse(
-  jobResult: Record<string, unknown>,
-  imageContent: string = 'test-binary-content',
-): { body: Buffer; contentType: string } {
-  const boundary = 'test-boundary-12345';
-  const jobJson = JSON.stringify(jobResult);
-  const imageBuffer = Buffer.from(imageContent);
-
-  const parts = [
-    `--${boundary}\r\n`,
-    'Content-Disposition: form-data; name="job"; filename="job.json"\r\n',
-    'Content-Type: application/json\r\n',
-    '\r\n',
-    jobJson,
-    '\r\n',
-    `--${boundary}\r\n`,
-    'Content-Disposition: form-data; name="output"; filename="output.png"\r\n',
-    'Content-Type: image/png\r\n',
-    '\r\n',
-  ];
-
-  const headerPart = Buffer.from(parts.join(''));
-  const endPart = Buffer.from(`\r\n--${boundary}--\r\n`);
-
-  const body = Buffer.concat([headerPart, imageBuffer, endPart]);
-
-  return {
-    body,
-    contentType: `multipart/form-data; boundary=${boundary}`,
-  };
-}
-
-const defaultJobResult = {
-  id: 'job-123',
-  created_at: '2025-01-01T00:00:00Z',
-  updated_at: '2025-01-01T00:00:05Z',
-  state: { current: 'completed' },
-  config: { prompt, seed: 42 },
-  metrics: { elapsed: 2.5, ips: 10.5 },
-};
-
-describe('ProdiaImageModel', () => {
-  const multipartResponse = createMultipartResponse(defaultJobResult);
-
-  const server = createTestServer({
-    'https://api.example.com/v2/job': {
-      response: {
-        type: 'binary',
-        body: multipartResponse.body,
-        headers: {
-          'content-type': multipartResponse.contentType,
-        },
-      },
-    },
-  });
-
-  describe('doGenerate', () => {
-    it('passes the correct parameters including providerOptions', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: 12345,
-        providerOptions: {
-          prodia: {
-            steps: 4,
-          },
-        },
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        type: 'inference.flux-fast.schnell.txt2img.v2',
-        config: {
-          prompt,
-          seed: 12345,
-          steps: 4,
-        },
-      });
-    });
-
-    it('includes width and height when size is provided', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: '1024x768',
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        type: 'inference.flux-fast.schnell.txt2img.v2',
-        config: {
-          prompt,
-          width: 1024,
-          height: 768,
-        },
-      });
-    });
-
-    it('provider options width/height take precedence over size', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: '1024x768',
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {
-          prodia: {
-            width: 512,
-            height: 512,
-          },
-        },
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        type: 'inference.flux-fast.schnell.txt2img.v2',
-        config: {
-          prompt,
-          width: 512,
-          height: 512,
-        },
-      });
-    });
-
-    it('includes style_preset when stylePreset is provided', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {
-          prodia: {
-            stylePreset: 'anime',
-          },
-        },
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        type: 'inference.flux-fast.schnell.txt2img.v2',
-        config: {
-          prompt,
-          style_preset: 'anime',
-        },
-      });
-    });
-
-    it('includes loras when provided', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {
-          prodia: {
-            loras: ['prodia/lora/flux/anime@v1', 'prodia/lora/flux/realism@v1'],
-          },
-        },
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        type: 'inference.flux-fast.schnell.txt2img.v2',
-        config: {
-          prompt,
-          loras: ['prodia/lora/flux/anime@v1', 'prodia/lora/flux/realism@v1'],
-        },
-      });
-    });
-
-    it('includes progressive when provided', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {
-          prodia: {
-            progressive: true,
-          },
-        },
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        type: 'inference.flux-fast.schnell.txt2img.v2',
-        config: {
-          prompt,
-          progressive: true,
-        },
-      });
-    });
-
-    it('calls the correct endpoint', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(server.calls[0].requestMethod).toBe('POST');
-      expect(server.calls[0].requestUrl).toBe('https://api.example.com/v2/job');
-    });
-
-    it('sends Accept: multipart/form-data header', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(server.calls[0].requestHeaders.accept).toBe(
-        'multipart/form-data; image/png',
-      );
-    });
-
-    it('merges provider and request headers', async () => {
-      const modelWithHeaders = createBasicModel({
-        headers: () => ({
-          'Custom-Provider-Header': 'provider-header-value',
-          Authorization: 'Bearer test-key',
-        }),
-      });
-
-      await modelWithHeaders.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        providerOptions: {},
-        headers: {
-          'Custom-Request-Header': 'request-header-value',
-        },
-        size: undefined,
-        seed: undefined,
-        aspectRatio: undefined,
-      });
-
-      expect(server.calls[0].requestHeaders).toMatchObject({
-        'content-type': 'application/json',
-        'custom-provider-header': 'provider-header-value',
-        'custom-request-header': 'request-header-value',
-        authorization: 'Bearer test-key',
-        accept: 'multipart/form-data; image/png',
-      });
-    });
-
-    it('returns image bytes from multipart response', async () => {
-      const model = createBasicModel();
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        seed: undefined,
-        aspectRatio: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.images).toHaveLength(1);
-      const image = result.images[0];
-      expect(image).toBeInstanceOf(Uint8Array);
-      expect(Buffer.from(image as Uint8Array<ArrayBufferLike>).toString()).toBe(
-        'test-binary-content',
-      );
-    });
-
-    it('returns provider metadata from job result', async () => {
-      const model = createBasicModel();
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        seed: undefined,
-        aspectRatio: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.providerMetadata?.prodia).toStrictEqual({
-        images: [
-          {
-            jobId: 'job-123',
-            seed: 42,
-            elapsed: 2.5,
-            iterationsPerSecond: 10.5,
-            createdAt: '2025-01-01T00:00:00Z',
-            updatedAt: '2025-01-01T00:00:05Z',
-          },
-        ],
-      });
-    });
-
-    it('omits optional metadata fields when not present in job result', async () => {
-      const minimalJobResult = {
-        id: 'job-456',
-        state: { current: 'completed' },
-        config: { prompt },
-      };
-      const response = createMultipartResponse(minimalJobResult);
-
-      server.urls['https://api.example.com/v2/job'].response = {
-        type: 'binary',
-        body: response.body,
-        headers: {
-          'content-type': response.contentType,
-        },
-      };
-
-      const model = createBasicModel();
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        seed: undefined,
-        aspectRatio: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.providerMetadata?.prodia).toStrictEqual({
-        images: [
-          {
-            jobId: 'job-456',
-          },
-        ],
-      });
-    });
-
-    it('warns on invalid size format', async () => {
-      const model = createBasicModel();
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: 'invalid' as `${number}x${number}`,
-        seed: undefined,
-        aspectRatio: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.warnings).toMatchInlineSnapshot(`
-        [
-          {
-            "details": "Invalid size format: invalid. Expected format: WIDTHxHEIGHT (e.g., 1024x1024)",
-            "feature": "size",
-            "type": "unsupported",
-          },
-        ]
-      `);
-    });
-
-    it('handles API errors', async () => {
-      server.urls['https://api.example.com/v2/job'].response = {
-        type: 'error',
-        status: 400,
-        body: JSON.stringify({
-          message: 'Invalid prompt',
-          detail: 'Prompt cannot be empty',
-        }),
-      };
-
-      const model = createBasicModel();
-
-      await expect(
-        model.doGenerate({
-          prompt,
-          files: undefined,
-          mask: undefined,
-          n: 1,
-          providerOptions: {},
-          size: undefined,
-          seed: undefined,
-          aspectRatio: undefined,
-        }),
-      ).rejects.toMatchObject({
-        message: 'Prompt cannot be empty',
-        statusCode: 400,
-        url: 'https://api.example.com/v2/job',
-      });
-    });
-
-    it('includes timestamp, headers, and modelId in response metadata', async () => {
-      const response = createMultipartResponse(defaultJobResult);
-      server.urls['https://api.example.com/v2/job'].response = {
-        type: 'binary',
-        body: response.body,
-        headers: {
-          'content-type': response.contentType,
-        },
-      };
-
-      const testDate = new Date('2025-01-01T00:00:00Z');
-      const model = createBasicModel({
-        currentDate: () => testDate,
-      });
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        providerOptions: {},
-        size: undefined,
-        seed: undefined,
-        aspectRatio: undefined,
-      });
-
-      expect(result.response).toStrictEqual({
-        timestamp: testDate,
-        modelId: 'inference.flux-fast.schnell.txt2img.v2',
-        headers: expect.any(Object),
-      });
-    });
-  });
-
-  describe('constructor', () => {
-    it('exposes correct provider and model information', () => {
-      const model = createBasicModel();
-
-      expect(model.provider).toBe('prodia.image');
-      expect(model.modelId).toBe('inference.flux-fast.schnell.txt2img.v2');
-      expect(model.specificationVersion).toBe('v3');
-      expect(model.maxImagesPerCall).toBe(1);
-    });
-  });
-});

package/src/prodia-provider.test.ts

@@ -1,123 +0,0 @@
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import { describe, expect, it } from 'vitest';
-import { createProdia } from './prodia-provider';
-
-function createMultipartResponse(
-  jobResult: Record<string, unknown>,
-  imageContent: string = 'test-image',
-): { body: Buffer; contentType: string } {
-  const boundary = 'test-boundary-12345';
-  const jobJson = JSON.stringify(jobResult);
-  const imageBuffer = Buffer.from(imageContent);
-
-  const parts = [
-    `--${boundary}\r\n`,
-    'Content-Disposition: form-data; name="job"; filename="job.json"\r\n',
-    'Content-Type: application/json\r\n',
-    '\r\n',
-    jobJson,
-    '\r\n',
-    `--${boundary}\r\n`,
-    'Content-Disposition: form-data; name="output"; filename="output.png"\r\n',
-    'Content-Type: image/png\r\n',
-    '\r\n',
-  ];
-
-  const headerPart = Buffer.from(parts.join(''));
-  const endPart = Buffer.from(`\r\n--${boundary}--\r\n`);
-
-  const body = Buffer.concat([headerPart, imageBuffer, endPart]);
-
-  return {
-    body,
-    contentType: `multipart/form-data; boundary=${boundary}`,
-  };
-}
-
-const defaultJobResult = {
-  id: 'job-123',
-  state: { current: 'completed' },
-  config: { prompt: 'test' },
-};
-
-const multipartResponse = createMultipartResponse(defaultJobResult);
-
-const server = createTestServer({
-  'https://api.example.com/v2/job': {
-    response: {
-      type: 'binary',
-      body: multipartResponse.body,
-      headers: {
-        'content-type': multipartResponse.contentType,
-      },
-    },
-  },
-});
-
-describe('Prodia provider', () => {
-  it('creates image models via .image and .imageModel', () => {
-    const provider = createProdia();
-
-    const imageModel = provider.image('inference.flux-fast.schnell.txt2img.v2');
-    const imageModel2 = provider.imageModel(
-      'inference.flux.schnell.txt2img.v2',
-    );
-
-    expect(imageModel.provider).toBe('prodia.image');
-    expect(imageModel.modelId).toBe('inference.flux-fast.schnell.txt2img.v2');
-    expect(imageModel2.modelId).toBe('inference.flux.schnell.txt2img.v2');
-    expect(imageModel.specificationVersion).toBe('v3');
-  });
-
-  it('configures baseURL and headers correctly', async () => {
-    const provider = createProdia({
-      apiKey: 'test-api-key',
-      baseURL: 'https://api.example.com/v2',
-      headers: {
-        'x-extra-header': 'extra',
-      },
-    });
-
-    const model = provider.image('inference.flux-fast.schnell.txt2img.v2');
-
-    await model.doGenerate({
-      prompt: 'A serene mountain landscape at sunset',
-      files: undefined,
-      mask: undefined,
-      n: 1,
-      size: undefined,
-      seed: undefined,
-      aspectRatio: undefined,
-      providerOptions: {},
-    });
-
-    expect(server.calls[0].requestUrl).toBe('https://api.example.com/v2/job');
-    expect(server.calls[0].requestMethod).toBe('POST');
-    expect(server.calls[0].requestHeaders.authorization).toBe(
-      'Bearer test-api-key',
-    );
-    expect(server.calls[0].requestHeaders['x-extra-header']).toBe('extra');
-    expect(server.calls[0].requestHeaders.accept).toBe(
-      'multipart/form-data; image/png',
-    );
-    expect(await server.calls[0].requestBodyJson).toMatchObject({
-      type: 'inference.flux-fast.schnell.txt2img.v2',
-      config: {
-        prompt: 'A serene mountain landscape at sunset',
-      },
-    });
-
-    expect(server.calls[0].requestUserAgent).toContain('ai-sdk/prodia/');
-  });
-
-  it('throws NoSuchModelError for unsupported model types', () => {
-    const provider = createProdia();
-
-    expect(() => provider.languageModel('some-id')).toThrowError(
-      'No such languageModel',
-    );
-    expect(() => provider.embeddingModel('some-id')).toThrowError(
-      'No such embeddingModel',
-    );
-  });
-});