upscale message content

trueai-org · Jul 9, 2024 · d807b51 · d807b51
1 parent dd4896c
commit d807b51
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/Midjourney.Infrastructure/Handle/MessageHandler.cs b/src/Midjourney.Infrastructure/Handle/MessageHandler.cs
@@ -110,6 +110,10 @@ protected void FinishTask(TaskInfo task, SocketMessage message)
             {
                 task.FailReason = "";
             }
+            if (string.IsNullOrWhiteSpace(task.State))
+            {
+                task.State = "";
+            }
 
             task.Success();
         }

diff --git a/src/Midjourney.Infrastructure/Handle/UpscaleSuccessHandler.cs b/src/Midjourney.Infrastructure/Handle/UpscaleSuccessHandler.cs
@@ -59,6 +59,8 @@ private void FindAndFinishUTask(IDiscordInstance instance, string finalPrompt, i
             task.SetProperty(Constants.MJ_MESSAGE_HANDLED, true);
             task.SetProperty(Constants.TASK_PROPERTY_FINAL_PROMPT, finalPrompt);
             task.SetProperty(Constants.TASK_PROPERTY_MESSAGE_HASH, messageHash);
+            task.SetProperty(Constants.TASK_PROPERTY_MESSAGE_CONTENT, message.Content);
+
             task.ImageUrl = imageUrl;
             FinishTask(task, message);
             task.Awake();