@ai-sdk/openai 0.0.0-2f1ae29d-20260122140908 → 0.0.0-4115c213-20260122152721

package/src/embedding/__snapshots__/openai-embedding-model.test.ts.snap

@@ -1,43 +0,0 @@
-// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
-
-exports[`doEmbed > should expose the raw response 1`] = `
-{
-  "body": {
-    "data": [
-      {
-        "embedding": [
-          0.1,
-          0.2,
-          0.3,
-          0.4,
-          0.5,
-        ],
-        "index": 0,
-        "object": "embedding",
-      },
-      {
-        "embedding": [
-          0.6,
-          0.7,
-          0.8,
-          0.9,
-          1,
-        ],
-        "index": 1,
-        "object": "embedding",
-      },
-    ],
-    "model": "text-embedding-3-large",
-    "object": "list",
-    "usage": {
-      "prompt_tokens": 8,
-      "total_tokens": 8,
-    },
-  },
-  "headers": {
-    "content-length": "236",
-    "content-type": "application/json",
-    "test-header": "test-value",
-  },
-}
-`;

package/src/embedding/openai-embedding-model.test.ts

@@ -1,146 +0,0 @@
-import { EmbeddingModelV3Embedding } from '@ai-sdk/provider';
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import { createOpenAI } from '../openai-provider';
-import { describe, it, expect, vi } from 'vitest';
-
-vi.mock('../version', () => ({
-  VERSION: '0.0.0-test',
-}));
-
-const dummyEmbeddings = [
-  [0.1, 0.2, 0.3, 0.4, 0.5],
-  [0.6, 0.7, 0.8, 0.9, 1.0],
-];
-const testValues = ['sunny day at the beach', 'rainy day in the city'];
-
-const provider = createOpenAI({ apiKey: 'test-api-key' });
-const model = provider.embedding('text-embedding-3-large');
-
-const server = createTestServer({
-  'https://api.openai.com/v1/embeddings': {},
-});
-
-describe('doEmbed', () => {
-  function prepareJsonResponse({
-    embeddings = dummyEmbeddings,
-    usage = { prompt_tokens: 8, total_tokens: 8 },
-    headers,
-  }: {
-    embeddings?: EmbeddingModelV3Embedding[];
-    usage?: { prompt_tokens: number; total_tokens: number };
-    headers?: Record<string, string>;
-  } = {}) {
-    server.urls['https://api.openai.com/v1/embeddings'].response = {
-      type: 'json-value',
-      headers,
-      body: {
-        object: 'list',
-        data: embeddings.map((embedding, i) => ({
-          object: 'embedding',
-          index: i,
-          embedding,
-        })),
-        model: 'text-embedding-3-large',
-        usage,
-      },
-    };
-  }
-
-  it('should extract embedding', async () => {
-    prepareJsonResponse();
-
-    const { embeddings } = await model.doEmbed({ values: testValues });
-
-    expect(embeddings).toStrictEqual(dummyEmbeddings);
-  });
-
-  it('should expose the raw response', async () => {
-    prepareJsonResponse({
-      headers: {
-        'test-header': 'test-value',
-      },
-    });
-
-    const { response } = await model.doEmbed({ values: testValues });
-
-    expect(response?.headers).toStrictEqual({
-      // default headers:
-      'content-length': '236',
-      'content-type': 'application/json',
-
-      // custom header
-      'test-header': 'test-value',
-    });
-    expect(response).toMatchSnapshot();
-  });
-
-  it('should extract usage', async () => {
-    prepareJsonResponse({
-      usage: { prompt_tokens: 20, total_tokens: 20 },
-    });
-
-    const { usage } = await model.doEmbed({ values: testValues });
-
-    expect(usage).toStrictEqual({ tokens: 20 });
-  });
-
-  it('should pass the model and the values', async () => {
-    prepareJsonResponse();
-
-    await model.doEmbed({ values: testValues });
-
-    expect(await server.calls[0].requestBodyJson).toStrictEqual({
-      model: 'text-embedding-3-large',
-      input: testValues,
-      encoding_format: 'float',
-    });
-  });
-
-  it('should pass the dimensions setting', async () => {
-    prepareJsonResponse();
-
-    await provider.embedding('text-embedding-3-large').doEmbed({
-      values: testValues,
-      providerOptions: { openai: { dimensions: 64 } },
-    });
-
-    expect(await server.calls[0].requestBodyJson).toStrictEqual({
-      model: 'text-embedding-3-large',
-      input: testValues,
-      encoding_format: 'float',
-      dimensions: 64,
-    });
-  });
-
-  it('should pass headers', async () => {
-    prepareJsonResponse();
-
-    const provider = createOpenAI({
-      apiKey: 'test-api-key',
-      organization: 'test-organization',
-      project: 'test-project',
-      headers: {
-        'Custom-Provider-Header': 'provider-header-value',
-      },
-    });
-
-    await provider.embedding('text-embedding-3-large').doEmbed({
-      values: testValues,
-      headers: {
-        'Custom-Request-Header': 'request-header-value',
-      },
-    });
-
-    expect(server.calls[0].requestHeaders).toStrictEqual({
-      authorization: 'Bearer test-api-key',
-      'content-type': 'application/json',
-      'custom-provider-header': 'provider-header-value',
-      'custom-request-header': 'request-header-value',
-      'openai-organization': 'test-organization',
-      'openai-project': 'test-project',
-    });
-    expect(server.calls[0].requestUserAgent).toContain(
-      `ai-sdk/openai/0.0.0-test`,
-    );
-  });
-});