Merge pull request #1903 from nextcloud/finalize-summary-on-failure

Remove summary tasks on AI failure/cancellation
nextcloud · Dec 2, 2024 · cd64b30 · cd64b30
2 parents 96ac7ca + a915dc0
commit cd64b30
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/NextcloudTalk/AiSummaryController.swift b/NextcloudTalk/AiSummaryController.swift
@@ -32,6 +32,7 @@ public class AiSummaryController {
         task.outputs.append(output)
     }
 
+    @discardableResult
     public func finalizeSummaryTask(forRoomInternalId internalId: String) -> [String] {
         let result = generateSummaryTasks[internalId]?.outputs ?? []
         generateSummaryTasks.removeValue(forKey: internalId)

diff --git a/NextcloudTalk/ChatViewController.swift b/NextcloudTalk/ChatViewController.swift
@@ -1525,11 +1525,13 @@ import SwiftyAttributes
                     }
 
                 } else if status == .failed {
+                    AiSummaryController.shared.finalizeSummaryTask(forRoomInternalId: self.room.internalId)
                     NotificationPresenter.shared().dismiss()
                     NotificationPresenter.shared().present(text: NSLocalizedString("Generating summary of unread messages failed", comment: ""), dismissAfterDelay: 7.0, includedStyle: .error)
 
                     return
                 } else if status == .cancelled {
+                    AiSummaryController.shared.finalizeSummaryTask(forRoomInternalId: self.room.internalId)
                     NotificationPresenter.shared().dismiss()
                     return
                 }