fix: add z.ai coding plan support (#1370)

This commit is contained in:
Ralph Khreish
2025-11-01 14:35:22 +01:00
committed by GitHub
parent 560a469f5c
commit 9c3b2737dd
15 changed files with 200 additions and 28 deletions

View File

@@ -0,0 +1,80 @@
/**
* Tests for ZAICodingProvider
*/
import { ZAICodingProvider } from '../../../src/ai-providers/zai-coding.js';
describe('ZAICodingProvider', () => {
let provider;
beforeEach(() => {
provider = new ZAICodingProvider();
});
describe('constructor', () => {
it('should initialize with correct name', () => {
expect(provider.name).toBe('Z.ai (Coding Plan)');
});
it('should initialize with correct coding endpoint baseURL', () => {
expect(provider.defaultBaseURL).toBe(
'https://api.z.ai/api/coding/paas/v4/'
);
});
it('should inherit from OpenAICompatibleProvider', () => {
expect(provider).toHaveProperty('generateText');
expect(provider).toHaveProperty('streamText');
expect(provider).toHaveProperty('generateObject');
});
});
describe('getRequiredApiKeyName', () => {
it('should return ZAI_API_KEY environment variable name', () => {
expect(provider.getRequiredApiKeyName()).toBe('ZAI_API_KEY');
});
});
describe('isRequiredApiKey', () => {
it('should return true as API key is required', () => {
expect(provider.isRequiredApiKey()).toBe(true);
});
});
describe('getClient', () => {
it('should create client with API key', () => {
const params = { apiKey: 'test-key' };
const client = provider.getClient(params);
expect(client).toBeDefined();
});
it('should use coding endpoint by default', () => {
const params = {
apiKey: 'test-key'
};
const client = provider.getClient(params);
expect(client).toBeDefined();
// The provider should use the coding endpoint
});
it('should throw error when API key is missing', () => {
expect(() => {
provider.getClient({});
}).toThrow('Z.ai (Coding Plan) API key is required.');
});
});
describe('validateAuth', () => {
it('should validate API key is present', () => {
expect(() => {
provider.validateAuth({});
}).toThrow('Z.ai (Coding Plan) API key is required');
});
it('should pass with valid API key', () => {
expect(() => {
provider.validateAuth({ apiKey: 'test-key' });
}).not.toThrow();
});
});
});

View File

@@ -261,6 +261,13 @@ jest.unstable_mockModule('../../src/ai-providers/index.js', () => ({
getRequiredApiKeyName: jest.fn(() => 'ZAI_API_KEY'),
isRequiredApiKey: jest.fn(() => true)
})),
ZAICodingProvider: jest.fn(() => ({
generateText: jest.fn(),
streamText: jest.fn(),
generateObject: jest.fn(),
getRequiredApiKeyName: jest.fn(() => 'ZAI_API_KEY'),
isRequiredApiKey: jest.fn(() => true)
})),
LMStudioProvider: jest.fn(() => ({
generateText: jest.fn(),
streamText: jest.fn(),