diff --git a/src/renderer/src/pages/home/Inputbar/Inputbar.tsx b/src/renderer/src/pages/home/Inputbar/Inputbar.tsx index 377c63c6d9..e2dd58a730 100644 --- a/src/renderer/src/pages/home/Inputbar/Inputbar.tsx +++ b/src/renderer/src/pages/home/Inputbar/Inputbar.tsx @@ -1,7 +1,13 @@ import { HolderOutlined } from '@ant-design/icons' import { QuickPanelListItem, QuickPanelView, useQuickPanel } from '@renderer/components/QuickPanel' import TranslateButton from '@renderer/components/TranslateButton' -import { isGenerateImageModel, isReasoningModel, isVisionModel, isWebSearchModel } from '@renderer/config/models' +import { + isGenerateImageModel, + isSupportedReasoningEffortModel, + isSupportedThinkingTokenModel, + isVisionModel, + isWebSearchModel +} from '@renderer/config/models' import db from '@renderer/databases' import { useAssistant } from '@renderer/hooks/useAssistant' import { useKnowledgeBases } from '@renderer/hooks/useKnowledge' @@ -854,6 +860,7 @@ const Inputbar: FC = ({ assistant: _assistant, setActiveTopic, topic }) = } const isExpended = expended || !!textareaHeight + const showThinkingButton = isSupportedThinkingTokenModel(model) || isSupportedReasoningEffortModel(model) return ( @@ -920,7 +927,7 @@ const Inputbar: FC = ({ assistant: _assistant, setActiveTopic, topic }) = setFiles={setFiles} ToolbarButton={ToolbarButton} /> - {isReasoningModel(model) && ( + {showThinkingButton && (