remove completion timeout

twinnydotdev · Mar 6, 2024 · 901d83e · 901d83e
1 parent ca434ed
commit 901d83e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/src/common/constants.ts b/src/common/constants.ts
@@ -17,7 +17,6 @@ export const ALL_BRACKETS = [...OPENING_BRACKETS, ...CLOSING_BRACKETS] as const
 export const BRACKET_REGEX = /^[()[\]{}]+$/
 export const NORMALIZE_REGEX = /\s*\r?\n|\r/g;
 export const LINE_BREAK_REGEX = /\r?\n|\r|\n/g
-export const COMPLETION_TIMEOUT = 20000 // 20 seconds
 export const MAX_CONTEXT_LINE_COUNT = 200
 export const SKIP_DECLARATION_SYMBOLS = ['=', ':']
 export const IMPORT_SEPARATOR = [',', '{']

diff --git a/src/extension/stream.ts b/src/extension/stream.ts
@@ -1,23 +1,12 @@
 import { StreamRequest } from '../common/types'
 import { logStreamOptions, safeParseJsonResponse } from './utils'
-import { COMPLETION_TIMEOUT } from '../common/constants'
 
 export async function streamResponse(request: StreamRequest) {
   logStreamOptions(request)
   const { body, options, onData, onEnd, onError, onStart } = request
   const controller = new AbortController()
-
   const { signal } = controller
 
-  const timeoutId = setTimeout(() => {
-    controller.abort()
-  }, COMPLETION_TIMEOUT)
-
-  const cleanup = () => {
-    clearTimeout(timeoutId)
-    controller.abort()
-  }
-
   try {
     const url = `${options.protocol}://${options.hostname}:${options.port}${options.path}`
     const fetchOptions = {
@@ -85,11 +74,11 @@ export async function streamResponse(request: StreamRequest) {
       if (done) break
     }
 
-    cleanup()
+    controller.abort()
     onEnd?.()
     reader.releaseLock()
   } catch (error: unknown) {
-    cleanup()
+    controller.abort()
     if (error instanceof Error) {
       if (error.name === 'AbortError') {
         onEnd?.()