Disable stop check when line limit <= 0

intitni · intitni · commit 497eacb9e9eb · 2024-04-15T19:24:16.000+08:00
diff --git a/Core/Sources/CodeCompletionService/StreamLineLimiter.swift b/Core/Sources/CodeCompletionService/StreamLineLimiter.swift
@@ -30,11 +30,15 @@ final class StreamLineLimiter {
             currentLine = lastLine ?? ""
         }
 
-        let stopResult = strategy.shouldStop(
-            existedLines: existedLines,
-            currentLine: currentLine,
-            proposedLineLimit: lineLimit
-        )
+        let stopResult = if lineLimit <= 0 {
+            StreamStopStrategyResult.continue
+        } else {
+            strategy.shouldStop(
+                existedLines: existedLines,
+                currentLine: currentLine,
+                proposedLineLimit: lineLimit
+            )
+        }
 
         switch stopResult {
         case .continue:
