From dfeb54fd94f24381440a5c821cbbf2e0e2ab89f9 Mon Sep 17 00:00:00 2001
From: suyao <sy20010504@gmail.com>
Date: Sun, 11 May 2025 03:50:05 +0800
Subject: [PATCH] fix: add support for inline base64 image data in image block

---
 .../src/providers/AiProvider/GeminiProvider.ts         | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/renderer/src/providers/AiProvider/GeminiProvider.ts b/src/renderer/src/providers/AiProvider/GeminiProvider.ts
index db30fa6c4b..c04632efd3 100644
--- a/src/renderer/src/providers/AiProvider/GeminiProvider.ts
+++ b/src/renderer/src/providers/AiProvider/GeminiProvider.ts
@@ -153,6 +153,16 @@ export default class GeminiProvider extends BaseProvider {
           }
         }
       }
+      const file = imageBlock.file
+      if (file) {
+        const base64Data = await window.api.file.base64Image(file.id + file.ext)
+        parts.push({
+          inlineData: {
+            data: base64Data.base64,
+            mimeType: base64Data.mime
+          } as Part['inlineData']
+        })
+      }
     }
 
     const fileBlocks = findFileBlocks(message)