From 0ca77a3ccb091982d77d9b312d8921987704946a Mon Sep 17 00:00:00 2001 From: Yasha Date: Sun, 26 Apr 2026 13:40:17 +0200 Subject: [PATCH] feat: added model --- packages/core/src/nvidia.ts | 8 +++++++- packages/core/src/providers.ts | 2 +- packages/core/tests/nvidia.test.ts | 4 ++-- 3 files changed, 10 insertions(+), 4 deletions(-) diff --git a/packages/core/src/nvidia.ts b/packages/core/src/nvidia.ts index 7f5dce8..fbf2440 100644 --- a/packages/core/src/nvidia.ts +++ b/packages/core/src/nvidia.ts @@ -17,7 +17,13 @@ export const NVIDIA_BASE_URL = 'https://integrate.api.nvidia.com/v1'; * Surfaces in the wizard as the placeholder; the user can override via the * model picker (which fetches the live list from `/v1/models`). */ -export const NVIDIA_DEFAULT_MODEL = 'deepseek-ai/deepseek-v4-pro'; +/** + * NVIDIA NIM advertises some models in `/v1/models` whose chat-completions + * endpoints stall with no response (e.g. `deepseek-v4-pro`, `deepseek-v4-flash`). + * Pick a model whose streaming endpoint is verified working so first-run users + * don't get stuck in an infinite loading state. + */ +export const NVIDIA_DEFAULT_MODEL = 'moonshotai/kimi-k2.5'; export interface NvidiaModelSummary { id: string; diff --git a/packages/core/src/providers.ts b/packages/core/src/providers.ts index 4beda11..ddb5430 100644 --- a/packages/core/src/providers.ts +++ b/packages/core/src/providers.ts @@ -41,7 +41,7 @@ export const PROVIDERS: ProviderInfo[] = [ id: 'nvidia', name: 'NVIDIA NIM', envVar: 'NVIDIA_API_KEY', - defaultModel: 'deepseek-ai/deepseek-v4-pro', + defaultModel: 'moonshotai/kimi-k2.5', defaultBaseUrl: 'https://integrate.api.nvidia.com/v1', customResolution: true, }, diff --git a/packages/core/tests/nvidia.test.ts b/packages/core/tests/nvidia.test.ts index d9c1597..d91aa98 100644 --- a/packages/core/tests/nvidia.test.ts +++ b/packages/core/tests/nvidia.test.ts @@ -36,7 +36,7 @@ describe('buildNvidiaModel', () => { }); it('exports a default model id matching the wizard\'s onboarding placeholder', () => { - expect(NVIDIA_DEFAULT_MODEL).toBe('deepseek-ai/deepseek-v4-pro'); + expect(NVIDIA_DEFAULT_MODEL).toBe('moonshotai/kimi-k2.5'); expect(NVIDIA_BASE_URL).toBe('https://integrate.api.nvidia.com/v1'); }); }); @@ -119,7 +119,7 @@ describe('PROVIDERS registry — nvidia entry', () => { const nvidia = getProviderById('nvidia'); expect(nvidia).toBeDefined(); expect(nvidia!.envVar).toBe('NVIDIA_API_KEY'); - expect(nvidia!.defaultModel).toBe('deepseek-ai/deepseek-v4-pro'); + expect(nvidia!.defaultModel).toBe('moonshotai/kimi-k2.5'); expect(nvidia!.defaultBaseUrl).toBe('https://integrate.api.nvidia.com/v1'); // Hosted (not local) — NVIDIA needs an API key, unlike ollama. expect(nvidia!.local).toBeUndefined();