fix: message errors should be added to the message

mrdjohnson · Nov 8, 2024 · 7797715 · 7797715
1 parent b27d84d
commit 7797715
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/core/IncomingMessageStore.ts b/src/core/IncomingMessageStore.ts
@@ -141,7 +141,7 @@ export class IncomingMessageStore {
 
         shouldDeleteMessage = _.isEmpty(messageToEdit.content)
       } else if (error instanceof Error) {
-        messageToEdit.setError(error)
+        await messageToEdit.setError(error)
 
         // make sure the server is still connected
         connectionStore.selectedConnection?.fetchLmModels()

diff --git a/src/core/message/MessageViewModel.ts b/src/core/message/MessageViewModel.ts
@@ -91,12 +91,12 @@ export class MessageViewModel {
     3000,
   )
 
-  setError(errorData: Error) {
-    const error = MessageErrorModel.safeParse(errorData)
+  async setError(errorData: Error) {
+    const { data: error } = MessageErrorModel.safeParse(errorData)
 
-    const extras = MessageExtrasModel.safeParse({ ...this.source.extras, error })
+    const { data: extras } = MessageExtrasModel.safeParse({ ...this.source.extras, error })
 
-    this.update({ extras })
+    return this.update({ extras })
   }
 
   async setExtraDetails(detailObject?: Record<string, unknown>) {