fix(web): allow unlimited plan models at zero balance (#7187)

* fix(web): skip zero-balance gate for unlimited models

* test(web): cover balance gate model argument

* fix(web): suppress low-balance warning for unlimited models

Author: Siri-Ray

apps/web/src/components/EntryShell.tsx

@@ -134,7 +134,11 @@ import type { OnboardingEntry } from '../onboarding/onboarding-entry';
 import type { PluginUseAction } from './plugins-home/useActions';
 import { Icon } from './Icon';
 import { Button } from '@open-design/components';
-import { defaultAgentModelId, effectiveAgentModelChoice } from './agentModelSelection';
+import {
+  defaultAgentModelId,
+  effectiveAgentModelChoice,
+  effectiveAgentModelId,
+} from './agentModelSelection';
 import { AgentIcon } from './AgentIcon';
 import { CommunityView } from './CommunityView';
 import { TeamSlotPlaceholder } from './TeamSlotPlaceholder';
@@ -1348,6 +1352,10 @@ export function EntryShell({
     let amrGatePrecheckWitness: AmrBalanceGateScope | undefined;
     let amrGatePrecheckPassed = false;
     if (config.mode === 'daemon' && config.agentId === 'amr') {
+      const amrModelId = effectiveAgentModelId(
+        agents.find((agent) => agent.id === 'amr'),
+        config.agentModels?.amr,
+      );
       // PRODUCT INVARIANT: Send never starts Workspace identity discovery.
       // Billing consumes the shell's current in-memory snapshot; if it has not
       // arrived yet, the existing account-scoped gate is used. The daemon's
@@ -1364,7 +1372,7 @@ export function EntryShell({
         const gateWorkspaceIdentity = workspaceIdentityCacheKey(gateWorkspaceContext);
         const gateScope = amrBalanceGateScopeForWorkspaceContext(gateWorkspaceContext);
         let gate = await retryUnavailableAmrBalanceGate(
-          () => checkAmrBalanceGate(gateScope),
+          () => checkAmrBalanceGate(gateScope, amrModelId),
         );
         // Hard blocks hold THIS submit open: the dialog resolves 'retry' when
         // its blocking condition clears (sign-in completed, recharge landed)
@@ -1383,7 +1391,7 @@ export function EntryShell({
           setAmrBalanceGateBlock(null);
           if (decision === 'dismiss') return 'blocked' as const;
           gate = await retryUnavailableAmrBalanceGate(
-            () => checkAmrBalanceGate(gateScope),
+            () => checkAmrBalanceGate(gateScope, amrModelId),
           );
         }
         if (gate.kind === 'unavailable') return false;

apps/web/src/components/ProjectView.tsx

@@ -333,7 +333,7 @@ import { buildContinueInCliToast } from '../lib/build-continue-in-cli-toast';
 import { buildClipboardPrompt } from '../lib/build-clipboard-prompt';
 import { copyToClipboard } from '../lib/copy-to-clipboard';
 import { effectiveMaxTokens } from '../state/maxTokens';
-import { effectiveAgentModelChoice } from './agentModelSelection';
+import { effectiveAgentModelChoice, effectiveAgentModelId } from './agentModelSelection';
 import { mediaExecutionPolicyForProjectMetadata } from '../media/execution-policy';
 import { mediaModelProviderId } from '../media/models';
 import { byokProviderRequiresApiKey } from '../utils/byokProvider';
@@ -6709,6 +6709,10 @@ export function ProjectView({
               persistedWorkspaceId.length > 0
               || projectWorkspaceScopeState.scope?.kind === 'unbound'
             );
+          const amrModelId = effectiveAgentModelId(
+            agentsById.get('amr'),
+            config.agentModels?.amr,
+          );
           const gate =
             deferAmrPreflightToDaemon
               ? { kind: 'allow' as const }
@@ -6721,6 +6725,7 @@ export function ProjectView({
                           projectRunPreflightContext.workspaceMemberId,
                       }
                     : undefined,
+                  amrModelId,
                 );
           // A blocked send parks in the conversation queue with its FULL
           // payload (prompt, attachments, comment context) — the composer

apps/web/src/components/agentModelSelection.ts

@@ -48,6 +48,16 @@ export function effectiveAgentModelChoice(
   return normalizeAgentModelChoice(agent, choice) ?? choice;
 }
 
+export function effectiveAgentModelId(
+  agent: AgentModelSource,
+  choice: AgentModelChoice | undefined,
+): string | null {
+  const configuredModel = effectiveAgentModelChoice(agent, choice)?.model?.trim();
+  return configuredModel && configuredModel !== 'default'
+    ? configuredModel
+    : defaultAgentModelId(agent);
+}
+
 /**
  * Whether `modelId` may be OFFERED to the user as a selectable model.
  *

apps/web/src/runtime/amr-balance-gate.ts

@@ -18,6 +18,8 @@ import type {
   WorkspaceBillingResponse,
 } from '@open-design/contracts';
 import { fetchAmrWalletSnapshot } from '../providers/daemon';
+import { resolveAmrPlan } from './amr-low-balance-plan';
+import { isUnlimitedAmrModelForPlan } from './amr-unlimited-models';
 
 /**
  * Hard-block line (USD): at or below this the wallet cannot fund any part of
@@ -229,6 +231,7 @@ async function fetchWorkspaceWalletSnapshot(
 
 async function checkWorkspaceBalanceGate(
   scope: AmrBalanceGateScope,
+  modelId?: string | null,
 ): Promise<AmrBalanceGateResult> {
   // The URL carries the selected workspace identity. The daemon authorizes
   // that exact directory membership and returns a v2 identity-stamped wallet.
@@ -259,6 +262,10 @@ async function checkWorkspaceBalanceGate(
   }
   const balance = amrWalletBalanceUsd(workspaceSnapshot);
   if (balance == null) return { kind: 'unavailable' };
+  if (balance <= AMR_LOW_BALANCE_WARN_USD && scope.workspaceType === 'personal') {
+    const plan = await resolveAmrPlan(workspaceSnapshot!);
+    if (isUnlimitedAmrModelForPlan(plan, modelId)) return { kind: 'allow' };
+  }
   if (balance <= AMR_HARD_BLOCK_BALANCE_USD) {
     return {
       kind: 'hard',
@@ -274,10 +281,11 @@ async function checkWorkspaceBalanceGate(
 
 export async function checkAmrBalanceGate(
   scope?: AmrBalanceGateScope,
+  modelId?: string | null,
 ): Promise<AmrBalanceGateResult> {
   try {
     if (scope) {
-      return await checkWorkspaceBalanceGate(scope);
+      return await checkWorkspaceBalanceGate(scope, modelId);
     }
     const cached = await fetchAmrWalletSnapshot().catch(() => null);
     const cachedBalance = amrWalletBalanceUsd(cached);
@@ -290,6 +298,8 @@ export async function checkAmrBalanceGate(
         return { kind: 'allow' };
       }
       // cached is non-null here: a definitive balance implies a snapshot.
+      const plan = await resolveAmrPlan(cached!);
+      if (isUnlimitedAmrModelForPlan(plan, modelId)) return { kind: 'allow' };
       return { kind: 'soft', snapshot: cached! };
     }
     // Hard-block candidate (signed out or empty): confirm against the live
@@ -309,6 +319,10 @@ export async function checkAmrBalanceGate(
     if (fresh.stale || fresh.error != null) return { kind: 'allow' };
     const freshBalance = amrWalletBalanceUsd(fresh);
     if (freshBalance == null) return { kind: 'allow' };
+    if (freshBalance <= AMR_LOW_BALANCE_WARN_USD) {
+      const plan = await resolveAmrPlan(fresh);
+      if (isUnlimitedAmrModelForPlan(plan, modelId)) return { kind: 'allow' };
+    }
     if (freshBalance <= AMR_HARD_BLOCK_BALANCE_USD) {
       return { kind: 'hard', reason: 'insufficient', snapshot: fresh };
     }

apps/web/src/runtime/amr-unlimited-models.ts

@@ -0,0 +1,46 @@
+const GO_UNLIMITED_MODELS = [
+  'deepseek-v4-flash',
+  'deepseek-v4-pro',
+  'glm-5.2',
+] as const;
+
+const PLUS_UNLIMITED_MODELS = [
+  ...GO_UNLIMITED_MODELS,
+  'kimi-k2.7-code',
+] as const;
+
+const PRO_UNLIMITED_MODELS = [
+  'deepseek-v4-flash',
+  'deepseek-v4-pro',
+  'glm-5.2',
+  'kimi-k2.7-code',
+  'mimo-v2.5-pro',
+] as const;
+
+const MAX_UNLIMITED_MODELS = [
+  ...PRO_UNLIMITED_MODELS,
+  'minimax-m2.7',
+  'kimi-k2.6',
+  'glm-5.1',
+] as const;
+
+// This table only decides whether the client-side balance preflight may
+// stand down. Vela remains authoritative for plan access and usage limits.
+const UNLIMITED_MODELS_BY_PLAN: Readonly<Record<string, ReadonlySet<string>>> = {
+  go: new Set(GO_UNLIMITED_MODELS),
+  plus: new Set(PLUS_UNLIMITED_MODELS),
+  pro: new Set(PRO_UNLIMITED_MODELS),
+  max: new Set(MAX_UNLIMITED_MODELS),
+};
+
+function normalize(value: string | null | undefined): string {
+  return value?.trim().toLowerCase() ?? '';
+}
+
+export function isUnlimitedAmrModelForPlan(
+  plan: string | null | undefined,
+  modelId: string | null | undefined,
+): boolean {
+  const models = UNLIMITED_MODELS_BY_PLAN[normalize(plan)];
+  return models?.has(normalize(modelId)) ?? false;
+}

apps/web/tests/components/EntryShell.amr-workspace-race.test.tsx

@@ -310,7 +310,9 @@ describe('EntryShell AMR workspace precheck race', () => {
     setHomeHeroPrompt('Create a poster after I sign in.');
     fireEvent.click(await screen.findByTestId('home-hero-submit'));
 
-    await waitFor(() => expect(mockedCheckAmrBalanceGate).toHaveBeenCalledWith(undefined));
+    await waitFor(() => {
+      expect(mockedCheckAmrBalanceGate).toHaveBeenCalledWith(undefined, 'glm-5');
+    });
     const dialog = await screen.findByTestId('amr-balance-dialog');
     expect(dialog.getAttribute('data-reason')).toBe('signed_out');
     expect(onCreateProject).not.toHaveBeenCalled();
@@ -393,7 +395,9 @@ describe('EntryShell AMR workspace precheck race', () => {
     setHomeHeroPrompt('Create through the old daemon compatibility lane.');
     fireEvent.click(await screen.findByTestId('home-hero-submit'));
 
-    await waitFor(() => expect(mockedCheckAmrBalanceGate).toHaveBeenCalledWith(undefined));
+    await waitFor(() => {
+      expect(mockedCheckAmrBalanceGate).toHaveBeenCalledWith(undefined, 'glm-5');
+    });
     await waitFor(() => expect(onCreateProject).toHaveBeenCalledTimes(1));
   });
 
@@ -890,7 +894,7 @@ describe('EntryShell AMR workspace precheck race', () => {
         workspaceType: 'team',
         workspaceId: 'workspace-a',
         workspaceMemberId: 'member-a',
-      });
+      }, 'glm-5');
     });
 
     currentContext = workspaceB;
@@ -905,7 +909,7 @@ describe('EntryShell AMR workspace precheck race', () => {
         workspaceType: 'team',
         workspaceId: 'workspace-b',
         workspaceMemberId: 'member-b',
-      });
+      }, 'glm-5');
     });
     await waitFor(() => expect(onCreateProject).toHaveBeenCalledTimes(1));
     expect(onCreateProject).toHaveBeenCalledWith(

apps/web/tests/components/ProjectView.run-workspace-identity.test.tsx

@@ -412,7 +412,16 @@ function projectViewElement(overrides: Partial<ComponentProps<typeof ProjectView
       project={project()}
       routeFileName={null}
       config={config}
-      agents={[{ id: 'amr', name: 'amr', available: true }] as unknown as AgentInfo[]}
+      agents={[{
+        id: 'amr',
+        name: 'amr',
+        available: true,
+        models: [{
+          id: 'deepseek-v4-flash',
+          label: 'DeepSeek V4 Flash',
+          default: true,
+        }],
+      }] as unknown as AgentInfo[]}
       skills={[] as SkillSummary[]}
       designTemplates={[] as SkillSummary[]}
       designSystems={[] as DesignSystemSummary[]}
@@ -502,7 +511,7 @@ describe('a Home auto-send identifies its caller before the project scope resolv
       workspaceType: 'team',
       workspaceId: TEAM_WORKSPACE,
       workspaceMemberId: TEAM_MEMBER,
-    });
+    }, 'deepseek-v4-flash');
     const options = mockedStreamViaDaemon.mock.calls[0]?.[0];
     expect(
       options?.workspaceContext,
@@ -997,7 +1006,7 @@ describe('a Home auto-send identifies its caller before the project scope resolv
       workspaceType: 'team',
       workspaceId: TEAM_WORKSPACE,
       workspaceMemberId: TEAM_MEMBER,
-    });
+    }, 'deepseek-v4-flash');
     expect(mockedStreamViaDaemon.mock.calls[0]?.[0].workspaceContext).toEqual(
       CALLER_CONTEXT,
     );
@@ -1018,7 +1027,16 @@ describe('a Home auto-send identifies its caller before the project scope resolv
 
     const stableOverrides: Partial<ComponentProps<typeof ProjectView>> = {
       project: project(),
-      agents: [{ id: 'amr', name: 'amr', available: true }] as unknown as AgentInfo[],
+      agents: [{
+        id: 'amr',
+        name: 'amr',
+        available: true,
+        models: [{
+          id: 'deepseek-v4-flash',
+          label: 'DeepSeek V4 Flash',
+          default: true,
+        }],
+      }] as unknown as AgentInfo[],
       skills: [] as SkillSummary[],
       designTemplates: [] as SkillSummary[],
       designSystems: [] as DesignSystemSummary[],
@@ -1080,7 +1098,7 @@ describe('a Home auto-send identifies its caller before the project scope resolv
         workspaceType: 'personal',
         workspaceId: PERSONAL_CONTEXT.workspaceId,
         workspaceMemberId: PERSONAL_CONTEXT.workspaceMemberId,
-      });
+      }, 'deepseek-v4-flash');
     });
     await waitFor(() => expect(mockedStreamViaDaemon).toHaveBeenCalled());
   });
@@ -1109,7 +1127,7 @@ describe('a Home auto-send identifies its caller before the project scope resolv
         workspaceType: 'personal',
         workspaceId: PERSONAL_CONTEXT.workspaceId,
         workspaceMemberId: PERSONAL_CONTEXT.workspaceMemberId,
-      });
+      }, 'deepseek-v4-flash');
     });
     await waitFor(() => expect(mockedStreamViaDaemon).toHaveBeenCalled());
     expect(mockedStreamViaDaemon.mock.calls[0]?.[0].workspaceContext).toEqual(
@@ -1205,7 +1223,16 @@ describe('a Home auto-send observes a project billing scope that settles after m
     // only dependency allowed to change in this regression.
     const stableOverrides: Partial<ComponentProps<typeof ProjectView>> = {
       project: project(),
-      agents: [{ id: 'amr', name: 'amr', available: true }] as unknown as AgentInfo[],
+      agents: [{
+        id: 'amr',
+        name: 'amr',
+        available: true,
+        models: [{
+          id: 'deepseek-v4-flash',
+          label: 'DeepSeek V4 Flash',
+          default: true,
+        }],
+      }] as unknown as AgentInfo[],
       skills: [] as SkillSummary[],
       designTemplates: [] as SkillSummary[],
       designSystems: [] as DesignSystemSummary[],
@@ -1249,7 +1276,7 @@ describe('a Home auto-send observes a project billing scope that settles after m
         workspaceType: 'team',
         workspaceId: TEAM_WORKSPACE,
         workspaceMemberId: TEAM_MEMBER,
-      });
+      }, 'deepseek-v4-flash');
     });
     await waitFor(() => expect(mockedStreamViaDaemon).toHaveBeenCalled());
   });

apps/web/tests/components/agentModelSelection.test.ts

@@ -3,6 +3,7 @@ import {
   agentModelIsSelectable,
   defaultAgentModelId,
   effectiveAgentModelChoice,
+  effectiveAgentModelId,
   normalizeAgentModelChoice,
 } from '../../src/components/agentModelSelection';
 import type { AgentInfo } from '../../src/types';
@@ -61,6 +62,7 @@ describe('agent model selection', () => {
 
     expect(normalizeAgentModelChoice(amrAgent, choice)).toBeNull();
     expect(effectiveAgentModelChoice(amrAgent, choice)).toEqual(choice);
+    expect(effectiveAgentModelId(amrAgent, choice)).toBe('glm-5');
   });
 
   it('does not select a disabled model as the AMR default when every catalog row is locked', () => {