@lobehub/chat 0.107.4 → 0.107.5

package/CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+### [Version 0.107.5](https://github.com/lobehub/lobe-chat/compare/v0.107.4...v0.107.5)
+
+<sup>Released on **2023-11-30**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: 修正调用插件查询的显示问题.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: 修正调用插件查询的显示问题 ([1457fe8](https://github.com/lobehub/lobe-chat/commit/1457fe8))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 0.107.4](https://github.com/lobehub/lobe-chat/compare/v0.107.3...v0.107.4)
 
 <sup>Released on **2023-11-30**</sup>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.107.4",
+  "version": "0.107.5",
   "description": "An open-source, extensible (Function Calling), high-performance chatbot framework. It supports one-click free deployment of your private ChatGPT/LLM web application",
   "keywords": [
     "chatbot",

package/src/database/models/__DEBUG.ts

@@ -34,7 +34,12 @@ class _DEBUG_MODEL extends BaseModel<'sessions'> {
     return array[randomIndex];
   }
 
-  createRandomData = async ({ sessionCount = 10, topicCount = 2000, messageCount = 10_000 }) => {
+  createRandomData = async ({
+    sessionCount = 10,
+    topicCount = 2000,
+    messageCount = 10_000,
+    startIndex = 1,
+  }) => {
     const numberOfSessions = sessionCount;
     const numberOfTopics = topicCount;
     const numberOfMessages = messageCount;
@@ -45,7 +50,7 @@ class _DEBUG_MODEL extends BaseModel<'sessions'> {
     const messagesData: DBModel<DB_Message>[] = [];
 
     // Prepare sessions
-    for (let i = 1; i <= numberOfSessions; i++) {
+    for (let i = startIndex; i < numberOfSessions + startIndex; i++) {
       sessionsData.push({
         config: DEFAULT_AGENT_CONFIG,
         createdAt: this.randomDate(new Date(2020, 0, 1), new Date()),
@@ -61,30 +66,30 @@ class _DEBUG_MODEL extends BaseModel<'sessions'> {
     }
 
     // Prepare topics
-    for (let i = 1; i <= numberOfTopics; i++) {
+    for (let i = startIndex; i < numberOfTopics + startIndex; i++) {
       topicsData.push({
         createdAt: this.randomDate(new Date(2020, 0, 1), new Date()),
         favorite: this.getRandomInt(0, 1),
         id: `topic_${i}`,
-        sessionId: `sess_${this.getRandomInt(1, numberOfSessions)}`,
+        sessionId: `sess_${this.getRandomInt(startIndex, numberOfSessions)}`,
         title: `Topic Title ${i}`,
         updatedAt: this.randomDate(new Date(2020, 0, 1), new Date()),
       });
     }
 
     // Prepare messages
-    for (let i = 1; i <= numberOfMessages; i++) {
+    for (let i = startIndex; i < numberOfMessages + startIndex; i++) {
       messagesData.push({
         content: this.randomString(300),
         createdAt: this.randomDate(new Date(2020, 0, 1), new Date()),
         favorite: this.getRandomInt(0, 1),
         fromModel: 'model',
         id: `msg_${i}`,
-        parentId: `msg_${this.getRandomInt(1, numberOfMessages)}`,
-        quotaId: `msg_${this.getRandomInt(1, numberOfMessages)}`,
+        parentId: `msg_${this.getRandomInt(startIndex, numberOfMessages)}`,
+        quotaId: `msg_${this.getRandomInt(startIndex, numberOfMessages)}`,
         role: this.randomPick(['user', 'assistant']),
-        sessionId: `sess_${this.getRandomInt(1, numberOfSessions)}`,
-        topicId: `topic_${this.getRandomInt(1, numberOfTopics)}`,
+        sessionId: `sess_${this.getRandomInt(startIndex, numberOfSessions)}`,
+        topicId: `topic_${this.getRandomInt(startIndex, numberOfTopics)}`,
         updatedAt: this.randomDate(new Date(2020, 0, 1), new Date()),
       });
     }

package/src/services/debug.ts

@@ -2,11 +2,38 @@ import { DEBUG_MODEL } from '@/database/models/__DEBUG';
 
 class DebugService {
   async insertLargeDataToDB() {
-    return DEBUG_MODEL.createRandomData({
-      messageCount: 400_000,
+    await DEBUG_MODEL.createRandomData({
+      messageCount: 100_000,
       sessionCount: 40,
+      startIndex: 0,
       topicCount: 200,
     });
+
+    console.log('已插入10w');
+
+    await DEBUG_MODEL.createRandomData({
+      messageCount: 300_000,
+      sessionCount: 40,
+      startIndex: 100_001,
+      topicCount: 200,
+    });
+    console.log('已插入40w');
+
+    await DEBUG_MODEL.createRandomData({
+      messageCount: 300_000,
+      sessionCount: 40,
+      startIndex: 400_001,
+      topicCount: 200,
+    });
+    console.log('已插入70w');
+
+    await DEBUG_MODEL.createRandomData({
+      messageCount: 300_000,
+      sessionCount: 40,
+      startIndex: 700_001,
+      topicCount: 200,
+    });
+    console.log('已插入100w');
   }
 }
 

package/src/services/message.ts

@@ -1,4 +1,5 @@
 import { CreateMessageParams, MessageModel } from '@/database/models/message';
+import { DB_Message } from '@/database/schemas/message';
 import {
   ChatMessage,
   ChatMessageError,
@@ -64,6 +65,10 @@ export class MessageService {
     return MessageModel.batchUpdate(messageIds, { topicId });
   }
 
+  async updateMessage(id: string, message: Partial<DB_Message>) {
+    return MessageModel.update(id, message);
+  }
+
   async updateMessageRole(id: string, role: LLMRoleType) {
     return MessageModel.update(id, { role });
   }

package/src/store/chat/actions/message.ts

@@ -249,6 +249,7 @@ export const chatMessage: StateCreator<
     // 2. fetch the AI response
     const { isFunctionCall, content, functionCallAtEnd, functionCallContent } =
       await fetchAIChatMessage(messages, mid);
+    await messageService.updateMessageContent(mid, content);
 
     // 3. if it's the function call message, trigger the function method
     if (isFunctionCall) {
@@ -271,9 +272,9 @@ export const chatMessage: StateCreator<
           topicId: activeTopicId,
         };
         functionId = await messageService.create(functionMessage);
-        await refreshMessages();
       }
 
+      await refreshMessages();
       triggerFunctionCall(functionId);
     }
   },
@@ -287,7 +288,7 @@ export const chatMessage: StateCreator<
     set({ messages }, false, t(`dispatchMessage/${payload.type}`, payload));
   },
   fetchAIChatMessage: async (messages, assistantId) => {
-    const { dispatchMessage, toggleChatLoading, refreshMessages } = get();
+    const { toggleChatLoading, refreshMessages } = get();
 
     const abortController = toggleChatLoading(
       true,
@@ -361,11 +362,11 @@ export const chatMessage: StateCreator<
         // update the content after fetch result
         await messageService.updateMessageContent(assistantId, content);
       },
-      onMessageHandle: (text) => {
+      onMessageHandle: async (text) => {
         output += text;
-        // Note: Don't need change this to  `messageService.updateMessageContent`
-        // because we don't need to update the message content to the database in the middle of the process
-        dispatchMessage({ id: assistantId, key: 'content', type: 'updateMessage', value: output });
+
+        await messageService.updateMessageContent(assistantId, output);
+        await refreshMessages();
 
         // is this message is just a function call
         if (isFunctionMessageAtStart(output)) isFunctionCall = true;

package/src/store/chat/actions/plugin.ts

@@ -81,14 +81,14 @@ export const chatPlugin: StateCreator<
       };
 
       await messageService.updateMessageContent(id, '');
+      await refreshMessages();
     } else {
       if (message.plugin) payload = message.plugin;
     }
 
     if (!payload.apiName) return;
 
-    await messageService.updateMessageRole(id, 'function');
-    await messageService.updateMessagePlugin(id, payload);
+    await messageService.updateMessage(id, { plugin: payload, role: 'function' });
     await refreshMessages();
 
     if (payload.type === 'standalone') {