fix: Merge pull request #28 from AI21Labs/update_model_names

feat: Update Model Names to Jamba-Mini and Jamba-Large

Author: Josephasafg

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,
 });

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);

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',

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,
     });
 

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,
     });

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,
           });

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<Stream<Models.ChatCompletionChunk> | 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<Models.ChatCompletionCreateParams, Models.ChatCompletionResponse>(
       '/chat/completions',
       {

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';