@lobehub/lobehub 2.0.0-next.40 → 2.0.0-next.41

package/packages/database/src/models/message.ts

@@ -14,7 +14,6 @@ import {
   UIChatMessage,
   UpdateMessageParams,
   UpdateMessageRAGParams,
-  UpdateMessageResult,
 } from '@lobechat/types';
 import type { HeatmapsProps } from '@lobehub/charts';
 import dayjs from 'dayjs';
@@ -550,13 +549,7 @@ export class MessageModel {
   update = async (
     id: string,
     { imageList, ...message }: Partial<UpdateMessageParams>,
-    options?: {
-      groupAssistantMessages?: boolean;
-      postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
-      sessionId?: string | null;
-      topicId?: string | null;
-    },
-  ): Promise<UpdateMessageResult> => {
+  ): Promise<{ success: boolean }> => {
     try {
       await this.db.transaction(async (trx) => {
         // 1. insert message files
@@ -574,22 +567,6 @@ export class MessageModel {
           .where(and(eq(messages.id, id), eq(messages.userId, this.userId)));
       });
 
-      // if sessionId or topicId provided, return the updated message list
-      if (options?.sessionId !== undefined || options?.topicId !== undefined) {
-        const messageList = await this.query(
-          {
-            sessionId: options.sessionId,
-            topicId: options.topicId,
-          },
-          {
-            groupAssistantMessages: options.groupAssistantMessages ?? false,
-            postProcessUrl: options.postProcessUrl,
-          },
-        );
-
-        return { messages: messageList, success: true };
-      }
-
       return { success: true };
     } catch (error) {
       console.error('Update message error:', error);
@@ -610,16 +587,7 @@ export class MessageModel {
       .where(and(eq(messages.userId, this.userId), eq(messages.id, id)));
   };
 
-  updatePluginState = async (
-    id: string,
-    state: Record<string, any>,
-    options?: {
-      groupAssistantMessages?: boolean;
-      postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
-      sessionId?: string | null;
-      topicId?: string | null;
-    },
-  ): Promise<UpdateMessageResult> => {
+  updatePluginState = async (id: string, state: Record<string, any>): Promise<void> => {
     const item = await this.db.query.messagePlugins.findFirst({
       where: eq(messagePlugins.id, id),
     });
@@ -629,22 +597,6 @@ export class MessageModel {
       .update(messagePlugins)
       .set({ state: merge(item.state || {}, state) })
       .where(eq(messagePlugins.id, id));
-
-    // Return updated messages if sessionId or topicId is provided
-    if (options?.sessionId !== undefined || options?.topicId !== undefined) {
-      const messageList = await this.query(
-        {
-          sessionId: options.sessionId,
-          topicId: options.topicId,
-        },
-        {
-          groupAssistantMessages: options.groupAssistantMessages ?? false,
-          postProcessUrl: options.postProcessUrl,
-        },
-      );
-      return { messages: messageList, success: true };
-    }
-    return { success: true };
   };
 
   updateMessagePlugin = async (id: string, value: Partial<MessagePluginItem>) => {

package/src/server/services/message/index.ts

@@ -1,5 +1,5 @@
 import { LobeChatDatabase } from '@lobechat/database';
-import { CreateMessageParams, UpdateMessageParams } from '@lobechat/types';
+import { CreateMessageParams, UIChatMessage, UpdateMessageParams } from '@lobechat/types';
 
 import { MessageModel } from '@/database/models/message';
 
@@ -51,7 +51,9 @@ export class MessageService {
   /**
    * Query messages and return response with success status (used after mutations)
    */
-  private async queryWithSuccess(options?: QueryOptions) {
+  private async queryWithSuccess(
+    options?: QueryOptions,
+  ): Promise<{ messages?: UIChatMessage[], success: boolean; }> {
     if (!options || (options.sessionId === undefined && options.topicId === undefined)) {
       return { success: true };
     }
@@ -139,7 +141,11 @@ export class MessageService {
    * Update plugin state and return message list
    * Pattern: update + conditional query
    */
-  async updatePluginState(id: string, value: any, options: QueryOptions): Promise<any> {
+  async updatePluginState(
+    id: string,
+    value: any,
+    options: QueryOptions,
+  ): Promise<{ messages?: UIChatMessage[], success: boolean; }> {
     await this.messageModel.updatePluginState(id, value);
     return this.queryWithSuccess(options);
   }
@@ -148,7 +154,11 @@ export class MessageService {
    * Update message and return message list
    * Pattern: update + conditional query
    */
-  async updateMessage(id: string, value: UpdateMessageParams, options: QueryOptions): Promise<any> {
+  async updateMessage(
+    id: string,
+    value: UpdateMessageParams,
+    options: QueryOptions,
+  ): Promise<{ messages?: UIChatMessage[], success: boolean; }> {
     await this.messageModel.update(id, value as any);
     return this.queryWithSuccess(options);
   }