Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 | 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x | import { ANTHROPIC_MAX_THINKING_BUDGET, ANTHROPIC_MIN_THINKING_BUDGET } from "@shared/api"
import { Mode } from "@shared/storage/types"
import { VSCodeCheckbox } from "@vscode/webview-ui-toolkit/react"
import { memo, useCallback, useEffect, useState } from "react"
import styled from "styled-components"
import { useExtensionState } from "@/context/ExtensionStateContext"
import { getModeSpecificFields } from "./utils/providerUtils"
import { useApiConfigurationHandlers } from "./utils/useApiConfigurationHandlers"
const THUMB_SIZE = 16
const Container = styled.div`
display: flex;
flex-direction: column;
margin-top: 5px;
margin-bottom: 10px;
`
const RangeInput = styled.input<{ $value: number; $min: number; $max: number }>`
width: 100%;
height: 8px;
appearance: none;
border-radius: 4px;
outline: none;
cursor: pointer;
margin: 5px 0 0;
padding: 0;
background: ${(props) => {
const percentage = ((props.$value - props.$min) / (props.$max - props.$min)) * 100
return `linear-gradient(to right,
var(--vscode-progressBar-background) 0%,
var(--vscode-progressBar-background) ${percentage}%,
var(--vscode-scrollbarSlider-background) ${percentage}%,
var(--vscode-scrollbarSlider-background) 100%)`
}};
&::-webkit-slider-thumb {
appearance: none;
width: ${THUMB_SIZE}px;
height: ${THUMB_SIZE}px;
border-radius: 50%;
background: var(--vscode-foreground);
cursor: pointer;
border: 0px solid var(--vscode-progressBar-background);
box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
}
&:focus {
outline: none;
}
&:focus::-webkit-slider-thumb,
&:hover::-webkit-slider-thumb {
border-color: var(--vscode-progressBar-background);
box-shadow: 0 2px 6px rgba(0, 0, 0, 0.3);
}
&:active::-webkit-slider-thumb {
outline: none;
border-color: var(--vscode-progressBar-background);
}
`
interface ThinkingBudgetSliderProps {
maxBudget?: number
currentMode: Mode
}
const ThinkingBudgetSlider = ({ currentMode }: ThinkingBudgetSliderProps) => {
const { apiConfiguration } = useExtensionState()
const { handleModeFieldChange } = useApiConfigurationHandlers()
const modeFields = getModeSpecificFields(apiConfiguration, currentMode)
// Add local state for the slider value
const [localValue, setLocalValue] = useState(modeFields.thinkingBudgetTokens || 0)
const [isEnabled, setIsEnabled] = useState<boolean>((modeFields.thinkingBudgetTokens || 0) > 0)
useEffect(() => {
const newThinkingBudgetValue = modeFields.thinkingBudgetTokens || 0
const newIsEnabled = newThinkingBudgetValue > 0
// Check if the value has changed, we could be getting the same value as feedback from the user's action of clicking the enabled checkbox or moving the slider
if (newThinkingBudgetValue !== localValue) {
setLocalValue(newThinkingBudgetValue)
}
if (newIsEnabled !== isEnabled) {
setIsEnabled(newIsEnabled)
}
}, [modeFields.thinkingBudgetTokens])
const handleSliderChange = useCallback((event: React.ChangeEvent<HTMLInputElement>) => {
const value = parseInt(event.target.value, 10)
const clampedValue = Math.max(value, ANTHROPIC_MIN_THINKING_BUDGET)
setLocalValue(clampedValue)
}, [])
const handleSliderComplete = () => {
handleModeFieldChange(
{ plan: "planModeThinkingBudgetTokens", act: "actModeThinkingBudgetTokens" },
localValue,
currentMode,
)
}
const handleToggleChange = (event: any) => {
const isChecked = (event.target as HTMLInputElement).checked
const newThinkingBudgetValue = isChecked ? ANTHROPIC_MIN_THINKING_BUDGET : 0
setIsEnabled(isChecked)
setLocalValue(newThinkingBudgetValue)
handleModeFieldChange(
{ plan: "planModeThinkingBudgetTokens", act: "actModeThinkingBudgetTokens" },
newThinkingBudgetValue,
currentMode,
)
}
return (
<>
<VSCodeCheckbox checked={isEnabled} onClick={handleToggleChange}>
Enable thinking{localValue && localValue > 0 ? ` (${localValue.toLocaleString()} tokens)` : ""}
</VSCodeCheckbox>
{isEnabled && (
<Container>
<RangeInput
$max={ANTHROPIC_MAX_THINKING_BUDGET}
$min={0}
$value={localValue}
aria-describedby="thinking-budget-description"
aria-label={`Thinking budget: ${localValue.toLocaleString()} tokens`}
aria-valuemax={ANTHROPIC_MAX_THINKING_BUDGET}
aria-valuemin={ANTHROPIC_MIN_THINKING_BUDGET}
aria-valuenow={localValue}
id="thinking-budget-slider"
max={ANTHROPIC_MAX_THINKING_BUDGET}
min={0}
onChange={handleSliderChange}
onMouseUp={handleSliderComplete}
onTouchEnd={handleSliderComplete}
step={1}
type="range"
value={localValue}
/>
</Container>
)}
</>
)
}
export default memo(ThinkingBudgetSlider)
|