Debounce prompt to code

Author: intitni

Core/Package.swift

@@ -275,6 +275,7 @@ let package = Package(
                 .product(name: "AppMonitoring", package: "Tool"),
                 .product(name: "ChatTab", package: "Tool"),
                 .product(name: "Logger", package: "Tool"),
+                .product(name: "CustomAsyncAlgorithms", package: "Tool"),
                 .product(name: "AsyncAlgorithms", package: "swift-async-algorithms"),
                 .product(name: "MarkdownUI", package: "swift-markdown-ui"),
                 .product(name: "ComposableArchitecture", package: "swift-composable-architecture"),

Core/Sources/SuggestionWidget/FeatureReducers/PromptToCode.swift

@@ -1,5 +1,6 @@
 import AppKit
 import ComposableArchitecture
+import AsyncAlgorithms
 import Dependencies
 import Foundation
 import PromptToCodeService
@@ -43,7 +44,7 @@ public struct PromptToCode: ReducerProtocol {
                 }
             }
         }
-        
+
         public enum FocusField: Equatable {
             case textField
         }
@@ -113,7 +114,7 @@ public struct PromptToCode: ReducerProtocol {
             self.generateDescriptionRequirement = generateDescriptionRequirement
             self.isAttachedToSelectionRange = isAttachedToSelectionRange
             self.commandName = commandName
-            
+
             if selectionRange?.isEmpty ?? true {
                 self.isAttachedToSelectionRange = false
             }
@@ -151,7 +152,7 @@ public struct PromptToCode: ReducerProtocol {
             switch action {
             case .binding:
                 return .none
-                
+
             case .focusOnTextField:
                 state.focusedField = .textField
                 return .none
@@ -187,13 +188,23 @@ public struct PromptToCode: ReducerProtocol {
                             generateDescriptionRequirement: copiedState
                                 .generateDescriptionRequirement
                         )
-                        #warning("TODO: make the action call debounced.")
-                        for try await fragment in stream {
-                            try Task.checkCancellation()
-                            await send(.modifyCodeChunkReceived(
-                                code: fragment.code,
-                                description: fragment.description
-                            ))
+
+                        if #available(macOS 13.0, *) {
+                            for try await fragment in stream.debounce(for: .milliseconds(400)) {
+                                try Task.checkCancellation()
+                                await send(.modifyCodeChunkReceived(
+                                    code: fragment.code,
+                                    description: fragment.description
+                                ))
+                            }
+                        } else {
+                            for try await fragment in stream {
+                                try Task.checkCancellation()
+                                await send(.modifyCodeChunkReceived(
+                                    code: fragment.code,
+                                    description: fragment.description
+                                ))
+                            }
                         }
                         try Task.checkCancellation()
                         await send(.modifyCodeFinished)

Tool/Package.swift

@@ -44,6 +44,7 @@ let package = Package(
         .library(name: "GitIgnoreCheck", targets: ["GitIgnoreCheck"]),
         .library(name: "DebounceFunction", targets: ["DebounceFunction"]),
         .library(name: "AsyncPassthroughSubject", targets: ["AsyncPassthroughSubject"]),
+        .library(name: "CustomAsyncAlgorithms", targets: ["CustomAsyncAlgorithms"]),
     ],
     dependencies: [
         // A fork of https://github.com/aespinilla/Tiktoken to allow loading from local files.
@@ -85,6 +86,13 @@ let package = Package(
 
         .target(name: "ObjectiveCExceptionHandling"),
 
+        .target(
+            name: "CustomAsyncAlgorithms",
+            dependencies: [
+                .product(name: "AsyncAlgorithms", package: "swift-async-algorithms")
+            ]
+        ),
+
         .target(
             name: "Keychain",
             dependencies: ["Configs", "Preferences"]

Tool/Sources/CustomAsyncAlgorithms/AccumulatedDebounce.swift

@@ -0,0 +1,34 @@
+import Foundation
+
+/// Debounce, at the same time accumulate the values
+public extension AsyncSequence {
+    func accumulatedDebounce<R>(
+        duration: TimeInterval,
+        initialValue: @escaping @autoclosure () -> R,
+        accumulate: @escaping (R, Element) -> R
+    ) -> AsyncThrowingStream<R, Error> {
+        AsyncThrowingStream { continuation in
+            Task {
+                var lastFireTime = Date()
+                var value = initialValue()
+                do {
+                    for try await item in self {
+                        value = accumulate(value, item)
+                        let now = Date()
+                        if now.timeIntervalSince(lastFireTime) > duration {
+                            lastFireTime = now
+                            continuation.yield(value)
+                            value = initialValue()
+                            print(value)
+                        }
+                    }
+                    continuation.yield(value)
+                    continuation.finish()
+                } catch {
+                    continuation.finish(throwing: error)
+                }
+            }
+        }
+    }
+}
+