Free => Lite mode in Codebuff!

Author: jahooma

agents/base2/base2-lite.ts

@@ -0,0 +1,8 @@
+import { createBase2 } from './base2'
+
+const definition = {
+  ...createBase2('lite'),
+  id: 'base2-lite',
+  displayName: 'Buffy the Lite Orchestrator',
+}
+export default definition

agents/base2/base2.ts

@@ -7,7 +7,7 @@ import {
 } from '../types/secret-agent-definition'
 
 export function createBase2(
-  mode: 'default' | 'free' | 'max' | 'fast',
+  mode: 'default' | 'free' | 'lite' | 'max' | 'fast',
   options?: {
     hasNoValidation?: boolean
     planOnly?: boolean
@@ -22,7 +22,7 @@ export function createBase2(
   const isDefault = mode === 'default'
   const isFast = mode === 'fast'
   const isMax = mode === 'max'
-  const isFree = mode === 'free'
+  const isFree = mode === 'free' || mode === 'lite'
 
   const isSonnet = false
   const model = isFree ? 'z-ai/glm-5.1' : 'anthropic/claude-opus-4.7'

cli/src/__tests__/rerender-perf.integration.test.ts

@@ -43,7 +43,7 @@ const RERENDER_THRESHOLDS = {
     'onToggleCollapsed',
     'onBuildFast',
     'onBuildMax',
-    'onBuildFree',
+    'onBuildLite',
     'onCloseFeedback',
   ],
 

cli/src/__tests__/unit/agent-mode-toggle.test.ts

@@ -17,7 +17,7 @@ describe('AgentModeToggle - buildExpandedSegments', () => {
   for (const mode of modes) {
     test(`returns segments with active indicator for ${mode}`, () => {
       const segs = buildExpandedSegments(mode)
-      // 4 mode options (DEFAULT, FREE, MAX, PLAN) + 1 active indicator
+      // 4 mode options (DEFAULT, LITE, MAX, PLAN) + 1 active indicator
       expect(segs.length).toBe(5)
 
       // Current mode is disabled among the choices

cli/src/chat.tsx

@@ -618,7 +618,7 @@ export const Chat = ({
     ],
   )
 
-  const { inputWidth, handleBuildFast, handleBuildMax, handleBuildFree } = useChatInput({
+  const { inputWidth, handleBuildFast, handleBuildMax, handleBuildLite } = useChatInput({
     setInputValue,
     agentMode,
     setAgentMode,
@@ -1246,15 +1246,15 @@ export const Chat = ({
       onToggleCollapsed: handleCollapseToggle,
       onBuildFast: handleBuildFast,
       onBuildMax: handleBuildMax,
-      onBuildFree: handleBuildFree,
+      onBuildLite: handleBuildLite,
       onFeedback: handleMessageFeedback,
       onCloseFeedback: handleCloseFeedback,
     })
   }, [
     handleCollapseToggle,
     handleBuildFast,
     handleBuildMax,
-    handleBuildFree,
+    handleBuildLite,
     handleMessageFeedback,
     handleCloseFeedback,
     setMessageBlockCallbacks,
@@ -1467,7 +1467,7 @@ export const Chat = ({
             <AdBanner
               ad={ad}
               onDisableAds={handleDisableAds}
-              isFreeMode={IS_FREEBUFF || agentMode === 'FREE'}
+              isFreeMode={IS_FREEBUFF}
             />
           )
         )}

cli/src/commands/ads.ts

@@ -38,15 +38,7 @@ export const handleAdsDisable = (): {
 export const getAdsEnabled = (): boolean => {
   if (IS_FREEBUFF) return true
 
-  // If no mode provided, get it from the store
-  const mode = useChatStore.getState().agentMode
-
-  // In FREE mode, ads are always enabled regardless of saved setting
-  if (mode === 'FREE') {
-    return true
-  }
-
-  // Otherwise, use the saved setting
+  // Codebuff LITE is a paid mode now, so use the normal saved setting.
   const settings = loadSettings()
   return settings.adsEnabled ?? false
 }

cli/src/components/__tests__/message-block.completion.test.tsx

@@ -46,7 +46,7 @@ const baseProps = {
   onToggleCollapsed: () => {},
   onBuildFast: () => {},
   onBuildMax: () => {},
-  onBuildFree: () => {},
+  onBuildLite: () => {},
   setCollapsedAgents: () => {},
   addAutoCollapsedAgent: () => {},
 }

cli/src/components/__tests__/message-block.streaming.test.tsx

@@ -42,7 +42,7 @@ const baseProps = {
   onToggleCollapsed: () => {},
   onBuildFast: () => {},
   onBuildMax: () => {},
-  onBuildFree: () => {},
+  onBuildLite: () => {},
   setCollapsedAgents: () => {},
   addAutoCollapsedAgent: () => {},
 }

cli/src/components/__tests__/message-with-agents.test.tsx

@@ -85,7 +85,7 @@ const defaultCallbacks = {
   onToggleCollapsed: () => {},
   onBuildFast: () => {},
   onBuildMax: () => {},
-  onBuildFree: () => {},
+  onBuildLite: () => {},
   onFeedback: () => {},
   onCloseFeedback: () => {},
 }
@@ -198,7 +198,7 @@ describe('MessageBlockStore', () => {
         onToggleCollapsed: mockToggle,
         onBuildFast: mockBuildFast,
         onBuildMax: mockBuildMax,
-        onBuildFree: mockBuildFree,
+        onBuildLite: mockBuildFree,
         onFeedback: mockFeedback,
         onCloseFeedback: mockCloseFeedback,
       })
@@ -207,7 +207,7 @@ describe('MessageBlockStore', () => {
       expect(state.callbacks.onToggleCollapsed).toBe(mockToggle)
       expect(state.callbacks.onBuildFast).toBe(mockBuildFast)
       expect(state.callbacks.onBuildMax).toBe(mockBuildMax)
-      expect(state.callbacks.onBuildFree).toBe(mockBuildFree)
+      expect(state.callbacks.onBuildLite).toBe(mockBuildFree)
       expect(state.callbacks.onFeedback).toBe(mockFeedback)
       expect(state.callbacks.onCloseFeedback).toBe(mockCloseFeedback)
     })
@@ -250,7 +250,7 @@ describe('MessageBlockStore', () => {
         onToggleCollapsed: mockFn,
         onBuildFast: mockFn,
         onBuildMax: mockFn,
-        onBuildFree: mockFn,
+        onBuildLite: mockFn,
         onFeedback: mockFn,
         onCloseFeedback: mockFn,
       })
@@ -261,7 +261,7 @@ describe('MessageBlockStore', () => {
       // Callbacks should be noop functions (not undefined)
       expect(typeof state.callbacks.onToggleCollapsed).toBe('function')
       expect(typeof state.callbacks.onBuildFast).toBe('function')
-      expect(typeof state.callbacks.onBuildFree).toBe('function')
+      expect(typeof state.callbacks.onBuildLite).toBe('function')
       // They should not throw when called
       expect(() => state.callbacks.onToggleCollapsed('test-id')).not.toThrow()
     })

cli/src/components/blocks/agent-branch-wrapper.tsx

@@ -87,7 +87,7 @@ interface AgentBodyProps {
   onToggleCollapsed: (id: string) => void
   onBuildFast: () => void
   onBuildMax: () => void
-  onBuildFree: () => void
+  onBuildLite: () => void
   isLastMessage?: boolean
 }
 
@@ -102,7 +102,7 @@ interface AgentBodyPropsRef {
   onToggleCollapsed: (id: string) => void
   onBuildFast: () => void
   onBuildMax: () => void
-  onBuildFree: () => void
+  onBuildLite: () => void
   isLastMessage?: boolean
   theme: ReturnType<typeof useTheme>
   getAgentMarkdownOptions: (indent: number) => {
@@ -121,7 +121,7 @@ const AgentBody = memo(
     onToggleCollapsed,
     onBuildFast,
     onBuildMax,
-    onBuildFree,
+    onBuildLite,
     isLastMessage,
   }: AgentBodyProps): ReactNode[] => {
     const theme = useTheme()
@@ -158,7 +158,7 @@ const AgentBody = memo(
       onToggleCollapsed,
       onBuildFast,
       onBuildMax,
-      onBuildFree,
+      onBuildLite,
       isLastMessage,
       theme,
       getAgentMarkdownOptions,
@@ -230,7 +230,7 @@ const AgentBody = memo(
                   onToggleCollapsed={p.onToggleCollapsed}
                   onBuildFast={p.onBuildFast}
                   onBuildMax={p.onBuildMax}
-                  onBuildFree={p.onBuildFree}
+                  onBuildLite={p.onBuildLite}
                   siblingBlocks={p.nestedBlocks}
                   isLastMessage={p.isLastMessage}
                 />
@@ -312,7 +312,7 @@ export interface AgentBranchWrapperProps {
   onToggleCollapsed: (id: string) => void
   onBuildFast: () => void
   onBuildMax: () => void
-  onBuildFree: () => void
+  onBuildLite: () => void
   siblingBlocks?: ContentBlock[]
   isLastMessage?: boolean
 }
@@ -326,7 +326,7 @@ export const AgentBranchWrapper = memo(
     onToggleCollapsed,
     onBuildFast,
     onBuildMax,
-    onBuildFree,
+    onBuildLite,
     siblingBlocks,
     isLastMessage,
   }: AgentBranchWrapperProps) => {
@@ -448,7 +448,7 @@ export const AgentBranchWrapper = memo(
             onToggleCollapsed={onToggleCollapsed}
             onBuildFast={onBuildFast}
             onBuildMax={onBuildMax}
-            onBuildFree={onBuildFree}
+            onBuildLite={onBuildLite}
             isLastMessage={isLastMessage}
           />
         </AgentBranchItem>