@ai-sdk/amazon-bedrock 4.0.28 → 4.0.29

package/src/inject-fetch-headers.test.ts

@@ -1,135 +0,0 @@
-import { injectFetchHeaders } from './inject-fetch-headers';
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-// Mock the version module
-vi.mock('./version', () => ({
-  VERSION: '0.0.0-test',
-}));
-
-// // Mock provider-utils to control runtime environment detection
-vi.mock('@ai-sdk/provider-utils', async () => {
-  const actual = await vi.importActual('@ai-sdk/provider-utils');
-  return {
-    ...actual,
-    getRuntimeEnvironmentUserAgent: vi.fn(() => 'runtime/testenv'),
-  };
-});
-
-describe('injectFetchHeaders', () => {
-  const originalFetch = globalThis.fetch;
-
-  beforeEach(() => {
-    globalThis.fetch = vi.fn();
-  });
-
-  afterEach(() => {
-    globalThis.fetch = originalFetch;
-    vi.clearAllMocks();
-  });
-
-  it('should inject custom headers into fetch request', async () => {
-    const mockFetch = vi.fn().mockResolvedValue('response');
-    globalThis.fetch = mockFetch;
-
-    const customHeaders = {
-      'x-custom-header': 'custom-value',
-      authorization: 'Bearer token',
-    };
-
-    const enhancedFetch = injectFetchHeaders(customHeaders);
-    await enhancedFetch('https://api.example.com');
-
-    expect(mockFetch).toHaveBeenCalledWith(
-      'https://api.example.com',
-      expect.objectContaining({
-        headers: expect.objectContaining({
-          'x-custom-header': 'custom-value',
-          authorization: 'Bearer token',
-          'user-agent': 'ai-sdk/amazon-bedrock/0.0.0-test runtime/testenv',
-        }),
-      }),
-    );
-  });
-
-  it('should merge custom headers with existing headers', async () => {
-    const mockFetch = vi.fn().mockResolvedValue('response');
-    globalThis.fetch = mockFetch;
-
-    const customHeaders = {
-      'x-custom-header': 'custom-value',
-    };
-
-    const existingHeaders = {
-      'Content-Type': 'application/json',
-    };
-
-    const enhancedFetch = injectFetchHeaders(customHeaders);
-    await enhancedFetch('https://api.example.com', {
-      headers: existingHeaders,
-    });
-
-    expect(mockFetch).toHaveBeenCalledWith(
-      'https://api.example.com',
-      expect.objectContaining({
-        headers: expect.objectContaining({
-          'content-type': 'application/json',
-          'x-custom-header': 'custom-value',
-          'user-agent': 'ai-sdk/amazon-bedrock/0.0.0-test runtime/testenv',
-        }),
-      }),
-    );
-  });
-
-  it('should handle undefined headers in init', async () => {
-    const mockFetch = vi.fn().mockResolvedValue('response');
-    globalThis.fetch = mockFetch;
-
-    const customHeaders = {
-      'x-custom-header': 'custom-value',
-    };
-
-    const enhancedFetch = injectFetchHeaders(customHeaders);
-    await enhancedFetch('https://api.example.com', {
-      headers: undefined,
-    });
-
-    expect(mockFetch).toHaveBeenCalledWith(
-      'https://api.example.com',
-      expect.objectContaining({
-        headers: expect.objectContaining({
-          'x-custom-header': 'custom-value',
-          'user-agent': 'ai-sdk/amazon-bedrock/0.0.0-test runtime/testenv',
-        }),
-      }),
-    );
-  });
-
-  it('should handle Headers instance in init', async () => {
-    const mockFetch = vi.fn().mockResolvedValue('response');
-    globalThis.fetch = mockFetch;
-
-    const customHeaders = {
-      'x-custom-header': 'custom-value',
-    };
-
-    const existingHeaders = new Headers({
-      'Content-Type': 'application/json',
-    });
-
-    const enhancedFetch = injectFetchHeaders(customHeaders);
-    await enhancedFetch('https://api.example.com', {
-      headers: existingHeaders,
-    });
-
-    expect(mockFetch).toHaveBeenCalledWith(
-      'https://api.example.com',
-      expect.objectContaining({
-        headers: expect.objectContaining({
-          'content-type': 'application/json',
-          'x-custom-header': 'custom-value',
-          'user-agent': 'ai-sdk/amazon-bedrock/0.0.0-test runtime/testenv',
-        }),
-      }),
-    );
-  });
-});

package/src/normalize-tool-call-id.test.ts

@@ -1,72 +0,0 @@
-import { describe, it, expect } from 'vitest';
-import { isMistralModel, normalizeToolCallId } from './normalize-tool-call-id';
-
-describe('isMistralModel', () => {
-  it('should return true for mistral models', () => {
-    expect(isMistralModel('mistral.mistral-7b-instruct-v0:2')).toBe(true);
-    expect(isMistralModel('mistral.mixtral-8x7b-instruct-v0:1')).toBe(true);
-    expect(isMistralModel('mistral.mistral-large-2402-v1:0')).toBe(true);
-    expect(isMistralModel('mistral.mistral-small-2402-v1:0')).toBe(true);
-    expect(isMistralModel('mistral.mistral-large-2407-v1:0')).toBe(true);
-    expect(isMistralModel('mistral.ministral-3-14b-instruct')).toBe(true);
-    expect(isMistralModel('mistral.ministral-3-8b-instruct')).toBe(true);
-  });
-
-  it('should return true for region-prefixed mistral models', () => {
-    expect(isMistralModel('us.mistral.pixtral-large-2502-v1:0')).toBe(true);
-    expect(isMistralModel('eu.mistral.mistral-large-2407-v1:0')).toBe(true);
-  });
-
-  it('should return false for non-mistral models', () => {
-    expect(isMistralModel('anthropic.claude-3-5-sonnet-20241022-v2:0')).toBe(
-      false,
-    );
-    expect(isMistralModel('amazon.nova-pro-v1:0')).toBe(false);
-    expect(isMistralModel('openai.gpt-4o')).toBe(false);
-    expect(isMistralModel('meta.llama3-70b-instruct-v1:0')).toBe(false);
-  });
-});
-
-describe('normalizeToolCallId', () => {
-  it('should return the original ID when not a Mistral model', () => {
-    const originalId = 'tooluse_bpe71yCfRu2b5i-nKGDr5g';
-    expect(normalizeToolCallId(originalId, false)).toBe(originalId);
-  });
-
-  it('should extract first 9 alphanumeric characters for Mistral models', () => {
-    // Bedrock format: tooluse_bpe71yCfRu2b5i-nKGDr5g
-    // After removing non-alphanumeric: toolusebpe71yCfRu2b5inKGDr5g
-    // First 9 chars: toolusebp
-    expect(normalizeToolCallId('tooluse_bpe71yCfRu2b5i-nKGDr5g', true)).toBe(
-      'toolusebp',
-    );
-  });
-
-  it('should handle IDs with various special characters', () => {
-    expect(normalizeToolCallId('tool-use_123ABC456', true)).toBe('tooluse12');
-    expect(normalizeToolCallId('___abc123DEF___', true)).toBe('abc123DEF');
-  });
-
-  it('should handle IDs that are already alphanumeric', () => {
-    expect(normalizeToolCallId('abcdefghi', true)).toBe('abcdefghi');
-    expect(normalizeToolCallId('abc123XYZ', true)).toBe('abc123XYZ');
-  });
-
-  it('should handle short IDs', () => {
-    expect(normalizeToolCallId('abc', true)).toBe('abc');
-    expect(normalizeToolCallId('12345', true)).toBe('12345');
-  });
-
-  it('should handle IDs with only special characters', () => {
-    expect(normalizeToolCallId('___---___', true)).toBe('');
-  });
-
-  it('should produce valid Mistral tool call IDs (9 alphanumeric chars)', () => {
-    const normalizedId = normalizeToolCallId(
-      'tooluse_bpe71yCfRu2b5i-nKGDr5g',
-      true,
-    );
-    // Verify the ID matches Mistral's requirements: ^[a-zA-Z0-9]{1,9}$
-    expect(normalizedId).toMatch(/^[a-zA-Z0-9]{1,9}$/);
-  });
-});

package/src/reranking/__fixtures__/bedrock-reranking.1.json

@@ -1,12 +0,0 @@
-{
-  "results": [
-    {
-      "index": 0,
-      "relevanceScore": 0.5110583305358887
-    },
-    {
-      "index": 5,
-      "relevanceScore": 0.30241215229034424
-    }
-  ]
-}

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

@@ -1,299 +0,0 @@
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import fs from 'node:fs';
-import { beforeEach, describe, expect, it } from 'vitest';
-import { injectFetchHeaders } from '../inject-fetch-headers';
-import { BedrockRerankingModel } from './bedrock-reranking-model';
-import { BedrockRerankingOptions } from './bedrock-reranking-options';
-
-const fakeFetchWithAuth = injectFetchHeaders({ 'x-amz-auth': 'test-auth' });
-
-const model = new BedrockRerankingModel('cohere.rerank-v3-5:0', {
-  baseUrl: () => 'https://bedrock-agent-runtime.us-east-1.amazonaws.com',
-  region: 'us-west-2',
-  headers: {
-    'config-header': 'config-value',
-    'shared-header': 'config-shared',
-  },
-  fetch: fakeFetchWithAuth,
-});
-
-describe('doRerank', () => {
-  const server = createTestServer({
-    'https://bedrock-agent-runtime.us-east-1.amazonaws.com/rerank': {},
-  });
-
-  function prepareJsonFixtureResponse(filename: string) {
-    server.urls[
-      'https://bedrock-agent-runtime.us-east-1.amazonaws.com/rerank'
-    ].response = {
-      type: 'binary',
-      headers: { 'content-type': 'application/json' },
-      body: Buffer.from(
-        fs.readFileSync(`src/reranking/__fixtures__/${filename}.json`, 'utf8'),
-      ),
-    };
-  }
-
-  describe('json documents', () => {
-    let result: Awaited<ReturnType<typeof model.doRerank>>;
-
-    beforeEach(async () => {
-      prepareJsonFixtureResponse('bedrock-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: {
-          bedrock: {
-            nextToken: 'test-token',
-            additionalModelRequestFields: {
-              test: 'test-value',
-            },
-          } satisfies BedrockRerankingOptions,
-        },
-      });
-    });
-
-    it('should send request with stringified json documents', async () => {
-      expect(await server.calls[0].requestBodyJson).toMatchInlineSnapshot(`
-        {
-          "nextToken": "test-token",
-          "queries": [
-            {
-              "textQuery": {
-                "text": "rainy day",
-              },
-              "type": "TEXT",
-            },
-          ],
-          "rerankingConfiguration": {
-            "bedrockRerankingConfiguration": {
-              "modelConfiguration": {
-                "additionalModelRequestFields": {
-                  "test": "test-value",
-                },
-                "modelArn": "arn:aws:bedrock:us-west-2::foundation-model/cohere.rerank-v3-5:0",
-              },
-              "numberOfResults": 2,
-            },
-            "type": "BEDROCK_RERANKING_MODEL",
-          },
-          "sources": [
-            {
-              "inlineDocumentSource": {
-                "jsonDocument": {
-                  "example": "sunny day at the beach",
-                },
-                "type": "JSON",
-              },
-              "type": "INLINE",
-            },
-            {
-              "inlineDocumentSource": {
-                "jsonDocument": {
-                  "example": "rainy day in the city",
-                },
-                "type": "JSON",
-              },
-              "type": "INLINE",
-            },
-          ],
-        }
-      `);
-    });
-
-    it('should send request with the correct headers', async () => {
-      expect(server.calls[0].requestHeaders).toMatchInlineSnapshot(`
-        {
-          "config-header": "config-value",
-          "content-type": "application/json",
-          "shared-header": "config-shared",
-          "x-amz-auth": "test-auth",
-        }
-      `);
-    });
-
-    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.5110583305358887,
-          },
-          {
-            "index": 5,
-            "relevanceScore": 0.30241215229034424,
-          },
-        ]
-      `);
-    });
-
-    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": {
-            "results": [
-              {
-                "index": 0,
-                "relevanceScore": 0.5110583305358887,
-              },
-              {
-                "index": 5,
-                "relevanceScore": 0.30241215229034424,
-              },
-            ],
-          },
-          "headers": {
-            "content-length": "171",
-            "content-type": "application/json",
-          },
-        }
-      `);
-    });
-  });
-
-  describe('text documents', () => {
-    let result: Awaited<ReturnType<typeof model.doRerank>>;
-
-    beforeEach(async () => {
-      prepareJsonFixtureResponse('bedrock-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: {
-          bedrock: {
-            nextToken: 'test-token',
-            additionalModelRequestFields: {
-              test: 'test-value',
-            },
-          } satisfies BedrockRerankingOptions,
-        },
-      });
-    });
-
-    it('should send request with text documents', async () => {
-      expect(await server.calls[0].requestBodyJson).toMatchInlineSnapshot(`
-        {
-          "nextToken": "test-token",
-          "queries": [
-            {
-              "textQuery": {
-                "text": "rainy day",
-              },
-              "type": "TEXT",
-            },
-          ],
-          "rerankingConfiguration": {
-            "bedrockRerankingConfiguration": {
-              "modelConfiguration": {
-                "additionalModelRequestFields": {
-                  "test": "test-value",
-                },
-                "modelArn": "arn:aws:bedrock:us-west-2::foundation-model/cohere.rerank-v3-5:0",
-              },
-              "numberOfResults": 2,
-            },
-            "type": "BEDROCK_RERANKING_MODEL",
-          },
-          "sources": [
-            {
-              "inlineDocumentSource": {
-                "textDocument": {
-                  "text": "sunny day at the beach",
-                },
-                "type": "TEXT",
-              },
-              "type": "INLINE",
-            },
-            {
-              "inlineDocumentSource": {
-                "textDocument": {
-                  "text": "rainy day in the city",
-                },
-                "type": "TEXT",
-              },
-              "type": "INLINE",
-            },
-          ],
-        }
-      `);
-    });
-
-    it('should send request with the correct headers', async () => {
-      expect(server.calls[0].requestHeaders).toMatchInlineSnapshot(`
-        {
-          "config-header": "config-value",
-          "content-type": "application/json",
-          "shared-header": "config-shared",
-          "x-amz-auth": "test-auth",
-        }
-      `);
-    });
-
-    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.5110583305358887,
-          },
-          {
-            "index": 5,
-            "relevanceScore": 0.30241215229034424,
-          },
-        ]
-      `);
-    });
-
-    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": {
-            "results": [
-              {
-                "index": 0,
-                "relevanceScore": 0.5110583305358887,
-              },
-              {
-                "index": 5,
-                "relevanceScore": 0.30241215229034424,
-              },
-            ],
-          },
-          "headers": {
-            "content-length": "171",
-            "content-type": "application/json",
-          },
-        }
-      `);
-    });
-  });
-});