From 50fc9c93c5f81bc181070d3acc67880e290436d4 Mon Sep 17 00:00:00 2001 From: root Date: Sat, 14 Mar 2026 06:41:08 +0000 Subject: [PATCH] test(llm-router): cover local ollama inference --- .../llm-router/src/__tests__/registry.test.ts | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/packages/llm-router/src/__tests__/registry.test.ts b/packages/llm-router/src/__tests__/registry.test.ts index a66729d6..793ab862 100644 --- a/packages/llm-router/src/__tests__/registry.test.ts +++ b/packages/llm-router/src/__tests__/registry.test.ts @@ -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);