@ai-sdk/cerebras 2.0.16 → 2.0.18

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @ai-sdk/cerebras
 
+## 2.0.18
+
+### Patch Changes
+
+- 8dc54db: chore: add src folders to package bundle
+- Updated dependencies [8dc54db]
+  - @ai-sdk/openai-compatible@2.0.17
+
+## 2.0.17
+
+### Patch Changes
+
+- Updated dependencies [78555ad]
+  - @ai-sdk/openai-compatible@2.0.16
+
 ## 2.0.16
 
 ### Patch Changes

package/dist/index.js

@@ -33,7 +33,7 @@ var import_provider_utils = require("@ai-sdk/provider-utils");
 var import_v4 = require("zod/v4");
 
 // src/version.ts
-var VERSION = true ? "2.0.16" : "0.0.0-test";
+var VERSION = true ? "2.0.18" : "0.0.0-test";
 
 // src/cerebras-provider.ts
 var cerebrasErrorSchema = import_v4.z.object({

package/dist/index.mjs

@@ -11,7 +11,7 @@ import {
 import { z } from "zod/v4";
 
 // src/version.ts
-var VERSION = true ? "2.0.16" : "0.0.0-test";
+var VERSION = true ? "2.0.18" : "0.0.0-test";
 
 // src/cerebras-provider.ts
 var cerebrasErrorSchema = z.object({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/cerebras",
-  "version": "2.0.16",
+  "version": "2.0.18",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -8,6 +8,7 @@
   "types": "./dist/index.d.ts",
   "files": [
     "dist/**/*",
+    "src",
     "CHANGELOG.md",
     "README.md"
   ],
@@ -20,7 +21,7 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/openai-compatible": "2.0.15",
+    "@ai-sdk/openai-compatible": "2.0.17",
     "@ai-sdk/provider": "3.0.4",
     "@ai-sdk/provider-utils": "4.0.8"
   },

package/src/cerebras-chat-options.ts

@@ -0,0 +1,13 @@
+// https://inference-docs.cerebras.ai/models/overview
+export type CerebrasChatModelId =
+  // production
+  | 'llama3.1-8b'
+  | 'llama-3.3-70b'
+  | 'gpt-oss-120b'
+  | 'qwen-3-32b'
+  // preview
+  | 'qwen-3-235b-a22b-instruct-2507'
+  | 'qwen-3-235b-a22b-thinking-2507'
+  | 'zai-glm-4.6'
+  | 'zai-glm-4.7'
+  | (string & {});

package/src/cerebras-provider.test.ts

@@ -0,0 +1,133 @@
+import { describe, it, expect, vi, beforeEach, Mock } from 'vitest';
+import { createCerebras } from './cerebras-provider';
+import { loadApiKey } from '@ai-sdk/provider-utils';
+import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';
+
+// Add type assertion for the mocked class
+const OpenAICompatibleChatLanguageModelMock =
+  OpenAICompatibleChatLanguageModel as unknown as Mock;
+
+vi.mock('@ai-sdk/openai-compatible', () => ({
+  OpenAICompatibleChatLanguageModel: vi.fn(),
+}));
+
+vi.mock('./version', () => ({
+  VERSION: '0.0.0-test',
+}));
+
+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),
+  };
+});
+
+describe('CerebrasProvider', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  describe('createCerebras', () => {
+    it('should create a CerebrasProvider instance with default options', () => {
+      const provider = createCerebras();
+      const model = provider('model-id');
+
+      const constructorCall =
+        OpenAICompatibleChatLanguageModelMock.mock.calls[0];
+      const config = constructorCall[1];
+      config.headers();
+
+      expect(loadApiKey).toHaveBeenCalledWith({
+        apiKey: undefined,
+        environmentVariableName: 'CEREBRAS_API_KEY',
+        description: 'Cerebras API key',
+      });
+    });
+
+    it('should create a CerebrasProvider instance with custom options', () => {
+      const options = {
+        apiKey: 'custom-key',
+        baseURL: 'https://custom.url',
+        headers: { 'Custom-Header': 'value' },
+      };
+      const provider = createCerebras(options);
+      provider('model-id');
+
+      const constructorCall =
+        OpenAICompatibleChatLanguageModelMock.mock.calls[0];
+      const config = constructorCall[1];
+      config.headers();
+
+      expect(loadApiKey).toHaveBeenCalledWith({
+        apiKey: 'custom-key',
+        environmentVariableName: 'CEREBRAS_API_KEY',
+        description: 'Cerebras API key',
+      });
+    });
+
+    it('should pass header', async () => {
+      const fetchMock = vi
+        .fn()
+        .mockResolvedValue(new Response('{}', { status: 200 }));
+
+      const provider = createCerebras({ fetch: fetchMock });
+      provider('model-id');
+
+      const constructorCall = vi.mocked(OpenAICompatibleChatLanguageModel).mock
+        .calls[0];
+      const config = constructorCall[1];
+      const headers = config.headers();
+
+      await fetchMock('https://api.cerebras.ai/v1/test', {
+        method: 'POST',
+        headers,
+      });
+
+      expect(fetchMock.mock.calls[0][1].headers['user-agent']).toContain(
+        'ai-sdk/cerebras/0.0.0-test',
+      );
+    });
+
+    it('should return a chat model when called as a function', () => {
+      const provider = createCerebras();
+      const modelId = 'foo-model-id';
+
+      const model = provider(modelId);
+      expect(model).toBeInstanceOf(OpenAICompatibleChatLanguageModel);
+    });
+  });
+
+  describe('languageModel', () => {
+    it('should construct a language model with correct configuration', () => {
+      const provider = createCerebras();
+      const modelId = 'foo-model-id';
+
+      const model = provider.languageModel(modelId);
+
+      expect(model).toBeInstanceOf(OpenAICompatibleChatLanguageModel);
+    });
+  });
+
+  describe('embeddingModel', () => {
+    it('should throw NoSuchModelError when attempting to create embedding model', () => {
+      const provider = createCerebras();
+
+      expect(() => provider.embeddingModel('any-model')).toThrow(
+        'No such embeddingModel: any-model',
+      );
+    });
+  });
+
+  describe('chat', () => {
+    it('should construct a chat model with correct configuration', () => {
+      const provider = createCerebras();
+      const modelId = 'foo-model-id';
+
+      const model = provider.chat(modelId);
+
+      expect(model).toBeInstanceOf(OpenAICompatibleChatLanguageModel);
+    });
+  });
+});

package/src/cerebras-provider.ts

@@ -0,0 +1,123 @@
+import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';
+import {
+  LanguageModelV3,
+  NoSuchModelError,
+  ProviderV3,
+} from '@ai-sdk/provider';
+import {
+  FetchFunction,
+  loadApiKey,
+  withoutTrailingSlash,
+  withUserAgentSuffix,
+} from '@ai-sdk/provider-utils';
+import { CerebrasChatModelId } from './cerebras-chat-options';
+import { z } from 'zod/v4';
+import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';
+import { VERSION } from './version';
+
+// Add error schema and structure
+const cerebrasErrorSchema = z.object({
+  message: z.string(),
+  type: z.string(),
+  param: z.string(),
+  code: z.string(),
+});
+
+export type CerebrasErrorData = z.infer<typeof cerebrasErrorSchema>;
+
+const cerebrasErrorStructure: ProviderErrorStructure<CerebrasErrorData> = {
+  errorSchema: cerebrasErrorSchema,
+  errorToMessage: data => data.message,
+};
+
+export interface CerebrasProviderSettings {
+  /**
+Cerebras API key.
+*/
+  apiKey?: string;
+  /**
+Base URL for the API calls.
+*/
+  baseURL?: string;
+  /**
+Custom headers to include in the requests.
+*/
+  headers?: Record<string, string>;
+  /**
+Custom fetch implementation. You can use it as a middleware to intercept requests,
+or to provide a custom fetch implementation for e.g. testing.
+*/
+  fetch?: FetchFunction;
+}
+
+export interface CerebrasProvider extends ProviderV3 {
+  /**
+Creates a Cerebras model for text generation.
+*/
+  (modelId: CerebrasChatModelId): LanguageModelV3;
+
+  /**
+Creates a Cerebras model for text generation.
+*/
+  languageModel(modelId: CerebrasChatModelId): LanguageModelV3;
+
+  /**
+Creates a Cerebras chat model for text generation.
+*/
+  chat(modelId: CerebrasChatModelId): LanguageModelV3;
+
+  /**
+   * @deprecated Use `embeddingModel` instead.
+   */
+  textEmbeddingModel(modelId: string): never;
+}
+
+export function createCerebras(
+  options: CerebrasProviderSettings = {},
+): CerebrasProvider {
+  const baseURL = withoutTrailingSlash(
+    options.baseURL ?? 'https://api.cerebras.ai/v1',
+  );
+  const getHeaders = () =>
+    withUserAgentSuffix(
+      {
+        Authorization: `Bearer ${loadApiKey({
+          apiKey: options.apiKey,
+          environmentVariableName: 'CEREBRAS_API_KEY',
+          description: 'Cerebras API key',
+        })}`,
+        ...options.headers,
+      },
+      `ai-sdk/cerebras/${VERSION}`,
+    );
+
+  const createLanguageModel = (modelId: CerebrasChatModelId) => {
+    return new OpenAICompatibleChatLanguageModel(modelId, {
+      provider: `cerebras.chat`,
+      url: ({ path }) => `${baseURL}${path}`,
+      headers: getHeaders,
+      fetch: options.fetch,
+      errorStructure: cerebrasErrorStructure,
+      supportsStructuredOutputs: true,
+    });
+  };
+
+  const provider = (modelId: CerebrasChatModelId) =>
+    createLanguageModel(modelId);
+
+  provider.specificationVersion = 'v3' as const;
+  provider.languageModel = createLanguageModel;
+  provider.chat = createLanguageModel;
+
+  provider.embeddingModel = (modelId: string) => {
+    throw new NoSuchModelError({ modelId, modelType: 'embeddingModel' });
+  };
+  provider.textEmbeddingModel = provider.embeddingModel;
+  provider.imageModel = (modelId: string) => {
+    throw new NoSuchModelError({ modelId, modelType: 'imageModel' });
+  };
+
+  return provider;
+}
+
+export const cerebras = createCerebras();

package/src/index.ts

@@ -0,0 +1,7 @@
+export { createCerebras, cerebras } from './cerebras-provider';
+export type {
+  CerebrasProvider,
+  CerebrasProviderSettings,
+} from './cerebras-provider';
+export type { CerebrasErrorData } from './cerebras-provider';
+export { VERSION } from './version';

package/src/version.ts

@@ -0,0 +1,6 @@
+// Version string of this package injected at build time.
+declare const __PACKAGE_VERSION__: string | undefined;
+export const VERSION: string =
+  typeof __PACKAGE_VERSION__ !== 'undefined'
+    ? __PACKAGE_VERSION__
+    : '0.0.0-test';