Merge pull request #383 from twinnydotdev/development

development > main

package.json

@@ -2,7 +2,7 @@
   "name": "twinny",
   "displayName": "twinny - AI Code Completion and Chat",
   "description": "Locally hosted AI code completion plugin for vscode",
-  "version": "3.18.11",
+  "version": "3.18.12",
   "icon": "assets/icon.png",
   "keywords": [
     "code-inference",

src/extension/api.ts

@@ -8,6 +8,11 @@ export async function streamResponse(request: StreamRequest) {
   const controller = new AbortController()
   const { signal } = controller
 
+  const timeOut = setTimeout(() => {
+    controller.abort()
+    onError?.(new Error("Request timed out"))
+  }, 25000)
+
   try {
     const url = `${options.protocol}://${options.hostname}${options.port ? `:${options.port}` : ""}${options.path}`
     const fetchOptions = {
@@ -18,6 +23,7 @@ export async function streamResponse(request: StreamRequest) {
     }
 
     const response = await fetch(url, fetchOptions)
+    clearTimeout(timeOut)
 
     if (!response.ok) {
       throw new Error(`Server responded with status code: ${response.status}`)
@@ -77,12 +83,14 @@ export async function streamResponse(request: StreamRequest) {
     onEnd?.()
     reader.releaseLock()
   } catch (error: unknown) {
+    clearTimeout(timeOut)
     controller.abort()
     if (error instanceof Error) {
       if (error.name === "AbortError") {
         onEnd?.()
       } else {
         console.error("Fetch error:", error)
+        onError?.(error)
       }
     }
   }

src/extension/review-service.ts

@@ -220,6 +220,15 @@ export class GithubService extends ConversationHistory {
             })
             this._completion = ""
           },
+          onError: (error: Error) => {
+            this.webView?.postMessage({
+              type: EVENT_NAME.twinnyOnEnd,
+              value: {
+                error: true,
+                errorMessage: error.message,
+              },
+            } as ServerMessage)
+          }
         })
       } catch (e) {
         return reject(e)