bytelyst/learning_ai_common_plat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(llm-router): cover local ollama inference

Browse Source
This commit is contained in:
root 2026-03-14 06:41:08 +00:00
parent 7616901683
commit 50fc9c93c5
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
packages/llm-router/src/__tests__/registry.test.ts
View File

@ -80,6 +80,26 @@ describe('getAvailableProviders', () => {
expect(result[0]!.name).toBe('local-ollama'); 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', () => { it('DEFAULT_PROVIDERS includes all 4 providers', () => {
expect(DEFAULT_PROVIDERS).toHaveLength(4); expect(DEFAULT_PROVIDERS).toHaveLength(4);
const names = DEFAULT_PROVIDERS.map(p => p.name); const names = DEFAULT_PROVIDERS.map(p => p.name);