Merge branch 'main' of https://github.com/lobehub/lobe-chat

CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+### [Version 1.39.1](https://github.com/lobehub/lobe-chat/compare/v1.39.0...v1.39.1)
+
+<sup>Released on **2024-12-24**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix image input on pglite.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix image input on pglite, closes [#5167](https://github.com/lobehub/lobe-chat/issues/5167) ([5c5b37d](https://github.com/lobehub/lobe-chat/commit/5c5b37d))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ## [Version 1.39.0](https://github.com/lobehub/lobe-chat/compare/v1.38.0...v1.39.0)
 
 <sup>Released on **2024-12-23**</sup>

changelog/v1.json

@@ -1,4 +1,11 @@
 [
+  {
+    "children": {
+      "fixes": ["Fix image input on pglite."]
+    },
+    "date": "2024-12-24",
+    "version": "1.39.1"
+  },
   {
     "children": {
       "features": ["Upgrade to next15 and react19."]

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.39.0",
+  "version": "1.39.1",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",

src/database/server/models/message.ts

@@ -48,7 +48,9 @@ export class MessageModel {
   // **************** Query *************** //
   query = async (
     { current = 0, pageSize = 1000, sessionId, topicId }: QueryMessageParams = {},
-    options: { postProcessUrl?: (path: string | null) => Promise<string> } = {},
+    options: {
+      postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
+    } = {},
   ): Promise<MessageItem[]> => {
     const offset = current * pageSize;
 
@@ -130,7 +132,9 @@ export class MessageModel {
     const relatedFileList = await Promise.all(
       rawRelatedFileList.map(async (file) => ({
         ...file,
-        url: options.postProcessUrl ? await options.postProcessUrl(file.url) : (file.url as string),
+        url: options.postProcessUrl
+          ? await options.postProcessUrl(file.url, file as any)
+          : (file.url as string),
       })),
     );
 

src/services/message/client.ts

@@ -5,6 +5,7 @@ import { clientDB } from '@/database/client/db';
 import { MessageItem } from '@/database/schemas';
 import { MessageModel } from '@/database/server/models/message';
 import { BaseClientService } from '@/services/baseClientService';
+import { clientS3Storage } from '@/services/file/ClientS3';
 import {
   ChatMessage,
   ChatMessageError,
@@ -34,10 +35,20 @@ export class ClientService extends BaseClientService implements IMessageService
   }
 
   async getMessages(sessionId: string, topicId?: string) {
-    const data = await this.messageModel.query({
-      sessionId: this.toDbSessionId(sessionId),
-      topicId,
-    });
+    const data = await this.messageModel.query(
+      {
+        sessionId: this.toDbSessionId(sessionId),
+        topicId,
+      },
+      {
+        postProcessUrl: async (url, file) => {
+          const hash = (url as string).replace('client-s3://', '');
+          const base64 = await this.getBase64ByFileHash(hash);
+
+          return `data:${file.fileType};base64,${base64}`;
+        },
+      },
+    );
 
     return data as unknown as ChatMessage[];
   }
@@ -115,4 +126,11 @@ export class ClientService extends BaseClientService implements IMessageService
   private toDbSessionId(sessionId: string | undefined) {
     return sessionId === INBOX_SESSION_ID ? undefined : sessionId;
   }
+
+  private async getBase64ByFileHash(hash: string) {
+    const fileItem = await clientS3Storage.getObject(hash);
+    if (!fileItem) throw new Error('file not found');
+
+    return Buffer.from(await fileItem.arrayBuffer()).toString('base64');
+  }
 }