test(llm-router): cover local ollama inference

This commit is contained in:
root 2026-03-14 06:41:08 +00:00
parent 7616901683
commit 50fc9c93c5

View File

@ -80,6 +80,26 @@ describe('getAvailableProviders', () => {
expect(result[0]!.name).toBe('local-ollama');
});
it('infers local ollama model metadata for routing', () => {
const local = createLocalOllamaProvider(['qwen2.5-coder:7b', 'llama3.1:70b']);
expect(local.baseUrl).toBe('http://localhost:11434/v1');
expect(local.models).toEqual([
expect.objectContaining({
id: 'qwen2.5-coder:7b',
contextWindow: 32_768,
speedTier: 1,
strengths: expect.arrayContaining(['code']),
}),
expect.objectContaining({
id: 'llama3.1:70b',
contextWindow: 8_192,
speedTier: 3,
strengths: expect.arrayContaining(['general']),
}),
]);
});
it('DEFAULT_PROVIDERS includes all 4 providers', () => {
expect(DEFAULT_PROVIDERS).toHaveLength(4);
const names = DEFAULT_PROVIDERS.map(p => p.name);