From bff7ccc479f523384a96d9d987514bff3f1d84a9 Mon Sep 17 00:00:00 2001 From: eeee0717 Date: Fri, 13 Jun 2025 21:25:39 +0800 Subject: [PATCH] fix: FileMetaData replace FileType --- .../src/aiCore/clients/gemini/GeminiAPIClient.ts | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts b/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts index bc848df7f..7903f5d4e 100644 --- a/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts +++ b/src/renderer/src/aiCore/clients/gemini/GeminiAPIClient.ts @@ -31,7 +31,7 @@ import { estimateTextTokens } from '@renderer/services/TokenService' import { Assistant, EFFORT_RATIO, - FileType, + FileMetadata, FileTypes, GenerateImageParams, MCPCallToolResponse, @@ -187,7 +187,7 @@ export class GeminiAPIClient extends BaseApiClient< * @param file - The file * @returns The part */ - private async handlePdfFile(file: FileType): Promise { + private async handlePdfFile(file: FileMetadata): Promise { const smallFileSize = 20 * MB const isSmallFile = file.size < smallFileSize @@ -726,7 +726,7 @@ export class GeminiAPIClient extends BaseApiClient< return sdkPayload.history || [] } - private async uploadFile(file: FileType): Promise { + private async uploadFile(file: FileMetadata): Promise { return await this.sdkInstance!.files.upload({ file: file.path, config: { @@ -737,7 +737,7 @@ export class GeminiAPIClient extends BaseApiClient< }) } - private async base64File(file: FileType) { + private async base64File(file: FileMetadata) { const { data } = await window.api.file.base64File(file.id + file.ext) return { data, @@ -745,7 +745,7 @@ export class GeminiAPIClient extends BaseApiClient< } } - private async retrieveFile(file: FileType): Promise { + private async retrieveFile(file: FileMetadata): Promise { const cachedResponse = CacheService.get('gemini_file_list') if (cachedResponse) { @@ -758,7 +758,7 @@ export class GeminiAPIClient extends BaseApiClient< return this.processResponse(response, file) } - private async processResponse(response: Pager, file: FileType) { + private async processResponse(response: Pager, file: FileMetadata) { for await (const f of response) { if (f.state === FileState.ACTIVE) { if (f.displayName === file.origin_name && Number(f.sizeBytes) === file.size) {