From 353e497642a606bee7c6166e5126d244cc8bb6bb Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Thu, 19 Sep 2024 22:58:12 +0800
Subject: [PATCH] feat: Improved layout and added file content filtering.

- Added a margin bottom to the Upload component in the MessageAttachments page for improved layout.
- Added support for not displaying file contents for specific providers.
---
 .../home/Messages/MessageAttachments.tsx      |  2 +-
 src/renderer/src/providers/OpenAIProvider.ts  | 34 +++++++++++++++++++
 2 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/pages/home/Messages/MessageAttachments.tsx b/src/renderer/src/pages/home/Messages/MessageAttachments.tsx
index 6228a8a1..fe98065d 100644
--- a/src/renderer/src/pages/home/Messages/MessageAttachments.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageAttachments.tsx
@@ -18,7 +18,7 @@ const MessageAttachments: FC<Props> = ({ message }) => {
   }
 
   return (
-    <Container style={{ marginTop: 2 }}>
+    <Container style={{ marginTop: 2, marginBottom: 8 }}>
       <Upload
         listType="picture"
         disabled
diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts
index 3f9619a7..84b67512 100644
--- a/src/renderer/src/providers/OpenAIProvider.ts
+++ b/src/renderer/src/providers/OpenAIProvider.ts
@@ -34,6 +34,11 @@ export default class OpenAIProvider extends BaseProvider {
     return true
   }
 
+  private get isNotSupportFiles() {
+    const providers = ['deepseek', 'baichuan', 'minimax', 'yi', 'doubao']
+    return providers.includes(this.provider.id)
+  }
+
   private async getMessageParam(
     message: Message,
     model: Model
@@ -47,6 +52,35 @@ export default class OpenAIProvider extends BaseProvider {
       }
     }
 
+    if (this.isNotSupportFiles) {
+      if (message.files) {
+        const textFiles = message.files.filter((file) => file.type === FileTypes.TEXT)
+
+        if (textFiles.length > 0) {
+          let text = ''
+          const divider = '\n\n---\n\n'
+
+          for (const file of textFiles) {
+            const fileContent = (await window.api.file.read(file.id + file.ext)).trim()
+            const fileNameRow = 'file: ' + file.origin_name + '\n\n'
+            text = text + fileNameRow + fileContent + divider
+          }
+
+          message.content = message.content + divider + text
+
+          return {
+            role: message.role,
+            content: message.content
+          }
+        }
+      }
+
+      return {
+        role: message.role,
+        content: message.content
+      }
+    }
+
     const parts: ChatCompletionContentPart[] = [
       {
         type: 'text',