@ai-sdk/togetherai 2.0.19 → 2.0.20

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # @ai-sdk/togetherai
 
+## 2.0.20
+
+### 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.19
 
 ### Patch Changes

package/dist/index.js

@@ -272,7 +272,7 @@ var togetheraiImageProviderOptionsSchema = (0, import_provider_utils4.lazySchema
 );
 
 // src/version.ts
-var VERSION = true ? "2.0.19" : "0.0.0-test";
+var VERSION = true ? "2.0.20" : "0.0.0-test";
 
 // src/togetherai-provider.ts
 function createTogetherAI(options = {}) {

package/dist/index.mjs

@@ -267,7 +267,7 @@ var togetheraiImageProviderOptionsSchema = lazySchema3(
 );
 
 // src/version.ts
-var VERSION = true ? "2.0.19" : "0.0.0-test";
+var VERSION = true ? "2.0.20" : "0.0.0-test";
 
 // src/togetherai-provider.ts
 function createTogetherAI(options = {}) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/togetherai",
-  "version": "2.0.19",
+  "version": "2.0.20",
   "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/reranking/__fixtures__/togetherai-reranking.1.json

@@ -1,22 +0,0 @@
-{
-  "id": "oGs6Zt9-62bZhn-99529372487b1b0a",
-  "object": "rerank",
-  "model": "Salesforce/Llama-Rank-v1",
-  "results": [
-    {
-      "index": 0,
-      "relevance_score": 0.6475887154399037,
-      "document": {}
-    },
-    {
-      "index": 5,
-      "relevance_score": 0.6323295373206566,
-      "document": {}
-    }
-  ],
-  "usage": {
-    "prompt_tokens": 2966,
-    "completion_tokens": 0,
-    "total_tokens": 2966
-  }
-}

package/src/reranking/togetherai-reranking-model.test.ts

@@ -1,245 +0,0 @@
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import fs from 'node:fs';
-import { beforeEach, describe, expect, it } from 'vitest';
-import { createTogetherAI } from '../togetherai-provider';
-import { TogetherAIRerankingOptions } from './togetherai-reranking-options';
-
-const provider = createTogetherAI({ apiKey: 'test-api-key' });
-const model = provider.rerankingModel('Salesforce/Llama-Rank-v1');
-
-describe('doRerank', () => {
-  const server = createTestServer({
-    'https://api.together.xyz/v1/rerank': {},
-  });
-
-  function prepareJsonFixtureResponse(filename: string) {
-    server.urls['https://api.together.xyz/v1/rerank'].response = {
-      type: 'json-value',
-      body: JSON.parse(
-        fs.readFileSync(`src/reranking/__fixtures__/${filename}.json`, 'utf8'),
-      ),
-    };
-    return;
-  }
-
-  describe('json documents', () => {
-    let result: Awaited<ReturnType<typeof model.doRerank>>;
-
-    beforeEach(async () => {
-      prepareJsonFixtureResponse('togetherai-reranking.1');
-
-      result = await model.doRerank({
-        documents: {
-          type: 'object',
-          values: [
-            { example: 'sunny day at the beach' },
-            { example: 'rainy day in the city' },
-          ],
-        },
-        query: 'rainy day',
-        topN: 2,
-        providerOptions: {
-          togetherai: {
-            rankFields: ['example'],
-          } satisfies TogetherAIRerankingOptions,
-        },
-      });
-    });
-
-    it('should send request with stringified json documents', async () => {
-      expect(await server.calls[0].requestBodyJson).toMatchInlineSnapshot(`
-        {
-          "documents": [
-            {
-              "example": "sunny day at the beach",
-            },
-            {
-              "example": "rainy day in the city",
-            },
-          ],
-          "model": "Salesforce/Llama-Rank-v1",
-          "query": "rainy day",
-          "rank_fields": [
-            "example",
-          ],
-          "return_documents": false,
-          "top_n": 2,
-        }
-      `);
-    });
-
-    it('should send request with the correct headers', async () => {
-      expect(server.calls[0].requestHeaders).toMatchInlineSnapshot(`
-        {
-          "authorization": "Bearer test-api-key",
-          "content-type": "application/json",
-        }
-      `);
-    });
-
-    it('should return result with warnings', async () => {
-      expect(result.warnings).toMatchInlineSnapshot(`undefined`);
-    });
-
-    it('should return result with the correct ranking', async () => {
-      expect(result.ranking).toMatchInlineSnapshot(`
-        [
-          {
-            "index": 0,
-            "relevanceScore": 0.6475887154399037,
-          },
-          {
-            "index": 5,
-            "relevanceScore": 0.6323295373206566,
-          },
-        ]
-      `);
-    });
-
-    it('should not return provider metadata (use response body instead)', async () => {
-      expect(result.providerMetadata).toMatchInlineSnapshot(`undefined`);
-    });
-
-    it('should return result with the correct response', async () => {
-      expect(result.response).toMatchInlineSnapshot(`
-        {
-          "body": {
-            "id": "oGs6Zt9-62bZhn-99529372487b1b0a",
-            "model": "Salesforce/Llama-Rank-v1",
-            "object": "rerank",
-            "results": [
-              {
-                "document": {},
-                "index": 0,
-                "relevance_score": 0.6475887154399037,
-              },
-              {
-                "document": {},
-                "index": 5,
-                "relevance_score": 0.6323295373206566,
-              },
-            ],
-            "usage": {
-              "completion_tokens": 0,
-              "prompt_tokens": 2966,
-              "total_tokens": 2966,
-            },
-          },
-          "headers": {
-            "content-length": "304",
-            "content-type": "application/json",
-          },
-          "id": "oGs6Zt9-62bZhn-99529372487b1b0a",
-          "modelId": "Salesforce/Llama-Rank-v1",
-        }
-      `);
-    });
-  });
-
-  describe('text documents', () => {
-    let result: Awaited<ReturnType<typeof model.doRerank>>;
-
-    beforeEach(async () => {
-      prepareJsonFixtureResponse('togetherai-reranking.1');
-
-      result = await model.doRerank({
-        documents: {
-          type: 'text',
-          values: ['sunny day at the beach', 'rainy day in the city'],
-        },
-        query: 'rainy day',
-        topN: 2,
-        providerOptions: {
-          togetherai: {
-            rankFields: ['example'],
-          } satisfies TogetherAIRerankingOptions,
-        },
-      });
-    });
-
-    it('should send request with text documents', async () => {
-      expect(await server.calls[0].requestBodyJson).toMatchInlineSnapshot(`
-        {
-          "documents": [
-            "sunny day at the beach",
-            "rainy day in the city",
-          ],
-          "model": "Salesforce/Llama-Rank-v1",
-          "query": "rainy day",
-          "rank_fields": [
-            "example",
-          ],
-          "return_documents": false,
-          "top_n": 2,
-        }
-      `);
-    });
-
-    it('should send request with the correct headers', async () => {
-      expect(server.calls[0].requestHeaders).toMatchInlineSnapshot(`
-        {
-          "authorization": "Bearer test-api-key",
-          "content-type": "application/json",
-        }
-      `);
-    });
-
-    it('should return result without warnings', async () => {
-      expect(result.warnings).toMatchInlineSnapshot(`undefined`);
-    });
-
-    it('should return result with the correct ranking', async () => {
-      expect(result.ranking).toMatchInlineSnapshot(`
-        [
-          {
-            "index": 0,
-            "relevanceScore": 0.6475887154399037,
-          },
-          {
-            "index": 5,
-            "relevanceScore": 0.6323295373206566,
-          },
-        ]
-      `);
-    });
-
-    it('should not return provider metadata (use response body instead)', async () => {
-      expect(result.providerMetadata).toMatchInlineSnapshot(`undefined`);
-    });
-
-    it('should return result with the correct response', async () => {
-      expect(result.response).toMatchInlineSnapshot(`
-        {
-          "body": {
-            "id": "oGs6Zt9-62bZhn-99529372487b1b0a",
-            "model": "Salesforce/Llama-Rank-v1",
-            "object": "rerank",
-            "results": [
-              {
-                "document": {},
-                "index": 0,
-                "relevance_score": 0.6475887154399037,
-              },
-              {
-                "document": {},
-                "index": 5,
-                "relevance_score": 0.6323295373206566,
-              },
-            ],
-            "usage": {
-              "completion_tokens": 0,
-              "prompt_tokens": 2966,
-              "total_tokens": 2966,
-            },
-          },
-          "headers": {
-            "content-length": "304",
-            "content-type": "application/json",
-          },
-          "id": "oGs6Zt9-62bZhn-99529372487b1b0a",
-          "modelId": "Salesforce/Llama-Rank-v1",
-        }
-      `);
-    });
-  });
-});

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

@@ -1,488 +0,0 @@
-import { FetchFunction } from '@ai-sdk/provider-utils';
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import { describe, expect, it } from 'vitest';
-import { TogetherAIImageModel } from './togetherai-image-model';
-
-const prompt = 'A cute baby sea otter';
-
-function createBasicModel({
-  headers,
-  fetch,
-  currentDate,
-}: {
-  headers?: () => Record<string, string>;
-  fetch?: FetchFunction;
-  currentDate?: () => Date;
-} = {}) {
-  return new TogetherAIImageModel('stabilityai/stable-diffusion-xl', {
-    provider: 'togetherai',
-    baseURL: 'https://api.example.com',
-    headers: headers ?? (() => ({ 'api-key': 'test-key' })),
-    fetch,
-    _internal: {
-      currentDate,
-    },
-  });
-}
-
-const server = createTestServer({
-  'https://api.example.com/*': {
-    response: {
-      type: 'json-value',
-      body: {
-        id: 'test-id',
-        data: [{ index: 0, b64_json: 'test-base64-content' }],
-        model: 'stabilityai/stable-diffusion-xl',
-        object: 'list',
-      },
-    },
-  },
-});
-
-describe('doGenerate', () => {
-  it('should pass the correct parameters including size and seed', async () => {
-    const model = createBasicModel();
-
-    await model.doGenerate({
-      prompt,
-      files: undefined,
-      mask: undefined,
-      n: 1,
-      size: '1024x1024',
-      seed: 42,
-      providerOptions: { togetherai: { additional_param: 'value' } },
-      aspectRatio: undefined,
-    });
-
-    expect(await server.calls[0].requestBodyJson).toStrictEqual({
-      model: 'stabilityai/stable-diffusion-xl',
-      prompt,
-      seed: 42,
-      width: 1024,
-      height: 1024,
-      response_format: 'base64',
-      additional_param: 'value',
-    });
-  });
-
-  it('should include n parameter when requesting multiple images', async () => {
-    const model = createBasicModel();
-
-    await model.doGenerate({
-      prompt,
-      files: undefined,
-      mask: undefined,
-      n: 3,
-      size: '1024x1024',
-      seed: 42,
-      providerOptions: {},
-      aspectRatio: undefined,
-    });
-
-    expect(await server.calls[0].requestBodyJson).toStrictEqual({
-      model: 'stabilityai/stable-diffusion-xl',
-      prompt,
-      seed: 42,
-      n: 3,
-      width: 1024,
-      height: 1024,
-      response_format: 'base64',
-    });
-  });
-
-  it('should call the correct url', async () => {
-    const model = createBasicModel();
-
-    await model.doGenerate({
-      prompt,
-      files: undefined,
-      mask: undefined,
-      n: 1,
-      size: '1024x1024',
-      seed: 42,
-      providerOptions: {},
-      aspectRatio: undefined,
-    });
-
-    expect(server.calls[0].requestMethod).toStrictEqual('POST');
-    expect(server.calls[0].requestUrl).toStrictEqual(
-      'https://api.example.com/images/generations',
-    );
-  });
-
-  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,
-      seed: undefined,
-      providerOptions: {},
-      aspectRatio: undefined,
-      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,
-        seed: undefined,
-        providerOptions: {},
-        aspectRatio: undefined,
-      }),
-    ).rejects.toMatchObject({
-      message: 'Bad Request',
-    });
-  });
-
-  describe('warnings', () => {
-    it('should return aspectRatio warning when aspectRatio is provided', async () => {
-      const model = createBasicModel();
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: '1024x1024',
-        aspectRatio: '1:1',
-        seed: 123,
-        providerOptions: {},
-      });
-
-      expect(result.warnings).toMatchInlineSnapshot(`
-        [
-          {
-            "details": "This model does not support the \`aspectRatio\` option. Use \`size\` instead.",
-            "feature": "aspectRatio",
-            "type": "unsupported",
-          },
-        ]
-      `);
-    });
-  });
-
-  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,
-      seed: undefined,
-      providerOptions: {},
-      aspectRatio: undefined,
-      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 testDate = new Date('2024-01-01T00:00:00Z');
-      const model = createBasicModel({
-        currentDate: () => testDate,
-      });
-
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        seed: undefined,
-        providerOptions: {},
-        aspectRatio: undefined,
-      });
-
-      expect(result.response).toStrictEqual({
-        timestamp: testDate,
-        modelId: 'stabilityai/stable-diffusion-xl',
-        headers: expect.any(Object),
-      });
-    });
-
-    it('should include response headers from API call', async () => {
-      server.urls['https://api.example.com/*'].response = {
-        type: 'json-value',
-        body: {
-          id: 'test-id',
-          data: [{ index: 0, b64_json: 'test-base64-content' }],
-          model: 'stabilityai/stable-diffusion-xl',
-          object: 'list',
-        },
-        headers: {
-          'x-request-id': 'test-request-id',
-          'content-length': '128',
-        },
-      };
-
-      const model = createBasicModel();
-      const result = await model.doGenerate({
-        prompt,
-        files: undefined,
-        mask: undefined,
-        n: 1,
-        size: undefined,
-        seed: undefined,
-        providerOptions: {},
-        aspectRatio: undefined,
-      });
-
-      expect(result.response.headers).toStrictEqual({
-        'x-request-id': 'test-request-id',
-        'content-type': 'application/json',
-        'content-length': '128',
-      });
-    });
-  });
-});
-
-describe('constructor', () => {
-  it('should expose correct provider and model information', () => {
-    const model = createBasicModel();
-
-    expect(model.provider).toBe('togetherai');
-    expect(model.modelId).toBe('stabilityai/stable-diffusion-xl');
-    expect(model.specificationVersion).toBe('v3');
-    expect(model.maxImagesPerCall).toBe(1);
-  });
-});
-
-describe('Image Editing', () => {
-  const server = createTestServer({
-    'https://api.example.com/*': {
-      response: {
-        type: 'json-value',
-        body: {
-          id: 'test-id',
-          data: [{ index: 0, b64_json: 'test-base64-content' }],
-          model: 'black-forest-labs/FLUX.1-kontext-pro',
-          object: 'list',
-        },
-      },
-    },
-  });
-
-  it('should send image_url when URL file is provided', async () => {
-    const model = createBasicModel();
-
-    await model.doGenerate({
-      prompt: 'Make the shirt yellow',
-      files: [
-        {
-          type: 'url',
-          url: 'https://example.com/input.jpg',
-        },
-      ],
-      mask: undefined,
-      n: 1,
-      size: undefined,
-      aspectRatio: undefined,
-      seed: undefined,
-      providerOptions: {},
-    });
-
-    const requestBody = await server.calls[0].requestBodyJson;
-    expect(requestBody).toMatchInlineSnapshot(`
-      {
-        "image_url": "https://example.com/input.jpg",
-        "model": "stabilityai/stable-diffusion-xl",
-        "prompt": "Make the shirt yellow",
-        "response_format": "base64",
-      }
-    `);
-  });
-
-  it('should convert Uint8Array file to data URI', async () => {
-    const model = createBasicModel();
-    const testImageData = new Uint8Array([137, 80, 78, 71, 13, 10, 26, 10]);
-
-    await model.doGenerate({
-      prompt: 'Transform this image',
-      files: [
-        {
-          type: 'file',
-          data: testImageData,
-          mediaType: 'image/png',
-        },
-      ],
-      mask: undefined,
-      n: 1,
-      size: undefined,
-      aspectRatio: undefined,
-      seed: undefined,
-      providerOptions: {},
-    });
-
-    const requestBody = await server.calls[0].requestBodyJson;
-    expect(requestBody.image_url).toMatch(/^data:image\/png;base64,/);
-    expect(requestBody.prompt).toBe('Transform this image');
-  });
-
-  it('should convert file with base64 string data to data URI', async () => {
-    const model = createBasicModel();
-
-    await model.doGenerate({
-      prompt: 'Edit this',
-      files: [
-        {
-          type: 'file',
-          data: 'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==',
-          mediaType: 'image/png',
-        },
-      ],
-      mask: undefined,
-      n: 1,
-      size: undefined,
-      aspectRatio: undefined,
-      seed: undefined,
-      providerOptions: {},
-    });
-
-    const requestBody = await server.calls[0].requestBodyJson;
-    expect(requestBody.image_url).toBe(
-      'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==',
-    );
-  });
-
-  it('should throw error when mask is provided', async () => {
-    const model = createBasicModel();
-
-    await expect(
-      model.doGenerate({
-        prompt: 'Inpaint this area',
-        files: [
-          {
-            type: 'url',
-            url: 'https://example.com/input.jpg',
-          },
-        ],
-        mask: {
-          type: 'url',
-          url: 'https://example.com/mask.png',
-        },
-        n: 1,
-        size: undefined,
-        aspectRatio: undefined,
-        seed: undefined,
-        providerOptions: {},
-      }),
-    ).rejects.toThrow(
-      'Together AI does not support mask-based image editing. ' +
-        'Use FLUX Kontext models (e.g., black-forest-labs/FLUX.1-kontext-pro) ' +
-        'with a reference image and descriptive prompt instead.',
-    );
-  });
-
-  it('should warn when multiple files are provided', async () => {
-    const model = createBasicModel();
-
-    const result = await model.doGenerate({
-      prompt: 'Edit multiple images',
-      files: [
-        {
-          type: 'url',
-          url: 'https://example.com/input1.jpg',
-        },
-        {
-          type: 'url',
-          url: 'https://example.com/input2.jpg',
-        },
-      ],
-      mask: undefined,
-      n: 1,
-      size: undefined,
-      aspectRatio: undefined,
-      seed: undefined,
-      providerOptions: {},
-    });
-
-    expect(result.warnings).toMatchInlineSnapshot(`
-      [
-        {
-          "message": "Together AI only supports a single input image. Additional images are ignored.",
-          "type": "other",
-        },
-      ]
-    `);
-
-    // Should only use the first image
-    const requestBody = await server.calls[0].requestBodyJson;
-    expect(requestBody.image_url).toBe('https://example.com/input1.jpg');
-  });
-
-  it('should pass provider options with image editing', async () => {
-    const model = createBasicModel();
-
-    await model.doGenerate({
-      prompt: 'Transform the style',
-      files: [
-        {
-          type: 'url',
-          url: 'https://example.com/input.jpg',
-        },
-      ],
-      mask: undefined,
-      n: 1,
-      size: undefined,
-      aspectRatio: undefined,
-      seed: undefined,
-      providerOptions: {
-        togetherai: {
-          steps: 28,
-          guidance: 3.5,
-        },
-      },
-    });
-
-    const requestBody = await server.calls[0].requestBodyJson;
-    expect(requestBody).toMatchInlineSnapshot(`
-      {
-        "guidance": 3.5,
-        "image_url": "https://example.com/input.jpg",
-        "model": "stabilityai/stable-diffusion-xl",
-        "prompt": "Transform the style",
-        "response_format": "base64",
-        "steps": 28,
-      }
-    `);
-  });
-});

package/src/togetherai-provider.test.ts

@@ -1,196 +0,0 @@
-import {
-  OpenAICompatibleChatLanguageModel,
-  OpenAICompatibleCompletionLanguageModel,
-  OpenAICompatibleEmbeddingModel,
-} from '@ai-sdk/openai-compatible';
-import {
-  EmbeddingModelV3,
-  LanguageModelV3,
-  RerankingModelV3,
-} from '@ai-sdk/provider';
-import { loadApiKey } from '@ai-sdk/provider-utils';
-import { beforeEach, describe, expect, it, Mock, vi } from 'vitest';
-import { TogetherAIRerankingModel } from './reranking/togetherai-reranking-model';
-import { TogetherAIImageModel } from './togetherai-image-model';
-import { createTogetherAI } from './togetherai-provider';
-
-// 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('./togetherai-image-model', () => ({
-  TogetherAIImageModel: vi.fn(),
-}));
-
-vi.mock('./reranking/togetherai-reranking-model', () => ({
-  TogetherAIRerankingModel: vi.fn(),
-}));
-
-describe('TogetherAIProvider', () => {
-  let mockLanguageModel: LanguageModelV3;
-  let mockEmbeddingModel: EmbeddingModelV3;
-  let mockRerankingModel: RerankingModelV3;
-
-  beforeEach(() => {
-    // Mock implementations of models
-    mockLanguageModel = {
-      // Add any required methods for LanguageModelV3
-    } as LanguageModelV3;
-    mockEmbeddingModel = {
-      // Add any required methods for EmbeddingModelV3
-    } as EmbeddingModelV3;
-    mockRerankingModel = {
-      // Add any required methods for RerankingModelV3
-    } as RerankingModelV3;
-
-    // Reset mocks
-    vi.clearAllMocks();
-  });
-
-  describe('createTogetherAI', () => {
-    it('should create a TogetherAIProvider instance with default options', () => {
-      const provider = createTogetherAI();
-      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: 'TOGETHER_AI_API_KEY',
-        description: 'TogetherAI',
-      });
-    });
-
-    it('should create a TogetherAIProvider instance with custom options', () => {
-      const options = {
-        apiKey: 'custom-key',
-        baseURL: 'https://custom.url',
-        headers: { 'Custom-Header': 'value' },
-      };
-      const provider = createTogetherAI(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: 'TOGETHER_AI_API_KEY',
-        description: 'TogetherAI',
-      });
-    });
-
-    it('should return a chat model when called as a function', () => {
-      const provider = createTogetherAI();
-      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 = createTogetherAI();
-      const modelId = 'together-chat-model';
-
-      const model = provider.chatModel(modelId);
-
-      expect(model).toBeInstanceOf(OpenAICompatibleChatLanguageModel);
-    });
-  });
-
-  describe('completionModel', () => {
-    it('should construct a completion model with correct configuration', () => {
-      const provider = createTogetherAI();
-      const modelId = 'together-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 = createTogetherAI();
-      const modelId = 'together-embedding-model';
-
-      const model = provider.embeddingModel(modelId);
-
-      expect(model).toBeInstanceOf(OpenAICompatibleEmbeddingModel);
-    });
-  });
-
-  describe('image', () => {
-    it('should construct an image model with correct configuration', () => {
-      const provider = createTogetherAI();
-      const modelId = 'stabilityai/stable-diffusion-xl';
-
-      const model = provider.image(modelId);
-
-      expect(TogetherAIImageModel).toHaveBeenCalledWith(
-        modelId,
-        expect.objectContaining({
-          provider: 'togetherai.image',
-          baseURL: 'https://api.together.xyz/v1/',
-        }),
-      );
-      expect(model).toBeInstanceOf(TogetherAIImageModel);
-    });
-
-    it('should pass custom baseURL to image model', () => {
-      const provider = createTogetherAI({
-        baseURL: 'https://custom.url/',
-      });
-      const modelId = 'stabilityai/stable-diffusion-xl';
-
-      provider.image(modelId);
-
-      expect(TogetherAIImageModel).toHaveBeenCalledWith(
-        modelId,
-        expect.objectContaining({
-          baseURL: 'https://custom.url/',
-        }),
-      );
-    });
-  });
-
-  describe('rerankingModel', () => {
-    it('should construct a reranking model with correct configuration', () => {
-      const provider = createTogetherAI();
-      const modelId = 'Salesforce/Llama-Rank-v1';
-      0;
-      const model = provider.rerankingModel(modelId);
-
-      expect(TogetherAIRerankingModel).toHaveBeenCalledWith(
-        modelId,
-        expect.objectContaining({
-          baseURL: 'https://api.together.xyz/v1/',
-        }),
-      );
-      expect(model).toBeInstanceOf(TogetherAIRerankingModel);
-    });
-  });
-});