diff --git a/packages/shared/data/api/schemas/messages.ts b/packages/shared/data/api/schemas/messages.ts
index b952156da5..a2462ca489 100644
--- a/packages/shared/data/api/schemas/messages.ts
+++ b/packages/shared/data/api/schemas/messages.ts
@@ -62,6 +62,10 @@ export interface UpdateMessageDto {
   siblingsGroupId?: number
   /** Update status */
   status?: MessageStatus
+  /** Update trace ID */
+  traceId?: string | null
+  /** Update statistics */
+  stats?: MessageStats | null
 }
 
 /**
diff --git a/src/main/data/services/MessageService.ts b/src/main/data/services/MessageService.ts
index 5f1c05e749..7542ffc02b 100644
--- a/src/main/data/services/MessageService.ts
+++ b/src/main/data/services/MessageService.ts
@@ -558,6 +558,8 @@ export class MessageService {
       if (dto.parentId !== undefined) updates.parentId = dto.parentId
       if (dto.siblingsGroupId !== undefined) updates.siblingsGroupId = dto.siblingsGroupId
       if (dto.status !== undefined) updates.status = dto.status
+      if (dto.traceId !== undefined) updates.traceId = dto.traceId
+      if (dto.stats !== undefined) updates.stats = dto.stats
 
       const [row] = await tx.update(messageTable).set(updates).where(eq(messageTable.id, id)).returning()