test(llm-router): cover local ollama inference
This commit is contained in:
parent
7616901683
commit
50fc9c93c5
@ -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);
|
||||
|
||||
Loading…
Reference in New Issue
Block a user