@ai-sdk/deepinfra 2.0.18 → 2.0.19

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # @ai-sdk/deepinfra
 
+## 2.0.19
+
+### Patch Changes
+
+- 4de5a1d: chore: excluded tests from src folder in npm package
+- Updated dependencies [4de5a1d]
+  - @ai-sdk/openai-compatible@2.0.18
+  - @ai-sdk/provider@3.0.5
+  - @ai-sdk/provider-utils@4.0.9
+
 ## 2.0.18
 
 ### Patch Changes

package/dist/index.js

@@ -161,7 +161,7 @@ async function fileToBlob(file) {
 }
 
 // src/version.ts
-var VERSION = true ? "2.0.18" : "0.0.0-test";
+var VERSION = true ? "2.0.19" : "0.0.0-test";
 
 // src/deepinfra-provider.ts
 function createDeepInfra(options = {}) {

package/dist/index.mjs

@@ -150,7 +150,7 @@ async function fileToBlob(file) {
 }
 
 // src/version.ts
-var VERSION = true ? "2.0.18" : "0.0.0-test";
+var VERSION = true ? "2.0.19" : "0.0.0-test";
 
 // src/deepinfra-provider.ts
 function createDeepInfra(options = {}) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/deepinfra",
-  "version": "2.0.18",
+  "version": "2.0.19",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -10,6 +10,10 @@
     "dist/**/*",
     "docs/**/*",
     "src",
+    "!src/**/*.test.ts",
+    "!src/**/*.test-d.ts",
+    "!src/**/__snapshots__",
+    "!src/**/__fixtures__",
     "CHANGELOG.md",
     "README.md"
   ],
@@ -25,16 +29,16 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/openai-compatible": "2.0.17",
-    "@ai-sdk/provider": "3.0.4",
-    "@ai-sdk/provider-utils": "4.0.8"
+    "@ai-sdk/openai-compatible": "2.0.18",
+    "@ai-sdk/provider": "3.0.5",
+    "@ai-sdk/provider-utils": "4.0.9"
   },
   "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/deepinfra-image-model.test.ts

@@ -1,431 +0,0 @@
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import { describe, expect, it } from 'vitest';
-import { DeepInfraImageModel } from './deepinfra-image-model';
-import { FetchFunction } from '@ai-sdk/provider-utils';
-
-const prompt = 'A cute baby sea otter';
-
-function createBasicModel({
-  headers,
-  fetch,
-  currentDate,
-}: {
-  headers?: () => Record<string, string>;
-  fetch?: FetchFunction;
-  currentDate?: () => Date;
-} = {}) {
-  return new DeepInfraImageModel('stability-ai/sdxl', {
-    provider: 'deepinfra',
-    baseURL: 'https://api.example.com',
-    headers: headers ?? (() => ({ 'api-key': 'test-key' })),
-    fetch,
-    _internal: {
-      currentDate,
-    },
-  });
-}
-
-describe('DeepInfraImageModel', () => {
-  const testDate = new Date('2024-01-01T00:00:00Z');
-  const server = createTestServer({
-    'https://api.example.com/*': {
-      response: {
-        type: 'json-value',
-        body: {
-          images: ['data:image/png;base64,test-image-data'],
-        },
-      },
-    },
-  });
-
-  describe('doGenerate', () => {
-    it('should pass the correct parameters including aspect ratio and seed', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: '16:9',
-        seed: 42,
-        providerOptions: { deepinfra: { additional_param: 'value' } },
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        prompt,
-        aspect_ratio: '16:9',
-        seed: 42,
-        num_images: 1,
-        additional_param: 'value',
-      });
-    });
-
-    it('should call the correct url', 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).toStrictEqual('POST');
-      expect(server.calls[0].requestUrl).toStrictEqual(
-        'https://api.example.com/stability-ai/sdxl',
-      );
-    });
-
-    it('should pass headers', async () => {
-      const modelWithHeaders = createBasicModel({
-        headers: () => ({
-          'Custom-Provider-Header': 'provider-header-value',
-        }),
-      });
-
-      await modelWithHeaders.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-        headers: {
-          'Custom-Request-Header': 'request-header-value',
-        },
-      });
-
-      expect(server.calls[0].requestHeaders).toStrictEqual({
-        'content-type': 'application/json',
-        'custom-provider-header': 'provider-header-value',
-        'custom-request-header': 'request-header-value',
-      });
-    });
-
-    it('should handle API errors', async () => {
-      server.urls['https://api.example.com/*'].response = {
-        type: 'error',
-        status: 400,
-        body: JSON.stringify({
-          error: {
-            message: 'Bad Request',
-          },
-        }),
-      };
-
-      const model = createBasicModel();
-      await expect(
-        model.doGenerate({
-          prompt,
-          files: undefined,
-          mask: undefined,
-          n: 1,
-          size: undefined,
-          aspectRatio: undefined,
-          seed: undefined,
-          providerOptions: {},
-        }),
-      ).rejects.toThrow('Bad Request');
-    });
-
-    it('should handle size parameter', async () => {
-      const model = createBasicModel();
-
-      await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: '1024x768',
-        aspectRatio: undefined,
-        seed: 42,
-        providerOptions: {},
-      });
-
-      expect(await server.calls[0].requestBodyJson).toStrictEqual({
-        prompt,
-        width: '1024',
-        height: '768',
-        seed: 42,
-        num_images: 1,
-      });
-    });
-
-    it('should respect the abort signal', async () => {
-      const model = createBasicModel();
-      const controller = new AbortController();
-
-      const generatePromise = model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-        abortSignal: controller.signal,
-      });
-
-      controller.abort();
-
-      await expect(generatePromise).rejects.toThrow(
-        'This operation was aborted',
-      );
-    });
-
-    describe('response metadata', () => {
-      it('should include timestamp, headers and modelId in response', async () => {
-        const model = createBasicModel({
-          currentDate: () => testDate,
-        });
-
-        const result = await model.doGenerate({
-          prompt,
-          files: undefined,
-          mask: undefined,
-          n: 1,
-          size: undefined,
-          aspectRatio: undefined,
-          seed: undefined,
-          providerOptions: {},
-        });
-
-        expect(result.response).toStrictEqual({
-          timestamp: testDate,
-          modelId: 'stability-ai/sdxl',
-          headers: expect.any(Object),
-        });
-      });
-
-      it('should include response headers from API call', async () => {
-        server.urls['https://api.example.com/*'].response = {
-          type: 'json-value',
-          headers: {
-            'x-request-id': 'test-request-id',
-          },
-          body: {
-            images: ['data:image/png;base64,test-image-data'],
-          },
-        };
-
-        const model = createBasicModel();
-        const result = await model.doGenerate({
-          prompt,
-          files: undefined,
-          mask: undefined,
-          n: 1,
-          size: undefined,
-          aspectRatio: undefined,
-          seed: undefined,
-          providerOptions: {},
-        });
-
-        expect(result.response.headers).toStrictEqual({
-          'content-length': '52',
-          'x-request-id': 'test-request-id',
-          'content-type': 'application/json',
-        });
-      });
-    });
-  });
-
-  describe('constructor', () => {
-    it('should expose correct provider and model information', () => {
-      const model = createBasicModel();
-
-      expect(model.provider).toBe('deepinfra');
-      expect(model.modelId).toBe('stability-ai/sdxl');
-      expect(model.specificationVersion).toBe('v3');
-      expect(model.maxImagesPerCall).toBe(1);
-    });
-  });
-
-  describe('Image Editing', () => {
-    const editServer = createTestServer({
-      'https://edit.example.com/openai/images/edits': {
-        response: {
-          type: 'json-value',
-          body: {
-            created: 1234567890,
-            data: [{ b64_json: 'edited-image-base64' }],
-          },
-        },
-      },
-    });
-
-    // Model with baseURL that will resolve to edit endpoint
-    const editModel = new DeepInfraImageModel(
-      'black-forest-labs/FLUX.1-Kontext-dev',
-      {
-        provider: 'deepinfra',
-        baseURL: 'https://edit.example.com/inference',
-        headers: () => ({ 'api-key': 'test-key' }),
-      },
-    );
-
-    it('should send edit request with files', async () => {
-      const imageData = new Uint8Array([137, 80, 78, 71]); // PNG magic bytes
-
-      const result = await editModel.doGenerate({
-        prompt: 'Turn the cat into a dog',
-        files: [
-          {
-            type: 'file',
-            data: imageData,
-            mediaType: 'image/png',
-          },
-        ],
-        mask: undefined,
-        n: 1,
-        size: '1024x1024',
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.images).toStrictEqual(['edited-image-base64']);
-      expect(editServer.calls[0].requestUrl).toBe(
-        'https://edit.example.com/openai/images/edits',
-      );
-    });
-
-    it('should send edit request with files and mask', async () => {
-      const imageData = new Uint8Array([137, 80, 78, 71]);
-      const maskData = new Uint8Array([255, 255, 255, 0]);
-
-      const result = await editModel.doGenerate({
-        prompt: 'Add a flamingo to the pool',
-        files: [
-          {
-            type: 'file',
-            data: imageData,
-            mediaType: 'image/png',
-          },
-        ],
-        mask: {
-          type: 'file',
-          data: maskData,
-          mediaType: 'image/png',
-        },
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.images).toStrictEqual(['edited-image-base64']);
-      expect(editServer.calls[0].requestUrl).toBe(
-        'https://edit.example.com/openai/images/edits',
-      );
-    });
-
-    it('should send edit request with multiple images', async () => {
-      const image1 = new Uint8Array([137, 80, 78, 71]);
-      const image2 = new Uint8Array([137, 80, 78, 71]);
-
-      const result = await editModel.doGenerate({
-        prompt: 'Combine these images',
-        files: [
-          {
-            type: 'file',
-            data: image1,
-            mediaType: 'image/png',
-          },
-          {
-            type: 'file',
-            data: image2,
-            mediaType: 'image/png',
-          },
-        ],
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.images).toStrictEqual(['edited-image-base64']);
-    });
-
-    it('should include response metadata for edit requests', async () => {
-      const testDate = new Date('2024-01-01T00:00:00Z');
-      const modelWithDate = new DeepInfraImageModel(
-        'black-forest-labs/FLUX.1-Kontext-dev',
-        {
-          provider: 'deepinfra',
-          baseURL: 'https://edit.example.com/inference',
-          headers: () => ({ 'api-key': 'test-key' }),
-          _internal: {
-            currentDate: () => testDate,
-          },
-        },
-      );
-
-      const imageData = new Uint8Array([137, 80, 78, 71]);
-
-      const result = await modelWithDate.doGenerate({
-        prompt: 'Edit this image',
-        files: [
-          {
-            type: 'file',
-            data: imageData,
-            mediaType: 'image/png',
-          },
-        ],
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      });
-
-      expect(result.response).toStrictEqual({
-        timestamp: testDate,
-        modelId: 'black-forest-labs/FLUX.1-Kontext-dev',
-        headers: expect.any(Object),
-      });
-    });
-
-    it('should pass provider options in edit request', async () => {
-      const imageData = new Uint8Array([137, 80, 78, 71]);
-
-      await editModel.doGenerate({
-        prompt: 'Edit with custom options',
-        files: [
-          {
-            type: 'file',
-            data: imageData,
-            mediaType: 'image/png',
-          },
-        ],
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {
-          deepinfra: {
-            guidance: 7.5,
-          },
-        },
-      });
-
-      // The request should have been made to the edit endpoint
-      expect(editServer.calls[0].requestUrl).toBe(
-        'https://edit.example.com/openai/images/edits',
-      );
-    });
-  });
-});

package/src/deepinfra-provider.test.ts

@@ -1,184 +0,0 @@
-import { DeepInfraImageModel } from './deepinfra-image-model';
-import { createDeepInfra } from './deepinfra-provider';
-import {
-  OpenAICompatibleChatLanguageModel,
-  OpenAICompatibleCompletionLanguageModel,
-  OpenAICompatibleEmbeddingModel,
-} from '@ai-sdk/openai-compatible';
-import { LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
-import { loadApiKey } from '@ai-sdk/provider-utils';
-import { describe, it, expect, vi, beforeEach, Mock } from 'vitest';
-
-// Add type assertion for the mocked class
-const OpenAICompatibleChatLanguageModelMock =
-  OpenAICompatibleChatLanguageModel as unknown as Mock;
-
-vi.mock('@ai-sdk/openai-compatible', () => ({
-  OpenAICompatibleChatLanguageModel: vi.fn(),
-  OpenAICompatibleCompletionLanguageModel: vi.fn(),
-  OpenAICompatibleEmbeddingModel: vi.fn(),
-}));
-
-vi.mock('@ai-sdk/provider-utils', async () => {
-  const actual = await vi.importActual('@ai-sdk/provider-utils');
-  return {
-    ...actual,
-    loadApiKey: vi.fn().mockReturnValue('mock-api-key'),
-    withoutTrailingSlash: vi.fn(url => url),
-  };
-});
-
-vi.mock('./deepinfra-image-model', () => ({
-  DeepInfraImageModel: vi.fn(),
-}));
-
-describe('DeepInfraProvider', () => {
-  let mockLanguageModel: LanguageModelV3;
-  let mockEmbeddingModel: EmbeddingModelV3;
-
-  beforeEach(() => {
-    // Mock implementations of models
-    mockLanguageModel = {
-      // Add any required methods for LanguageModelV3
-    } as LanguageModelV3;
-    mockEmbeddingModel = {
-      // Add any required methods for EmbeddingModelV3
-    } as EmbeddingModelV3;
-
-    // Reset mocks
-    vi.clearAllMocks();
-  });
-
-  describe('createDeepInfra', () => {
-    it('should create a DeepInfraProvider instance with default options', () => {
-      const provider = createDeepInfra();
-      const model = provider('model-id');
-
-      // Use the mocked version
-      const constructorCall =
-        OpenAICompatibleChatLanguageModelMock.mock.calls[0];
-      const config = constructorCall[1];
-      config.headers();
-
-      expect(loadApiKey).toHaveBeenCalledWith({
-        apiKey: undefined,
-        environmentVariableName: 'DEEPINFRA_API_KEY',
-        description: "DeepInfra's API key",
-      });
-    });
-
-    it('should create a DeepInfraProvider instance with custom options', () => {
-      const options = {
-        apiKey: 'custom-key',
-        baseURL: 'https://custom.url',
-        headers: { 'Custom-Header': 'value' },
-      };
-      const provider = createDeepInfra(options);
-      const model = provider('model-id');
-
-      const constructorCall =
-        OpenAICompatibleChatLanguageModelMock.mock.calls[0];
-      const config = constructorCall[1];
-      config.headers();
-
-      expect(loadApiKey).toHaveBeenCalledWith({
-        apiKey: 'custom-key',
-        environmentVariableName: 'DEEPINFRA_API_KEY',
-        description: "DeepInfra's API key",
-      });
-    });
-
-    it('should return a chat model when called as a function', () => {
-      const provider = createDeepInfra();
-      const modelId = 'foo-model-id';
-
-      const model = provider(modelId);
-      expect(model).toBeInstanceOf(OpenAICompatibleChatLanguageModel);
-    });
-  });
-
-  describe('chatModel', () => {
-    it('should construct a chat model with correct configuration', () => {
-      const provider = createDeepInfra();
-      const modelId = 'deepinfra-chat-model';
-
-      const model = provider.chatModel(modelId);
-
-      expect(model).toBeInstanceOf(OpenAICompatibleChatLanguageModel);
-      expect(OpenAICompatibleChatLanguageModelMock).toHaveBeenCalledWith(
-        modelId,
-        expect.objectContaining({
-          provider: 'deepinfra.chat',
-        }),
-      );
-    });
-  });
-
-  describe('completionModel', () => {
-    it('should construct a completion model with correct configuration', () => {
-      const provider = createDeepInfra();
-      const modelId = 'deepinfra-completion-model';
-
-      const model = provider.completionModel(modelId);
-
-      expect(model).toBeInstanceOf(OpenAICompatibleCompletionLanguageModel);
-    });
-  });
-
-  describe('embeddingModel', () => {
-    it('should construct a text embedding model with correct configuration', () => {
-      const provider = createDeepInfra();
-      const modelId = 'deepinfra-embedding-model';
-
-      const model = provider.embeddingModel(modelId);
-
-      expect(model).toBeInstanceOf(OpenAICompatibleEmbeddingModel);
-    });
-  });
-
-  describe('image', () => {
-    it('should construct an image model with correct configuration', () => {
-      const provider = createDeepInfra();
-      const modelId = 'deepinfra-image-model';
-
-      const model = provider.image(modelId);
-
-      expect(model).toBeInstanceOf(DeepInfraImageModel);
-      expect(DeepInfraImageModel).toHaveBeenCalledWith(
-        modelId,
-        expect.objectContaining({
-          provider: 'deepinfra.image',
-          baseURL: 'https://api.deepinfra.com/v1/inference',
-        }),
-      );
-    });
-
-    it('should use default settings when none provided', () => {
-      const provider = createDeepInfra();
-      const modelId = 'deepinfra-image-model';
-
-      const model = provider.image(modelId);
-
-      expect(model).toBeInstanceOf(DeepInfraImageModel);
-      expect(DeepInfraImageModel).toHaveBeenCalledWith(
-        modelId,
-        expect.any(Object),
-      );
-    });
-
-    it('should respect custom baseURL', () => {
-      const customBaseURL = 'https://custom.api.deepinfra.com';
-      const provider = createDeepInfra({ baseURL: customBaseURL });
-      const modelId = 'deepinfra-image-model';
-
-      const model = provider.image(modelId);
-
-      expect(DeepInfraImageModel).toHaveBeenCalledWith(
-        modelId,
-        expect.objectContaining({
-          baseURL: `${customBaseURL}/inference`,
-        }),
-      );
-    });
-  });
-});