William Valentin
0eb1f7a073
Add native GeminiClient using @google/generative-ai SDK and BedrockClient using @aws-sdk/client-bedrock-runtime. Replace the previous Gemini fallback (OpenAI-compatible shim) with the real implementation. Add OpenRouter as a provider option (OpenAI-compatible with custom baseURL). Update model costs, doctor CLI checks, and client factory tests.
97 lines
2.9 KiB
TypeScript
97 lines
2.9 KiB
TypeScript
import { describe, it, expect } from 'vitest';
|
|
import { createClientFromConfig } from './index.js';
|
|
import { AnthropicClient } from '../models/anthropic.js';
|
|
import { OpenAIClient } from '../models/openai.js';
|
|
import { OllamaClient } from '../models/local/ollama.js';
|
|
import { LlamaCppClient } from '../models/local/llamacpp.js';
|
|
import { GeminiClient } from '../models/gemini.js';
|
|
import { BedrockClient } from '../models/bedrock.js';
|
|
|
|
describe('createClientFromConfig', () => {
|
|
it('creates AnthropicClient for anthropic provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'anthropic',
|
|
model: 'claude-sonnet-4-5-20250514',
|
|
api_key: 'sk-ant-test',
|
|
});
|
|
expect(client).toBeInstanceOf(AnthropicClient);
|
|
});
|
|
|
|
it('creates OpenAIClient for openai provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'openai',
|
|
model: 'gpt-4o',
|
|
api_key: 'sk-test',
|
|
});
|
|
expect(client).toBeInstanceOf(OpenAIClient);
|
|
});
|
|
|
|
it('creates OllamaClient for ollama provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'ollama',
|
|
model: 'llama3.2:1b',
|
|
endpoint: 'http://localhost:11434',
|
|
});
|
|
expect(client).toBeInstanceOf(OllamaClient);
|
|
});
|
|
|
|
it('creates OllamaClient with num_gpu option', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'ollama',
|
|
model: 'llama3.2:1b',
|
|
num_gpu: 0,
|
|
});
|
|
expect(client).toBeInstanceOf(OllamaClient);
|
|
});
|
|
|
|
it('creates LlamaCppClient for llamacpp provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'llamacpp',
|
|
model: 'ministral-reasoning',
|
|
endpoint: 'http://localhost:8080',
|
|
});
|
|
expect(client).toBeInstanceOf(LlamaCppClient);
|
|
});
|
|
|
|
it('defaults llamacpp endpoint to localhost:8080', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'llamacpp',
|
|
model: 'test-model',
|
|
});
|
|
expect(client).toBeInstanceOf(LlamaCppClient);
|
|
});
|
|
|
|
it('creates GeminiClient for gemini provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'gemini',
|
|
model: 'gemini-2.5-pro',
|
|
api_key: 'test-key',
|
|
});
|
|
expect(client).toBeInstanceOf(GeminiClient);
|
|
});
|
|
|
|
it('throws for unknown provider', () => {
|
|
expect(() => createClientFromConfig({
|
|
provider: 'unknown' as 'anthropic',
|
|
model: 'test',
|
|
})).toThrow('Unknown model provider: unknown');
|
|
});
|
|
|
|
it('creates OpenAIClient with OpenRouter baseURL for openrouter provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'openrouter',
|
|
model: 'meta-llama/llama-3.1-70b',
|
|
api_key: 'test-key',
|
|
});
|
|
expect(client).toBeInstanceOf(OpenAIClient);
|
|
});
|
|
|
|
it('creates BedrockClient for bedrock provider', () => {
|
|
const client = createClientFromConfig({
|
|
provider: 'bedrock',
|
|
model: 'anthropic.claude-3-sonnet',
|
|
});
|
|
expect(client).toBeInstanceOf(BedrockClient);
|
|
});
|
|
});
|