Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-06 05:09:09 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

revert: generate image button (#5414)

Browse Source 
* revert: generate image button

* chore
This commit is contained in:
Chen Tao 2025-04-27 21:19:27 +08:00 committed by GitHub
parent 56cd5cc3df
commit dd16ad9d67
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/renderer/src/pages/home/Inputbar/Inputbar.tsx
View File

@ -55,6 +55,7 @@ import styled from 'styled-components'
import NarrowLayout from '../Messages/NarrowLayout' import NarrowLayout from '../Messages/NarrowLayout'
import AttachmentButton, { AttachmentButtonRef } from './AttachmentButton' import AttachmentButton, { AttachmentButtonRef } from './AttachmentButton'
import AttachmentPreview from './AttachmentPreview' import AttachmentPreview from './AttachmentPreview'
import GenerateImageButton from './GenerateImageButton'
import KnowledgeBaseButton, { KnowledgeBaseButtonRef } from './KnowledgeBaseButton' import KnowledgeBaseButton, { KnowledgeBaseButtonRef } from './KnowledgeBaseButton'
import KnowledgeBaseInput from './KnowledgeBaseInput' import KnowledgeBaseInput from './KnowledgeBaseInput'
import MCPToolsButton, { MCPToolsButtonRef } from './MCPToolsButton' import MCPToolsButton, { MCPToolsButtonRef } from './MCPToolsButton'
@ -795,6 +796,10 @@ const Inputbar: FC<Props> = ({ assistant: _assistant, setActiveTopic, topic }) =
updateAssistant({ ...assistant, enableWebSearch: !assistant.enableWebSearch }) updateAssistant({ ...assistant, enableWebSearch: !assistant.enableWebSearch })
} }
const onEnableGenerateImage = () => {
updateAssistant({ ...assistant, enableGenerateImage: !assistant.enableGenerateImage })
}
useEffect(() => { useEffect(() => {
if (!isWebSearchModel(model) && !WebSearchService.isWebSearchEnabled() && assistant.enableWebSearch) { if (!isWebSearchModel(model) && !WebSearchService.isWebSearchEnabled() && assistant.enableWebSearch) {
updateAssistant({ ...assistant, enableWebSearch: false }) updateAssistant({ ...assistant, enableWebSearch: false })
@ -802,7 +807,7 @@ const Inputbar: FC<Props> = ({ assistant: _assistant, setActiveTopic, topic }) =
if (!isGenerateImageModel(model) && assistant.enableGenerateImage) { if (!isGenerateImageModel(model) && assistant.enableGenerateImage) {
updateAssistant({ ...assistant, enableGenerateImage: false }) updateAssistant({ ...assistant, enableGenerateImage: false })
} }
if (isGenerateImageModel(model) && !assistant.enableGenerateImage) { if (isGenerateImageModel(model) && !assistant.enableGenerateImage && model.id !== 'gemini-2.0-flash-exp') {
updateAssistant({ ...assistant, enableGenerateImage: true }) updateAssistant({ ...assistant, enableGenerateImage: true })
} }
}, [assistant, model, updateAssistant]) }, [assistant, model, updateAssistant])
@ -951,6 +956,13 @@ const Inputbar: FC<Props> = ({ assistant: _assistant, setActiveTopic, topic }) =
setInputValue={setText} setInputValue={setText}
resizeTextArea={resizeTextArea} resizeTextArea={resizeTextArea}
/> />
<GenerateImageButton
model={model}
assistant={assistant}
onEnableGenerateImage={onEnableGenerateImage}
ToolbarButton={ToolbarButton}
/>
<MentionModelsButton <MentionModelsButton
ref={mentionModelsButtonRef} ref={mentionModelsButtonRef}
mentionModels={mentionModels} mentionModels={mentionModels}