diff --git a/packages/shared/config/constant.ts b/packages/shared/config/constant.ts index e4545d44cb..2ed5dbc7cc 100644 --- a/packages/shared/config/constant.ts +++ b/packages/shared/config/constant.ts @@ -416,6 +416,6 @@ export enum UpgradeChannel { BETA = 'beta' // 预览版本 } -export const defaultTimeout = 5 * 1000 * 60 +export const defaultTimeout = 10 * 1000 * 60 export const occupiedDirs = ['logs', 'Network', 'Partitions/webview/Network'] diff --git a/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts b/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts index e4bfafe14d..bfd2aff3f2 100644 --- a/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts +++ b/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts @@ -60,7 +60,7 @@ import { } from '@renderer/utils/mcp-tools' import { findFileBlocks, findImageBlocks, getMainTextContent } from '@renderer/utils/messageUtils/find' import { buildSystemPrompt } from '@renderer/utils/prompt' -import { MB } from '@shared/config/constant' +import { defaultTimeout, MB } from '@shared/config/constant' import { BaseApiClient } from '../BaseApiClient' import { RequestTransformer, ResponseChunkTransformer } from '../types' @@ -118,7 +118,7 @@ export class GeminiAPIClient extends BaseApiClient< aspectRatio: imageSize, abortSignal: signal, httpOptions: { - timeout: 5 * 60 * 1000 + timeout: defaultTimeout } } const response = await sdk.models.generateImages({ diff --git a/src/renderer/src/aiCore/middleware/feat/ImageGenerationMiddleware.ts b/src/renderer/src/aiCore/middleware/feat/ImageGenerationMiddleware.ts index 6c01759bec..ceb8d791d7 100644 --- a/src/renderer/src/aiCore/middleware/feat/ImageGenerationMiddleware.ts +++ b/src/renderer/src/aiCore/middleware/feat/ImageGenerationMiddleware.ts @@ -3,6 +3,7 @@ import { isDedicatedImageGenerationModel } from '@renderer/config/models' import FileManager from '@renderer/services/FileManager' import { ChunkType } from '@renderer/types/chunk' import { findImageBlocks, getMainTextContent } from '@renderer/utils/messageUtils/find' +import { defaultTimeout } from '@shared/config/constant' import OpenAI from 'openai' import { toFile } from 'openai/uploads' @@ -74,8 +75,7 @@ export const ImageGenerationMiddleware: CompletionsMiddleware = const startTime = Date.now() let response: OpenAI.Images.ImagesResponse - - const options = { signal, timeout: 300_000 } + const options = { signal, timeout: defaultTimeout } if (imageFiles.length > 0) { response = await sdk.images.edit(