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

feat: Enable image generation in assistant based on model selection (#5364)

Browse Source 
* feat: Enable image generation in assistant based on model selection

* chore: remove generate image button
This commit is contained in:
Chen Tao 2025-04-26 22:15:48 +08:00 committed by GitHub
parent bcbb3f294e
commit 6b113c19a3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 11 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,7 +55,6 @@ 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'
@ -796,10 +795,6 @@ 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 })
@ -807,6 +802,9 @@ 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) {
updateAssistant({ ...assistant, enableGenerateImage: true })
}
}, [assistant, model, updateAssistant]) }, [assistant, model, updateAssistant])
const onMentionModel = (model: Model) => { const onMentionModel = (model: Model) => {
@ -953,12 +951,6 @@ 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}