feat: added tests

Author: MikeSquall

src/types/chat/ChatModel.ts

@@ -4,6 +4,4 @@ export type ChatModel =
   | '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';
+  | 'jamba-mini-1.6';

tests/unittests/resources/chat/completions.test.ts

@@ -30,7 +30,7 @@ describe('Completions', () => {
 
   it('should create a non-streaming completion when stream is false', async () => {
     const body: Models.ChatCompletionCreateParamsNonStreaming = { 
-        model: "jamba-1.5-mini",
+        model: "jamba-mini",
         messages: [{role: "user", content: "Hello"}],
     };
     const options: Models.RequestOptions = { headers: { 'Authorization': `Bearer ${dummyAPIKey}` } };
@@ -58,7 +58,7 @@ describe('Completions', () => {
   });
 
   it('should create a streaming completion when stream is true', async () => {
-    const body: Models.ChatCompletionCreateParamsStreaming = { model: "jamba-1.5-mini", messages: [{role: "user", content: "Hello"}], stream: true };
+    const body: Models.ChatCompletionCreateParamsStreaming = { model: "jamba-mini", messages: [{role: "user", content: "Hello"}], stream: true };
     const options: Models.RequestOptions = { headers: { 'Authorization': `Bearer ${dummyAPIKey}` } };
     const expectedStream: Models.ChatCompletionChunk = {
         id: "test-id",
@@ -85,7 +85,7 @@ describe('Completions', () => {
   });
 
   it('should use default options when options parameter is omitted', async () => {
-    const body: Models.ChatCompletionCreateParamsNonStreaming = { model: "jamba-1.5-mini", messages: [{role: "user", content: "Hello"}], stream: false };
+    const body: Models.ChatCompletionCreateParamsNonStreaming = { model: "jamba-mini", messages: [{role: "user", content: "Hello"}], stream: false };
     const expectedResponse: Models.ChatCompletionResponse = {
         id: "test-id",
         usage: {
@@ -108,7 +108,7 @@ describe('Completions', () => {
   });
 
   it('should handle errors thrown by the API client', async () => {
-    const body: Models.ChatCompletionCreateParamsNonStreaming = { model: "jamba-1.5-mini", messages: [{role: "user", content: "Hello"}], stream: false };
+    const body: Models.ChatCompletionCreateParamsNonStreaming = { model: "jamba-mini", messages: [{role: "user", content: "Hello"}], stream: false };
     const error = new AI21Error();
 
     mockClient.post.mockRejectedValue(error);

tests/unittests/resources/maestro/maestro.test.ts

@@ -0,0 +1,70 @@
+import { Maestro } from '../../../../src/resources/maestro/maestro';
+import { Beta } from '../../../../src/resources/maestro/beta';
+import { Runs } from '../../../../src/resources/maestro/runs';
+import { APIClient } from '../../../../src/APIClient';
+
+class MockAPIClient extends APIClient {
+  public post = jest.fn();
+  public get = jest.fn();
+}
+
+describe('Maestro', () => {
+  let maestro: Maestro;
+  let mockClient: MockAPIClient;
+
+  beforeEach(() => {
+    mockClient = new MockAPIClient({
+      baseURL: 'https://api.example.com',
+      maxRetries: 3,
+      timeout: 5000,
+    });
+
+    maestro = new Maestro(mockClient);
+  });
+
+  afterEach(() => {
+    jest.clearAllMocks();
+  });
+
+  it('should initialize with a Runs instance', () => {
+    expect(maestro).toBeInstanceOf(Maestro);
+    expect(maestro.runs).toBeInstanceOf(Runs);
+  });
+
+  it('should share the same client with runs', () => {
+    expect(maestro.runs['client']).toBe(mockClient);
+  });
+});
+
+describe('Beta', () => {
+  let beta: Beta;
+  let mockClient: MockAPIClient;
+
+  beforeEach(() => {
+    mockClient = new MockAPIClient({
+      baseURL: 'https://api.example.com',
+      maxRetries: 3,
+      timeout: 5000,
+    });
+
+    beta = new Beta(mockClient);
+  });
+
+  afterEach(() => {
+    jest.clearAllMocks();
+  });
+
+  it('should initialize with a Maestro instance', () => {
+    expect(beta).toBeInstanceOf(Beta);
+    expect(beta.maestro).toBeInstanceOf(Maestro);
+  });
+
+  it('should have access to runs through maestro', () => {
+    expect(beta.maestro.runs).toBeInstanceOf(Runs);
+  });
+
+  it('should share the same client through the hierarchy', () => {
+    expect(beta.maestro['client']).toBe(mockClient);
+    expect(beta.maestro.runs['client']).toBe(mockClient);
+  });
+});