diff --git a/src/renderer/src/config/models/utils.ts b/src/renderer/src/config/models/utils.ts index 1d5c9a6443..8a52d6e4ff 100644 --- a/src/renderer/src/config/models/utils.ts +++ b/src/renderer/src/config/models/utils.ts @@ -150,7 +150,7 @@ export const isGeminiModel = (model: Model) => { export const ZHIPU_RESULT_TOKENS = ['<|begin_of_box|>', '<|end_of_box|>'] as const export const agentModelFilter = (model: Model): boolean => { - return !isEmbeddingModel(model) && !isRerankModel(model) && !isTextToImageModel(model) + return !isEmbeddingModel(model) && !isRerankModel(model) && !isTextToImageModel(model) && !isGenerateImageModel(model) } export const isMaxTemperatureOneModel = (model: Model): boolean => {