fix a issue that cannot edit-and-save-new

sdcb · Jan 22, 2025 · 6715276 · 6715276
1 parent 1c490cf
commit 6715276
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/BE/Controllers/Chats/Messages/MessagesController.cs b/src/BE/Controllers/Chats/Messages/MessagesController.cs
@@ -155,6 +155,9 @@ public async Task<ActionResult<RequestMessageDto>> EditAndSaveNew(string encrypt
         Message? message = await db.Messages
             .Include(x => x.Chat)
             .Include(x => x.Usage)
+            .Include(x => x.Usage!.UserModel)
+            .Include(x => x.Usage!.UserModel.Model)
+            .Include(x => x.Usage!.UserModel.Model.ModelKey)
             .FirstOrDefaultAsync(x => x.Id == messageId, cancellationToken);
         if (message == null)
         {
@@ -181,6 +184,7 @@ public async Task<ActionResult<RequestMessageDto>> EditAndSaveNew(string encrypt
             newMessage.Usage = new UserModelUsage()
             {
                 UserModelId = message.Usage.UserModelId,
+                UserModel = message.Usage.UserModel,
                 FinishReasonId = (byte)DBFinishReason.Success,
                 SegmentCount = 1,
                 InputTokens = message.Usage.InputTokens,