diff --git a/README.md b/README.md index b09bec3..b841004 100644 --- a/README.md +++ b/README.md @@ -81,7 +81,7 @@ const client = new AI21({ }); const response = await client.chat.completions.create({ - model: 'jamba-1.5-mini', + model: 'jamba-mini', messages: [{ role: 'user', content: 'Hello, how are you? tell me a 100 line story about a cat named "Fluffy"' }], }); @@ -96,7 +96,7 @@ The client supports streaming responses for real-time processing. Here are examp ```typescript const streamResponse = await client.chat.completions.create({ - model: 'jamba-1.5-mini', + model: 'jamba-mini', messages: [{ role: 'user', content: 'Write a story about a space cat' }], stream: true, }); diff --git a/examples/studio/chat/chat-completions.ts b/examples/studio/chat/chat-completions.ts index 5944091..0d64a7c 100644 --- a/examples/studio/chat/chat-completions.ts +++ b/examples/studio/chat/chat-completions.ts @@ -5,7 +5,7 @@ async function main() { try { const response = await client.chat.completions.create({ - model: 'jamba-1.5-mini', + model: 'jamba-mini', messages: [{ role: 'user', content: 'Hello, how are you? tell me a 100 line story about a cat' }], }); console.log(response); diff --git a/examples/studio/chat/chat-documents.ts b/examples/studio/chat/chat-documents.ts index 942c67f..da79987 100644 --- a/examples/studio/chat/chat-documents.ts +++ b/examples/studio/chat/chat-documents.ts @@ -29,7 +29,7 @@ async function main() { try { const response = await client.chat.completions.create({ - model: 'jamba-1.5-mini', + model: 'jamba-mini', messages: [ { role: 'system', diff --git a/examples/studio/chat/chat-response-format.ts b/examples/studio/chat/chat-response-format.ts index 540fd8d..5360300 100644 --- a/examples/studio/chat/chat-response-format.ts +++ b/examples/studio/chat/chat-response-format.ts @@ -22,7 +22,7 @@ async function main() { try { const response = await client.chat.completions.create({ messages, - model: 'jamba-1.5-large', + model: 'jamba-large', responseFormat, }); diff --git a/examples/studio/chat/stream-chat-completions.ts b/examples/studio/chat/stream-chat-completions.ts index efe0786..15ffaad 100644 --- a/examples/studio/chat/stream-chat-completions.ts +++ b/examples/studio/chat/stream-chat-completions.ts @@ -5,7 +5,7 @@ async function main() { try { const streamResponse = await client.chat.completions.create({ - model: 'jamba-1.5-mini', + model: 'jamba-mini', messages: [{ role: 'user', content: 'Hello, how are you? tell me a short story' }], stream: true, }); diff --git a/examples/studio/chat/tools-chat-completions.ts b/examples/studio/chat/tools-chat-completions.ts index d6121ab..0bf2d2d 100644 --- a/examples/studio/chat/tools-chat-completions.ts +++ b/examples/studio/chat/tools-chat-completions.ts @@ -42,7 +42,7 @@ async function main() { try { // First response with streaming const response = await client.chat.completions.create({ - model: 'jamba-1.5-large', + model: 'jamba-large', messages, tools, }); @@ -69,7 +69,7 @@ async function main() { // Get final response const finalResponse = await client.chat.completions.create({ - model: 'jamba-1.5-large', + model: 'jamba-large', messages, tools, }); diff --git a/src/resources/chat/completions.ts b/src/resources/chat/completions.ts index fb14d8b..7a521a9 100644 --- a/src/resources/chat/completions.ts +++ b/src/resources/chat/completions.ts @@ -2,6 +2,8 @@ import * as Models from '../../types'; import { APIResource } from '../../APIResource'; import { Stream } from '../../streaming'; +const deprecatedModels = ['jamba-1.5-mini', 'jamba-1.5-large']; + export class Completions extends APIResource { create( body: Models.ChatCompletionCreateParamsNonStreaming, @@ -19,6 +21,14 @@ export class Completions extends APIResource { ): Promise | Models.ChatCompletionResponse>; create(body: Models.ChatCompletionCreateParams, options?: Models.RequestOptions) { + // Check for deprecated models + if (deprecatedModels.includes(body.model)) { + console.warn( + `Warning: The model "${body.model}" is deprecated and will be removed in a future release. + Please use jamba-mini or jamba-large instead.`, + ); + } + return this.client.post( '/chat/completions', { diff --git a/src/types/chat/ChatModel.ts b/src/types/chat/ChatModel.ts index 3a19050..230f442 100644 --- a/src/types/chat/ChatModel.ts +++ b/src/types/chat/ChatModel.ts @@ -1 +1,9 @@ -export type ChatModel = 'jamba-1.5-mini' | 'jamba-1.5-large'; +export type ChatModel = + | 'jamba-mini' + | 'jamba-large' + | 'jamba-large-1.6-2025-03' + | 'jamba-mini-1.6-2025-03' + | 'jamba-large-1.6' + | 'jamba-mini-1.6' + | 'jamba-1.5-mini' + | 'jamba-1.5-large';