@lobehub/chat 0.107.5 → 0.107.6

package/CHANGELOG.md

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.107.5",
+  "version": "0.107.6",
   "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/__tests__/message.test.ts

@@ -260,6 +260,19 @@ describe('MessageModel', () => {
 
       expect(updatedMessage).toHaveProperty('content', 'Updated content');
     });
+
+    it('should update a role and plugins', async () => {
+      const createdMessage = await MessageModel.create(messageData);
+      const updateData = {
+        role: 'function' as const,
+        plugin: { apiName: 'a', identifier: 'b', arguments: 'abc' },
+      };
+
+      await MessageModel.update(createdMessage.id, updateData);
+      const updatedMessage = await MessageModel.findById(createdMessage.id);
+
+      expect(updatedMessage).toHaveProperty('role', 'function');
+    });
   });
 
   describe('batchUpdate', () => {

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

@@ -249,7 +249,6 @@ 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) {
@@ -361,6 +360,7 @@ export const chatMessage: StateCreator<
       onFinish: async (content) => {
         // update the content after fetch result
         await messageService.updateMessageContent(assistantId, content);
+        await refreshMessages();
       },
       onMessageHandle: async (text) => {
         output += text;

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

@@ -97,7 +97,7 @@ export const chatTopic: StateCreator<
       onError: () => {
         updateTopicTitleInSummary(topicId, topic.title);
       },
-      onFinish: (text) => {
+      onFinish: async (text) => {
         topicService.updateTitle(topicId, text);
       },
       onLoadingChange: (loading) => {

package/src/utils/fetch.ts

@@ -27,7 +27,7 @@ export const getMessageError = async (response: Response) => {
 
 export interface FetchSSEOptions {
   onErrorHandle?: (error: ChatMessageError) => void;
-  onFinish?: (text: string) => void;
+  onFinish?: (text: string) => Promise<void>;
   onMessageHandle?: (text: string) => void;
 }
 
@@ -67,7 +67,7 @@ export const fetchSSE = async (fetchFn: () => Promise<Response>, options: FetchS
     options.onMessageHandle?.(chunkValue);
   }
 
-  options?.onFinish?.(output);
+  await options?.onFinish?.(output);
 
   return returnRes;
 };
@@ -78,7 +78,7 @@ interface FetchAITaskResultParams<T> {
    * 错误处理函数
    */
   onError?: (e: Error, rawError?: any) => void;
-  onFinish?: (text: string) => void;
+  onFinish?: (text: string) => Promise<void>;
   /**
    * 加载状态变化处理函数
    * @param loading - 是否处于加载状态