From 5159cc0b6a981aa7eafd3e65fabddc58c248f86f Mon Sep 17 00:00:00 2001
From: Alex <yanglongwei06@gmail.com>
Date: Mon, 18 May 2026 18:16:43 +0800
Subject: [PATCH 1/2] feat: add Novita AI provider

---
 .../components/settings/panels/AIPanel.tsx    | 37 ++++++---
 .../panels/__tests__/AIPanel.test.tsx         | 82 +++++++++++++++++++
 2 files changed, 109 insertions(+), 10 deletions(-)

diff --git a/app/src/components/settings/panels/AIPanel.tsx b/app/src/components/settings/panels/AIPanel.tsx
index 0ab696d18a..4de18ee0f0 100644
--- a/app/src/components/settings/panels/AIPanel.tsx
+++ b/app/src/components/settings/panels/AIPanel.tsx
@@ -103,12 +103,22 @@ const BUILTIN_PROVIDER_META: Record<string, { tone: string; label: string }> = {
     label: 'OpenRouter',
     tone: 'bg-slate-100 dark:bg-slate-500/15 ring-slate-300 text-slate-900 dark:text-slate-100',
   },
+  novita: {
+    label: 'Novita AI',
+    tone: 'bg-cyan-50 dark:bg-cyan-500/10 ring-cyan-200 text-cyan-900 dark:text-cyan-100',
+  },
   custom: {
     label: 'Custom',
     tone: 'bg-stone-100 dark:bg-neutral-800 ring-stone-300 text-stone-900 dark:text-neutral-100',
   },
 };
 
+const BUILTIN_CLOUD_SLUGS = ['openai', 'anthropic', 'openrouter', 'novita', 'custom'] as const;
+
+const BUILTIN_PROVIDER_DEFAULT_MODELS: Record<string, string> = {
+  novita: 'deepseek/deepseek-v4-pro',
+};
+
 const WORKLOADS: Workload[] = [
   {
     id: 'reasoning',
@@ -451,7 +461,9 @@ const ProviderKeyDialog = ({
         ? 'sk-ant-...'
         : slug === 'openrouter'
           ? 'sk-or-...'
-          : 'your-api-key';
+          : slug === 'novita'
+            ? 'Novita API key'
+            : 'your-api-key';
 
   const handleSave = async () => {
     const trimmed = apiKey.trim();
@@ -1515,7 +1527,7 @@ const CustomRoutingDialog = ({
     if (initial.kind === 'cloud' || initial.kind === 'local') return initial.model;
     if (initialSource?.kind === 'cloud') {
       const p = customCloud.find(c => c.slug === initialSource.providerSlug);
-      return p ? '' : '';
+      return p ? (BUILTIN_PROVIDER_DEFAULT_MODELS[p.slug] ?? '') : '';
     }
     return localModels[0]?.id ?? '';
   });
@@ -1591,7 +1603,7 @@ const CustomRoutingDialog = ({
                     setModel(localModels[0]?.id ?? '');
                   } else if (kind === 'cloud') {
                     setSource({ kind: 'cloud', providerSlug: slug });
-                    setModel('');
+                    setModel(BUILTIN_PROVIDER_DEFAULT_MODELS[slug] ?? '');
                   }
                 }}
                 className="rounded-lg border border-stone-300 dark:border-neutral-700 bg-white dark:bg-neutral-900 px-3 py-2 text-sm text-stone-900 dark:text-neutral-100 focus:border-primary-500 focus:outline-none focus:ring-1 focus:ring-primary-500">
@@ -1624,7 +1636,12 @@ const CustomRoutingDialog = ({
                   type="text"
                   value={model}
                   onChange={e => setModel(e.target.value)}
-                  placeholder={selectedCloud ? `${selectedCloud.slug} model id` : 'model-id'}
+                  placeholder={
+                    selectedCloud
+                      ? (BUILTIN_PROVIDER_DEFAULT_MODELS[selectedCloud.slug] ??
+                        `${selectedCloud.slug} model id`)
+                      : 'model-id'
+                  }
                   className="rounded-lg border border-stone-300 dark:border-neutral-700 bg-white dark:bg-neutral-900 px-3 py-2 text-sm font-mono text-stone-900 dark:text-neutral-100 placeholder-stone-400 dark:placeholder-neutral-500 focus:border-primary-500 focus:outline-none focus:ring-1 focus:ring-primary-500"
                 />
               )}
@@ -1797,8 +1814,8 @@ const AIPanel = ({ embedded = false }: AIPanelProps = {}) => {
             )}
 
             <div className="flex flex-wrap gap-2">
-              {/* Built-in cloud providers — openai/anthropic/openrouter/custom */}
-              {(['openai', 'anthropic', 'openrouter', 'custom'] as const).map(slug => {
+              {/* Built-in cloud providers — openai/anthropic/openrouter/novita/custom */}
+              {BUILTIN_CLOUD_SLUGS.map(slug => {
                 const meta = BUILTIN_PROVIDER_META[slug];
                 const label = meta?.label ?? slug;
                 const existing = draft.cloudProviders.find(cp => cp.slug === slug);
@@ -2112,9 +2129,7 @@ const CloudProviderEditor = ({
   const { t } = useT();
   const defaultSlug: string =
     initial?.slug ??
-    (['openai', 'anthropic', 'openrouter', 'custom'] as const).find(
-      s => !existingSlugs.includes(s)
-    ) ??
+    BUILTIN_CLOUD_SLUGS.find(s => !existingSlugs.includes(s)) ??
     'custom';
   const [slug, setSlug] = useState<string>(defaultSlug);
   const [label, setLabel] = useState<string>(
@@ -2157,7 +2172,7 @@ const CloudProviderEditor = ({
               }}
               disabled={!!initial}
               className="mt-1 w-full rounded-lg border border-stone-200 dark:border-neutral-800 bg-white dark:bg-neutral-900 px-3 py-2 text-sm text-stone-900 dark:text-neutral-100 disabled:opacity-60 focus:border-primary-400 focus:outline-none focus:ring-1 focus:ring-primary-200">
-              {(['openai', 'anthropic', 'openrouter', 'custom'] as const)
+              {BUILTIN_CLOUD_SLUGS
                 .filter(s => s === slug || !existingSlugs.includes(s))
                 .map(s => (
                   <option key={s} value={s}>
@@ -2261,6 +2276,8 @@ function defaultEndpointFor(slug: string): string {
       return 'https://api.anthropic.com/v1';
     case 'openrouter':
       return 'https://openrouter.ai/api/v1';
+    case 'novita':
+      return 'https://api.novita.ai/openai';
     default:
       return '';
   }
diff --git a/app/src/components/settings/panels/__tests__/AIPanel.test.tsx b/app/src/components/settings/panels/__tests__/AIPanel.test.tsx
index 7b857c99a2..a6034e7b00 100644
--- a/app/src/components/settings/panels/__tests__/AIPanel.test.tsx
+++ b/app/src/components/settings/panels/__tests__/AIPanel.test.tsx
@@ -298,6 +298,88 @@ describe('AIPanel', () => {
     expect(screen.getByLabelText(/API key/i)).toBeInTheDocument();
   });
 
+  it('connects Novita AI with the built-in OpenAI-compatible endpoint', async () => {
+    vi.mocked(loadAISettings).mockResolvedValue({ ...baseSettings, cloudProviders: [] });
+    vi.mocked(setCloudProviderKey).mockResolvedValue(undefined);
+    vi.mocked(saveAISettings).mockResolvedValue(undefined);
+
+    renderWithProviders(<AIPanel />);
+    await waitFor(() =>
+      expect(screen.getByRole('switch', { name: /Connect Novita AI/i })).toBeInTheDocument()
+    );
+
+    fireEvent.click(screen.getByRole('switch', { name: /Connect Novita AI/i }));
+    await waitFor(() =>
+      expect(screen.getByRole('dialog', { name: /Connect Novita AI/i })).toBeInTheDocument()
+    );
+
+    fireEvent.change(screen.getByLabelText(/API key/i), {
+      target: { value: 'novita-test-key' },
+    });
+    fireEvent.click(screen.getByRole('button', { name: /^Save$/i }));
+
+    await waitFor(() =>
+      expect(screen.getByRole('switch', { name: /Disconnect Novita AI/i })).toBeInTheDocument()
+    );
+    expect(vi.mocked(setCloudProviderKey)).toHaveBeenCalledWith('novita', 'novita-test-key');
+
+    fireEvent.click(screen.getByRole('button', { name: /^Save$/i }));
+    await waitFor(() => expect(vi.mocked(saveAISettings)).toHaveBeenCalled());
+
+    const [, nextSettings] = vi.mocked(saveAISettings).mock.calls[0];
+    const novitaProvider = nextSettings.cloudProviders.find(
+      (p: { slug: string }) => p.slug === 'novita'
+    );
+    expect(novitaProvider).toMatchObject({
+      slug: 'novita',
+      label: 'Novita AI',
+      endpoint: 'https://api.novita.ai/openai',
+      auth_style: 'bearer',
+      has_api_key: true,
+    });
+  });
+
+  it('prefills the Novita default model when routing a workload to Novita AI', async () => {
+    const settingsWithNovita = {
+      cloudProviders: [
+        {
+          id: 'p_novita_1',
+          slug: 'novita',
+          label: 'Novita AI',
+          endpoint: 'https://api.novita.ai/openai',
+          auth_style: 'bearer' as const,
+          has_api_key: true,
+        },
+      ],
+      routing: baseSettings.routing,
+    };
+    vi.mocked(loadAISettings).mockResolvedValue(settingsWithNovita);
+    vi.mocked(saveAISettings).mockResolvedValue(undefined);
+
+    renderWithProviders(<AIPanel />);
+    await waitFor(() => expect(screen.getAllByText(/Novita AI/i).length).toBeGreaterThan(0));
+
+    const reasoningRow = screen.getByText('Reasoning').parentElement!.parentElement!;
+    fireEvent.click(within(reasoningRow).getAllByRole('button')[1]);
+
+    await waitFor(() =>
+      expect(screen.getByDisplayValue('deepseek/deepseek-v4-pro')).toBeInTheDocument()
+    );
+    const dialog = screen.getByRole('dialog', { name: /Custom routing for Reasoning/i });
+    fireEvent.click(within(dialog).getByRole('button', { name: /^Save$/i }));
+
+    await waitFor(() => expect(screen.getByText(/unsaved change/i)).toBeInTheDocument());
+    fireEvent.click(screen.getByRole('button', { name: /^Save$/i }));
+    await waitFor(() => expect(vi.mocked(saveAISettings)).toHaveBeenCalled());
+
+    const [, nextSettings] = vi.mocked(saveAISettings).mock.calls[0];
+    expect(nextSettings.routing.reasoning).toEqual({
+      kind: 'cloud',
+      providerSlug: 'novita',
+      model: 'deepseek/deepseek-v4-pro',
+    });
+  });
+
   it('clicking the Custom chip (when disabled) opens the CloudProviderEditor, not the key dialog', async () => {
     // Load with no custom provider → chip is off.
     vi.mocked(loadAISettings).mockResolvedValue({ ...baseSettings, cloudProviders: [] });

From 67bc76520561b4fc8a565d990a094b1b8ff5904a Mon Sep 17 00:00:00 2001
From: Alex-wuhu <yanglongwei06@gmail.com>
Date: Mon, 18 May 2026 23:22:50 +0800
Subject: [PATCH 2/2] fix: keep custom provider editor default

---
 .../components/settings/panels/AIPanel.tsx    | 10 +++--
 .../panels/__tests__/AIPanel.test.tsx         | 43 +++++++++++++++++++
 2 files changed, 49 insertions(+), 4 deletions(-)

diff --git a/app/src/components/settings/panels/AIPanel.tsx b/app/src/components/settings/panels/AIPanel.tsx
index 4de18ee0f0..48c42be207 100644
--- a/app/src/components/settings/panels/AIPanel.tsx
+++ b/app/src/components/settings/panels/AIPanel.tsx
@@ -114,6 +114,7 @@ const BUILTIN_PROVIDER_META: Record<string, { tone: string; label: string }> = {
 };
 
 const BUILTIN_CLOUD_SLUGS = ['openai', 'anthropic', 'openrouter', 'novita', 'custom'] as const;
+const CLOUD_EDITOR_SLUGS = ['openai', 'anthropic', 'openrouter', 'custom'] as const;
 
 const BUILTIN_PROVIDER_DEFAULT_MODELS: Record<string, string> = {
   novita: 'deepseek/deepseek-v4-pro',
@@ -2128,9 +2129,7 @@ const CloudProviderEditor = ({
 }) => {
   const { t } = useT();
   const defaultSlug: string =
-    initial?.slug ??
-    BUILTIN_CLOUD_SLUGS.find(s => !existingSlugs.includes(s)) ??
-    'custom';
+    initial?.slug ?? CLOUD_EDITOR_SLUGS.find(s => !existingSlugs.includes(s)) ?? 'custom';
   const [slug, setSlug] = useState<string>(defaultSlug);
   const [label, setLabel] = useState<string>(
     initial?.label ?? BUILTIN_PROVIDER_META[defaultSlug]?.label ?? defaultSlug
@@ -2140,6 +2139,9 @@ const CloudProviderEditor = ({
   const [saving, setSaving] = useState(false);
   const isOpenHuman = slug === 'openhuman';
   const hasExistingKey = (initial?.maskedKey ?? '').startsWith('••••');
+  const editorSlugOptions = CLOUD_EDITOR_SLUGS.includes(slug as (typeof CLOUD_EDITOR_SLUGS)[number])
+    ? CLOUD_EDITOR_SLUGS
+    : ([slug, ...CLOUD_EDITOR_SLUGS] as const);
 
   return (
     <div className="fixed inset-0 z-50 flex items-center justify-center bg-stone-900/30 p-4">
@@ -2172,7 +2174,7 @@ const CloudProviderEditor = ({
               }}
               disabled={!!initial}
               className="mt-1 w-full rounded-lg border border-stone-200 dark:border-neutral-800 bg-white dark:bg-neutral-900 px-3 py-2 text-sm text-stone-900 dark:text-neutral-100 disabled:opacity-60 focus:border-primary-400 focus:outline-none focus:ring-1 focus:ring-primary-200">
-              {BUILTIN_CLOUD_SLUGS
+              {editorSlugOptions
                 .filter(s => s === slug || !existingSlugs.includes(s))
                 .map(s => (
                   <option key={s} value={s}>
diff --git a/app/src/components/settings/panels/__tests__/AIPanel.test.tsx b/app/src/components/settings/panels/__tests__/AIPanel.test.tsx
index a6034e7b00..f23759198a 100644
--- a/app/src/components/settings/panels/__tests__/AIPanel.test.tsx
+++ b/app/src/components/settings/panels/__tests__/AIPanel.test.tsx
@@ -397,6 +397,49 @@ describe('AIPanel', () => {
     expect(screen.queryByRole('dialog', { name: /Connect Custom/i })).not.toBeInTheDocument();
   });
 
+  it('keeps the Custom editor default on custom when Novita AI is the only unconfigured built-in provider', async () => {
+    vi.mocked(loadAISettings).mockResolvedValue({
+      ...baseSettings,
+      cloudProviders: [
+        {
+          id: 'p_openai_1',
+          slug: 'openai',
+          label: 'OpenAI',
+          endpoint: 'https://api.openai.com/v1',
+          auth_style: 'bearer' as const,
+          has_api_key: true,
+        },
+        {
+          id: 'p_anthropic_1',
+          slug: 'anthropic',
+          label: 'Anthropic',
+          endpoint: 'https://api.anthropic.com/v1',
+          auth_style: 'anthropic' as const,
+          has_api_key: true,
+        },
+        {
+          id: 'p_openrouter_1',
+          slug: 'openrouter',
+          label: 'OpenRouter',
+          endpoint: 'https://openrouter.ai/api/v1',
+          auth_style: 'bearer' as const,
+          has_api_key: true,
+        },
+      ],
+    });
+
+    renderWithProviders(<AIPanel />);
+    await waitFor(() =>
+      expect(screen.getByRole('switch', { name: /Connect Custom/i })).toBeInTheDocument()
+    );
+
+    fireEvent.click(screen.getByRole('switch', { name: /Connect Custom/i }));
+
+    await waitFor(() => expect(screen.getByText(/Add cloud provider/i)).toBeInTheDocument());
+    expect(screen.getByDisplayValue('Custom')).toBeInTheDocument();
+    expect(screen.getByPlaceholderText('https://api.example.com/v1')).toHaveValue('');
+  });
+
   // ─── chip toggle: toggle OFF scrubs routing entries ──────────────────────────
 
   it('toggling OFF an enabled provider scrubs routing entries that reference it', async () => {