Set as system prompt

Author: intitni

Core/Sources/ChatService/ChatService.swift

@@ -61,7 +61,7 @@ public final class ChatService: ObservableObject {
         await pluginController.cancel()
         await chatGPTService.clearHistory()
     }
-    
+
     public func resetPrompt() async {
         systemPrompt = defaultSystemPrompt
         extraSystemPrompt = ""
@@ -79,6 +79,19 @@ public final class ChatService: ObservableObject {
         }
     }
 
+    public func setMessageAsExtraPrompt(id: String) async {
+        if let message = (await chatGPTService.history).first(where: { $0.id == id }) {
+            mutateExtraSystemPrompt(message.content)
+            await mutateHistory { history in
+                history.append(.init(
+                    role: .assistant,
+                    content: "",
+                    summary: "System prompt updated"
+                ))
+            }
+        }
+    }
+
     /// Setting it to `nil` to reset the system prompt
     public func mutateSystemPrompt(_ newPrompt: String?) {
         systemPrompt = newPrompt ?? defaultSystemPrompt

Core/Sources/Service/GUI/ChatProvider+Service.swift

@@ -92,6 +92,12 @@ extension ChatProvider {
                 await commandHandler.handleCustomCommand(command)
             }
         }
+        
+        onSetAsExtraPrompt = { id in
+            Task {
+                await service.setMessageAsExtraPrompt(id: id)
+            }
+        }
     }
 }
 

Core/Sources/SuggestionWidget/ChatProvider.swift

@@ -3,6 +3,7 @@ import Preferences
 import SwiftUI
 
 public final class ChatProvider: ObservableObject {
+    public typealias MessageID = String
     let id = UUID()
     @Published public var history: [ChatMessage] = []
     @Published public var isReceivingMessage = false
@@ -13,10 +14,11 @@ public final class ChatProvider: ObservableObject {
     public var onClear: () -> Void
     public var onClose: () -> Void
     public var onSwitchContext: () -> Void
-    public var onDeleteMessage: (String) -> Void
-    public var onResendMessage: (String) -> Void
+    public var onDeleteMessage: (MessageID) -> Void
+    public var onResendMessage: (MessageID) -> Void
     public var onResetPrompt: () -> Void
     public var onRunCustomCommand: (CustomCommand) -> Void = { _ in }
+    public var onSetAsExtraPrompt: (MessageID) -> Void
 
     public init(
         history: [ChatMessage] = [],
@@ -26,10 +28,11 @@ public final class ChatProvider: ObservableObject {
         onClear: @escaping () -> Void = {},
         onClose: @escaping () -> Void = {},
         onSwitchContext: @escaping () -> Void = {},
-        onDeleteMessage: @escaping (String) -> Void = { _ in },
-        onResendMessage: @escaping (String) -> Void = { _ in },
+        onDeleteMessage: @escaping (MessageID) -> Void = { _ in },
+        onResendMessage: @escaping (MessageID) -> Void = { _ in },
         onResetPrompt: @escaping () -> Void = {},
-        onRunCustomCommand: @escaping (CustomCommand) -> Void = { _ in }
+        onRunCustomCommand: @escaping (CustomCommand) -> Void = { _ in },
+        onSetAsExtraPrompt: @escaping (MessageID) -> Void = { _ in }
     ) {
         self.history = history
         self.isReceivingMessage = isReceivingMessage
@@ -42,19 +45,21 @@ public final class ChatProvider: ObservableObject {
         self.onResendMessage = onResendMessage
         self.onResetPrompt = onResetPrompt
         self.onRunCustomCommand = onRunCustomCommand
+        self.onSetAsExtraPrompt = onSetAsExtraPrompt
     }
 
     public func send(_ message: String) { onMessageSend(message) }
     public func stop() { onStop() }
     public func clear() { onClear() }
     public func close() { onClose() }
     public func switchContext() { onSwitchContext() }
-    public func deleteMessage(id: String) { onDeleteMessage(id) }
-    public func resendMessage(id: String) { onResendMessage(id) }
+    public func deleteMessage(id: MessageID) { onDeleteMessage(id) }
+    public func resendMessage(id: MessageID) { onResendMessage(id) }
     public func resetPrompt() { onResetPrompt() }
     public func triggerCustomCommand(_ command: CustomCommand) {
         onRunCustomCommand(command)
     }
+    public func setAsExtraPrompt(id: MessageID) { onSetAsExtraPrompt(id) }
 }
 
 public struct ChatMessage: Equatable {

Core/Sources/SuggestionWidget/SuggestionPanelContent/ChatPanel.swift

@@ -177,6 +177,10 @@ private struct UserMessage: View {
                 Button("Send Again") {
                     chat.resendMessage(id: id)
                 }
+                
+                Button("Set as Extra System Prompt") {
+                    chat.setAsExtraPrompt(id: id)
+                }
 
                 Divider()
 
@@ -230,6 +234,10 @@ private struct BotMessage: View {
                         NSPasteboard.general.clearContents()
                         NSPasteboard.general.setString(text, forType: .string)
                     }
+                    
+                    Button("Set as Extra System Prompt") {
+                        chat.setAsExtraPrompt(id: id)
+                    }
 
                     Divider()