♻️ refactor: refactor chat message model to speed up (lobehub#10053)

* refactor new chat message

* ♻️ refactor: Unify message creation methods into single `internal_createMessage`

## Changes

### Method Consolidation
- Merged `internal_createMessage` and `internal_createNewMessage` into a single unified method
- All message creation now returns `{ id: string, messages: UIChatMessage[] }`
- Eliminated redundant API calls by always using `createNewMessage` backend endpoint

### Updated Call Sites (11 locations)
**Store Actions:**
- `addAIMessage` & `addUserMessage` - Added result validation

**AI Chat:**
- `generateAIChat.ts` - Extract `result.id` from response
- `generateAIChatV2.ts` - Renamed from `internal_createNewMessage` to `internal_createMessage`

**Group Chat:**
- `generateAIGroupChat.ts` - Extract `result.id` in 3 locations

**Thread & Tools:**
- `thread/action.ts` - Extract `result.id`
- `builtinTool/actions/search.ts` - Extract `result.id`
- `plugin/action.ts` - Extract `result.id`

### Test Updates
- Updated mocks to return `{ id, messages }` structure
- `thread/action.test.ts` - 4 mock updates
- `plugin/action.test.ts` - 2 mock updates

## Benefits
- **Performance**: All message creation now uses single-request pattern
- **Consistency**: Unified return type across all creation flows
- **Maintainability**: Single method to maintain instead of two similar ones

## Testing
- ✅ Type check: 0 errors
- ✅ Unit tests: 175/175 passed
  - message/action.test.ts: 33/33
  - plugin/action.test.ts: 26/26
  - thread/action.test.ts: 39/39
  - generateAIChat.test.ts: 41/41
  - generateAIChatV2.test.ts: 36/36

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

* ♻️ refactor: optimize message update operations to reduce API calls

Optimized two message update operations to reduce network requests:

1. **updatePluginState**: Modified to return updated messages
   - Backend: `MessageModel.updatePluginState` now accepts options and returns `UpdateMessageResult`
   - Router: Added `sessionId`, `topicId`, and `useGroup` parameters
   - Frontend: Service layer passes lab preferences, store uses `replaceMessages` instead of `refreshMessages`
   - Reduction: 2 requests → 1 request

2. **message.update**: Added `groupAssistantMessages` support
   - Service: `updateMessage` now passes `useGroup` flag based on lab preferences
   - Backend model already had infrastructure for returning messages
   - Reduction: Ensures consistent 1-request pattern

Tests passing (26/26 plugin tests, 14/14 integration tests).

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

* ♻️ refactor: optimize all internal message methods to reduce API calls

Optimized 6 internal message methods and 3 error handling scenarios to reduce API calls from 2 requests (update + refresh) to 1 request (update with messages returned):

**Internal methods optimized:**
- internal_updateMessagePluginError
- internal_updateMessageRAG
- internal_deleteMessage
- internal_refreshToUpdateMessageTools
- internal_updatePluginError

**Error handling optimized:**
- internal_callPluginApi error scenarios (2 locations)
- invokeStandaloneTypePlugin invalid settings

**Changes:**
- Backend: Updated message routers to accept sessionId/topicId/useGroup and return messages
- Service: Added getUseGroupPreference() getter to simplify lab preference checks
- Service: Updated methods to use getter and return UpdateMessageResult
- Store: Changed from refreshMessages() to replaceMessages(result.messages)
- Tests: Updated 4 plugin tests to verify replaceMessages instead of refreshMessages

**Performance impact:**
Each optimized method now makes 1 request instead of 2, reducing network overhead and improving UI responsiveness.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

* ♻️ refactor: optimize message deletion to reduce API calls and fix group message children deletion

**Problem 1 - Group message children not deleted:**
- When deleting a `role: 'group'` message, children messages (linked via `parentId`) were not deleted
- Tool result messages were also not included in deletion

**Problem 2 - Delete operations using refresh pattern:**
- `deleteMessage`, `clearMessage`, `clearAllMessages` all used refreshMessages after deletion
- This resulted in 2 requests: delete + refresh

**Solutions:**

1. **Enhanced deleteMessage in UI layer:**
   - Added logic to find all children messages via `parentId` for group role messages
   - Combined with existing tool message deletion logic
   - All related message IDs are collected and passed to backend in one call
   - Business logic stays in UI layer, model layer remains simple

2. **Optimized delete operations:**
   - Backend: `removeMessages` now accepts sessionId/topicId/useGroup and returns messages
   - Service: `removeMessages` updated to pass options and return UpdateMessageResult
   - Store: `deleteMessage` now uses replaceMessages with returned data (2 requests → 1 request)
   - Store: `clearMessage` and `clearAllMessages` directly replace with empty array

3. **Updated tests:**
   - Fixed 4 tests to verify replaceMessages instead of refreshMessages
   - Added mock for service to return messages in delete operations
   - All 33 message action tests passing
   - All 14 integration tests passing

**Performance impact:**
- deleteMessage: 2 requests → 1 request
- clearMessage/clearAllMessages: 1 delete + 1 refresh → 1 delete + direct clear

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

* ✅ test: add comprehensive tests for group message deletion with children

Added 2 new test cases to verify group message deletion behavior:

1. **Basic group message with children deletion:**
   - Verifies that deleting a `role: 'group'` message also deletes all children (via `parentId`)
   - Tests that unrelated messages are preserved

2. **Group message with children that have tool calls:**
   - Verifies that deleting a group message also deletes:
     - The group message itself
     - All children messages (via `parentId`)
     - Tool result messages from children (via `tool_call_id`)
   - Ensures complete cleanup of the entire message tree

**Implementation enhancement:**
- Updated `deleteMessage` to also collect and delete tool results from children messages
- Ensures no orphaned tool result messages remain after group deletion

**Test results:**
- All 35 message action tests passing (2 new tests added)
- Verifies complete cascading deletion of group message trees

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

---------

Co-authored-by: Claude <noreply@anthropic.com>

Author: arvinxx

packages/database/package.json

@@ -24,7 +24,7 @@
   "peerDependencies": {
     "@electric-sql/pglite": "^0.2.17",
     "dayjs": ">=1.11.18",
-    "drizzle-orm": ">=0.44.6",
+    "drizzle-orm": ">=0.44.7",
     "nanoid": ">=5.1.5",
     "pg": ">=8.16.3"
   }

packages/database/src/models/message.ts

@@ -603,6 +603,7 @@ export class MessageModel {
     id: string,
     { imageList, ...message }: Partial<UpdateMessageParams>,
     options?: {
+      groupAssistantMessages?: boolean;
       postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
       sessionId?: string | null;
       topicId?: string | null;
@@ -633,6 +634,7 @@ export class MessageModel {
             topicId: options.topicId,
           },
           {
+            groupAssistantMessages: options.groupAssistantMessages ?? false,
             postProcessUrl: options.postProcessUrl,
           },
         );
@@ -660,16 +662,41 @@ export class MessageModel {
       .where(and(eq(messages.userId, this.userId), eq(messages.id, id)));
   };
 
-  updatePluginState = async (id: string, state: Record<string, any>) => {
+  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> => {
     const item = await this.db.query.messagePlugins.findFirst({
       where: eq(messagePlugins.id, id),
     });
     if (!item) throw new Error('Plugin not found');
 
-    return this.db
+    await this.db
       .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>) => {

packages/types/src/message/ui/params.ts

@@ -120,55 +120,6 @@ const ChatPluginPayloadSchema = z.object({
   type: z.string(),
 });
 
-export const CreateMessageParamsSchema = z
-  .object({
-    content: z.string(),
-    role: UIMessageRoleTypeSchema,
-    sessionId: z.string().nullable().optional(),
-    error: ChatMessageErrorSchema.nullable().optional(),
-    fileChunks: z.array(SemanticSearchChunkSchema).optional(),
-    files: z.array(z.string()).optional(),
-    fromModel: z.string().optional(),
-    fromProvider: z.string().optional(),
-    groupId: z.string().nullable().optional(),
-    targetId: z.string().nullable().optional(),
-    threadId: z.string().nullable().optional(),
-    topicId: z.string().nullable().optional(),
-    traceId: z.string().optional(),
-    // Allow additional fields from UIChatMessage (many can be null)
-    agentId: z.string().optional(),
-    children: z.any().optional(),
-    chunksList: z.any().optional(),
-    createdAt: z.number().optional(),
-    extra: z.any().optional(),
-    favorite: z.boolean().optional(),
-    fileList: z.any().optional(),
-    id: z.string().optional(),
-    imageList: z.any().optional(),
-    meta: z.any().optional(),
-    metadata: z.any().nullable().optional(),
-    model: z.string().nullable().optional(),
-    observationId: z.string().optional(),
-    parentId: z.string().optional(),
-    performance: z.any().optional(),
-    plugin: z.any().optional(),
-    pluginError: z.any().optional(),
-    pluginState: z.any().optional(),
-    provider: z.string().nullable().optional(),
-    quotaId: z.string().optional(),
-    ragQuery: z.string().nullable().optional(),
-    ragQueryId: z.string().nullable().optional(),
-    reasoning: z.any().optional(),
-    search: z.any().optional(),
-    tool_call_id: z.string().optional(),
-    toolCalls: z.any().optional(),
-    tools: z.any().optional(),
-    translate: z.any().optional(),
-    tts: z.any().optional(),
-    updatedAt: z.number().optional(),
-  })
-  .passthrough();
-
 export const CreateNewMessageParamsSchema = z
   .object({
     // Required fields