Skip to content

Commit 7e8fe65

Browse files
jahoomajahooma
committed
Add build free button in plan mode
1 parent 30d3216 commit 7e8fe65

14 files changed

+79
-3
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ const RERENDER_THRESHOLDS = {
4343
'onToggleCollapsed',
4444
'onBuildFast',
4545
'onBuildMax',
46+
'onBuildFree',
4647
'onCloseFeedback',
4748
],
4849

‎cli/src/chat.tsx‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,7 @@ export const Chat = ({
611611
],
612612
)
613613

614-
const { inputWidth, handleBuildFast, handleBuildMax } = useChatInput({
614+
const { inputWidth, handleBuildFast, handleBuildMax, handleBuildFree } = useChatInput({
615615
setInputValue,
616616
agentMode,
617617
setAgentMode,
@@ -1230,13 +1230,15 @@ export const Chat = ({
12301230
onToggleCollapsed: handleCollapseToggle,
12311231
onBuildFast: handleBuildFast,
12321232
onBuildMax: handleBuildMax,
1233+
onBuildFree: handleBuildFree,
12331234
onFeedback: handleMessageFeedback,
12341235
onCloseFeedback: handleCloseFeedback,
12351236
})
12361237
}, [
12371238
handleCollapseToggle,
12381239
handleBuildFast,
12391240
handleBuildMax,
1241+
handleBuildFree,
12401242
handleMessageFeedback,
12411243
handleCloseFeedback,
12421244
setMessageBlockCallbacks,

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ const baseProps = {
4646
onToggleCollapsed: () => {},
4747
onBuildFast: () => {},
4848
onBuildMax: () => {},
49+
onBuildFree: () => {},
4950
setCollapsedAgents: () => {},
5051
addAutoCollapsedAgent: () => {},
5152
}

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ const baseProps = {
4242
onToggleCollapsed: () => {},
4343
onBuildFast: () => {},
4444
onBuildMax: () => {},
45+
onBuildFree: () => {},
4546
setCollapsedAgents: () => {},
4647
addAutoCollapsedAgent: () => {},
4748
}

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ const defaultCallbacks = {
8585
onToggleCollapsed: () => {},
8686
onBuildFast: () => {},
8787
onBuildMax: () => {},
88+
onBuildFree: () => {},
8889
onFeedback: () => {},
8990
onCloseFeedback: () => {},
9091
}
@@ -189,13 +190,15 @@ describe('MessageBlockStore', () => {
189190
const mockToggle = () => {}
190191
const mockBuildFast = () => {}
191192
const mockBuildMax = () => {}
193+
const mockBuildFree = () => {}
192194
const mockFeedback = () => {}
193195
const mockCloseFeedback = () => {}
194196

195197
useMessageBlockStore.getState().setCallbacks({
196198
onToggleCollapsed: mockToggle,
197199
onBuildFast: mockBuildFast,
198200
onBuildMax: mockBuildMax,
201+
onBuildFree: mockBuildFree,
199202
onFeedback: mockFeedback,
200203
onCloseFeedback: mockCloseFeedback,
201204
})
@@ -204,6 +207,7 @@ describe('MessageBlockStore', () => {
204207
expect(state.callbacks.onToggleCollapsed).toBe(mockToggle)
205208
expect(state.callbacks.onBuildFast).toBe(mockBuildFast)
206209
expect(state.callbacks.onBuildMax).toBe(mockBuildMax)
210+
expect(state.callbacks.onBuildFree).toBe(mockBuildFree)
207211
expect(state.callbacks.onFeedback).toBe(mockFeedback)
208212
expect(state.callbacks.onCloseFeedback).toBe(mockCloseFeedback)
209213
})
@@ -246,6 +250,7 @@ describe('MessageBlockStore', () => {
246250
onToggleCollapsed: mockFn,
247251
onBuildFast: mockFn,
248252
onBuildMax: mockFn,
253+
onBuildFree: mockFn,
249254
onFeedback: mockFn,
250255
onCloseFeedback: mockFn,
251256
})
@@ -256,6 +261,7 @@ describe('MessageBlockStore', () => {
256261
// Callbacks should be noop functions (not undefined)
257262
expect(typeof state.callbacks.onToggleCollapsed).toBe('function')
258263
expect(typeof state.callbacks.onBuildFast).toBe('function')
264+
expect(typeof state.callbacks.onBuildFree).toBe('function')
259265
// They should not throw when called
260266
expect(() => state.callbacks.onToggleCollapsed('test-id')).not.toThrow()
261267
})

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ interface AgentBodyProps {
8888
onToggleCollapsed: (id: string) => void
8989
onBuildFast: () => void
9090
onBuildMax: () => void
91+
onBuildFree: () => void
9192
isLastMessage?: boolean
9293
}
9394

@@ -102,6 +103,7 @@ interface AgentBodyPropsRef {
102103
onToggleCollapsed: (id: string) => void
103104
onBuildFast: () => void
104105
onBuildMax: () => void
106+
onBuildFree: () => void
105107
isLastMessage?: boolean
106108
theme: ReturnType<typeof useTheme>
107109
getAgentMarkdownOptions: (indent: number) => {
@@ -120,6 +122,7 @@ const AgentBody = memo(
120122
onToggleCollapsed,
121123
onBuildFast,
122124
onBuildMax,
125+
onBuildFree,
123126
isLastMessage,
124127
}: AgentBodyProps): ReactNode[] => {
125128
const theme = useTheme()
@@ -156,6 +159,7 @@ const AgentBody = memo(
156159
onToggleCollapsed,
157160
onBuildFast,
158161
onBuildMax,
162+
onBuildFree,
159163
isLastMessage,
160164
theme,
161165
getAgentMarkdownOptions,
@@ -227,6 +231,7 @@ const AgentBody = memo(
227231
onToggleCollapsed={p.onToggleCollapsed}
228232
onBuildFast={p.onBuildFast}
229233
onBuildMax={p.onBuildMax}
234+
onBuildFree={p.onBuildFree}
230235
siblingBlocks={p.nestedBlocks}
231236
isLastMessage={p.isLastMessage}
232237
/>
@@ -313,6 +318,7 @@ export interface AgentBranchWrapperProps {
313318
onToggleCollapsed: (id: string) => void
314319
onBuildFast: () => void
315320
onBuildMax: () => void
321+
onBuildFree: () => void
316322
siblingBlocks?: ContentBlock[]
317323
isLastMessage?: boolean
318324
}
@@ -326,6 +332,7 @@ export const AgentBranchWrapper = memo(
326332
onToggleCollapsed,
327333
onBuildFast,
328334
onBuildMax,
335+
onBuildFree,
329336
siblingBlocks,
330337
isLastMessage,
331338
}: AgentBranchWrapperProps) => {
@@ -448,6 +455,7 @@ export const AgentBranchWrapper = memo(
448455
onToggleCollapsed={onToggleCollapsed}
449456
onBuildFast={onBuildFast}
450457
onBuildMax={onBuildMax}
458+
onBuildFree={onBuildFree}
451459
isLastMessage={isLastMessage}
452460
/>
453461
</AgentBranchItem>

‎cli/src/components/blocks/blocks-renderer.tsx‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ interface BlocksRendererProps {
2424
onToggleCollapsed: (id: string) => void
2525
onBuildFast: () => void
2626
onBuildMax: () => void
27+
onBuildFree: () => void
2728
isLastMessage?: boolean
2829
contentToCopy?: string
2930
}
@@ -41,6 +42,7 @@ interface BlocksRendererPropsRef {
4142
onToggleCollapsed: (id: string) => void
4243
onBuildFast: () => void
4344
onBuildMax: () => void
45+
onBuildFree: () => void
4446
isLastMessage?: boolean
4547
contentToCopy?: string
4648
lastTextBlockIndex: number
@@ -59,6 +61,7 @@ export const BlocksRenderer = memo(
5961
onToggleCollapsed,
6062
onBuildFast,
6163
onBuildMax,
64+
onBuildFree,
6265
isLastMessage,
6366
contentToCopy,
6467
}: BlocksRendererProps) => {
@@ -84,6 +87,7 @@ export const BlocksRenderer = memo(
8487
onToggleCollapsed,
8588
onBuildFast,
8689
onBuildMax,
90+
onBuildFree,
8791
isLastMessage,
8892
contentToCopy,
8993
lastTextBlockIndex,
@@ -163,6 +167,7 @@ export const BlocksRenderer = memo(
163167
onToggleCollapsed={p.onToggleCollapsed}
164168
onBuildFast={p.onBuildFast}
165169
onBuildMax={p.onBuildMax}
170+
onBuildFree={p.onBuildFree}
166171
siblingBlocks={p.sourceBlocks}
167172
isLastMessage={p.isLastMessage}
168173
/>
@@ -189,6 +194,7 @@ export const BlocksRenderer = memo(
189194
onToggleCollapsed={p.onToggleCollapsed}
190195
onBuildFast={p.onBuildFast}
191196
onBuildMax={p.onBuildMax}
197+
onBuildFree={p.onBuildFree}
192198
isLastMessage={p.isLastMessage}
193199
contentToCopy={index === p.lastTextBlockIndex ? p.contentToCopy : undefined}
194200
/>

‎cli/src/components/blocks/single-block.tsx‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ interface SingleBlockProps {
3333
onToggleCollapsed: (id: string) => void
3434
onBuildFast: () => void
3535
onBuildMax: () => void
36+
onBuildFree: () => void
3637
isLastMessage?: boolean
3738
contentToCopy?: string
3839
}
@@ -52,6 +53,7 @@ export const SingleBlock = memo(
5253
onToggleCollapsed,
5354
onBuildFast,
5455
onBuildMax,
56+
onBuildFree,
5557
isLastMessage,
5658
contentToCopy,
5759
}: SingleBlockProps): ReactNode => {
@@ -120,6 +122,7 @@ export const SingleBlock = memo(
120122
markdownPalette={markdownPalette}
121123
onBuildFast={onBuildFast}
122124
onBuildMax={onBuildMax}
125+
onBuildFree={onBuildFree}
123126
/>
124127
</box>
125128
)
@@ -178,6 +181,7 @@ export const SingleBlock = memo(
178181
onToggleCollapsed={onToggleCollapsed}
179182
onBuildFast={onBuildFast}
180183
onBuildMax={onBuildMax}
184+
onBuildFree={onBuildFree}
181185
siblingBlocks={blocks}
182186
isLastMessage={isLastMessage}
183187
/>

‎cli/src/components/build-mode-buttons.tsx‎

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,14 @@ export const BuildModeButtons = ({
1010
theme,
1111
onBuildFast,
1212
onBuildMax,
13+
onBuildFree,
1314
}: {
1415
theme: ChatTheme
1516
onBuildFast: () => void
1617
onBuildMax: () => void
18+
onBuildFree: () => void
1719
}) => {
18-
const [hoveredButton, setHoveredButton] = useState<'fast' | 'max' | null>(
20+
const [hoveredButton, setHoveredButton] = useState<'fast' | 'max' | 'free' | null>(
1921
null,
2022
)
2123
const { width } = useTerminalLayout()
@@ -80,6 +82,25 @@ export const BuildModeButtons = ({
8082
<span fg={theme.foreground}>Build MAX</span>
8183
</text>
8284
</Button>
85+
<Button
86+
style={{
87+
flexDirection: 'row',
88+
alignItems: 'center',
89+
paddingLeft: 2,
90+
paddingRight: 2,
91+
borderStyle: 'single',
92+
borderColor:
93+
hoveredButton === 'free' ? theme.foreground : theme.secondary,
94+
customBorderChars: BORDER_CHARS,
95+
}}
96+
onClick={onBuildFree}
97+
onMouseOver={() => setHoveredButton('free')}
98+
onMouseOut={() => setHoveredButton(null)}
99+
>
100+
<text wrapMode="none">
101+
<span fg={theme.foreground}>Build FREE</span>
102+
</text>
103+
</Button>
83104
</box>
84105
</box>
85106
)

‎cli/src/components/message-block.tsx‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ interface MessageBlockProps {
4545
onToggleCollapsed: (id: string) => void
4646
onBuildFast: () => void
4747
onBuildMax: () => void
48+
onBuildFree: () => void
4849
onFeedback?: (messageId: string) => void
4950
onCloseFeedback?: () => void
5051
validationErrors?: Array<{ id: string; message: string }>
@@ -118,6 +119,7 @@ export const MessageBlock = memo(({
118119
onToggleCollapsed,
119120
onBuildFast,
120121
onBuildMax,
122+
onBuildFree,
121123
onFeedback,
122124
onCloseFeedback,
123125
validationErrors,
@@ -155,6 +157,7 @@ export const MessageBlock = memo(({
155157
onToggleCollapsed,
156158
onBuildFast,
157159
onBuildMax,
160+
onBuildFree,
158161
onFeedback,
159162
onCloseFeedback,
160163
validationErrors,
@@ -278,6 +281,7 @@ export const MessageBlock = memo(({
278281
onToggleCollapsed={onToggleCollapsed}
279282
onBuildFast={onBuildFast}
280283
onBuildMax={onBuildMax}
284+
onBuildFree={onBuildFree}
281285
isLastMessage={isLastMessage}
282286
contentToCopy={isUser ? content : undefined}
283287
/>

0 commit comments

Comments
 (0)