development > main (#393)

* fittencode_prevent

* fix#392

* lint

* 3.19.2

* lint

* lint

* lint

---------

Co-authored-by: ZhongGs <[email protected]>

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.19.1",
+  "version": "3.19.2",
   "icon": "assets/icon.png",
   "keywords": [
     "code-inference",

src/common/constants.ts

@@ -368,3 +368,9 @@ export const MULTI_LINE_DELIMITERS = ["\n\n", "\r\n\r\n"]
 export const SYMMETRY_EMITTER_KEY = {
   inference: "inference",
 }
+
+//Define an array containing all the error messages that need to be detected when fetch error occurred
+export const knownErrorMessages = [
+"First parameter has member 'readable' that is not a ReadableStream.", //This error occurs When plugins such as Fitten Code are enabled
+"The 'transform.readable' property must be an instance of ReadableStream. Received an instance of h" //When you try to enable the Node.js compatibility mode Compat to solve the problem, this error may pop up
+];

src/common/logger.ts

@@ -3,7 +3,14 @@ import { Base } from "../extension/base"
 export class Logger extends Base {
   private static instance: Logger
 
-  constructor () {
+  public static ErrorType = {
+    Default: 0, // default color
+    Fetch_Error: 91, // red -- fetch error
+    Abort: 90, // gray -- abort
+    Timeout: 33 // yellow -- timeout
+  }
+
+  constructor() {
     super()
   }
 
@@ -23,6 +30,20 @@ export class Logger extends Base {
     if (!this.config.enableLogging) return
     console.error(err.message)
   }
+
+  public logConsoleError(
+    errorKey: number,
+    message: string,
+    error: Error | string
+  ): void {
+    const color = errorKey
+    const coloredMessage = `\x1b[91m [ERROR_twinny] \x1b[32m Message: ${message} \n \x1b[${color}m Error Type: ${
+      error instanceof Error ? error.name : "Unknown Error"
+    } \n  Error Message: ${
+      error instanceof Error ? error.message : error
+    } \n \x1b[31m`
+    console.error(coloredMessage, error)
+  }
 }
 
 export const logger = Logger.getInstance()

src/common/types.ts

@@ -92,6 +92,7 @@ export interface ServerMessage<T = LanguageType> {
     type: string
   }
 }
+
 export interface Message {
   role: string
   content: string | undefined

src/extension/api.ts

@@ -1,6 +1,13 @@
+import { Logger } from "../common/logger"
 import { StreamRequest } from "../common/types"
 
-import { logStreamOptions, safeParseJsonResponse } from "./utils"
+import {
+  logStreamOptions,
+  notifyKnownErrors,
+  safeParseJsonResponse
+} from "./utils"
+
+const log = Logger.getInstance()
 
 export async function streamResponse(request: StreamRequest) {
   logStreamOptions(request)
@@ -11,15 +18,22 @@ export async function streamResponse(request: StreamRequest) {
   const timeOut = setTimeout(() => {
     controller.abort()
     onError?.(new Error("Request timed out"))
+    log.logConsoleError(
+      Logger.ErrorType.Timeout,
+      "Failed to establish connection",
+      new Error("Request timed out")
+    )
   }, 25000)
 
   try {
-    const url = `${options.protocol}://${options.hostname}${options.port ? `:${options.port}` : ""}${options.path}`
+    const url = `${options.protocol}://${options.hostname}${
+      options.port ? `:${options.port}` : ""
+    }${options.path}`
     const fetchOptions = {
       method: options.method,
       headers: options.headers,
       body: JSON.stringify(body),
-      signal: controller.signal,
+      signal: controller.signal
     }
 
     const response = await fetch(url, fetchOptions)
@@ -67,7 +81,7 @@ export async function streamResponse(request: StreamRequest) {
                 onError?.(new Error("Error parsing JSON data from event"))
               }
             }
-          },
+          }
         })
       )
       .getReader()
@@ -89,8 +103,9 @@ export async function streamResponse(request: StreamRequest) {
       if (error.name === "AbortError") {
         onEnd?.()
       } else {
-        console.error("Fetch error:", error)
+        log.logConsoleError(Logger.ErrorType.Fetch_Error, "Fetch error", error)
         onError?.(error)
+        notifyKnownErrors(error)
       }
     }
   }
@@ -100,14 +115,15 @@ export async function fetchEmbedding(request: StreamRequest) {
   const { body, options, onData } = request
   const controller = new AbortController()
 
-
   try {
-    const url = `${options.protocol}://${options.hostname}${options.port ? `:${options.port}` : ""}${options.path}`
+    const url = `${options.protocol}://${options.hostname}${
+      options.port ? `:${options.port}` : ""
+    }${options.path}`
     const fetchOptions = {
       method: options.method,
       headers: options.headers,
       body: JSON.stringify(body),
-      signal: controller.signal,
+      signal: controller.signal
     }
 
     const response = await fetch(url, fetchOptions)
@@ -123,10 +139,10 @@ export async function fetchEmbedding(request: StreamRequest) {
     const data = await response.json()
 
     onData(data)
-
   } catch (error: unknown) {
     if (error instanceof Error) {
-      console.error("Fetch error:", error)
+      log.logConsoleError(Logger.ErrorType.Fetch_Error, "Fetch error", error)
+      notifyKnownErrors(error)
     }
   }
 }

src/extension/chat-service.ts

@@ -363,7 +363,7 @@ export class ChatService extends Base {
       type: EVENT_NAME.twinnyOnEnd,
       value: {
         error: true,
-        errorMessage: error.message,
+        errorMessage: `==## ERROR ##== : ${error.message}`, // Highlight errors on webview
       },
     } as ServerMessage)
   }

src/extension/utils.ts

@@ -3,6 +3,7 @@ import fs from "fs"
 import ignore from "ignore"
 import path from "path"
 import * as util from "util"
+import * as vscode from "vscode"
 import {
   ColorThemeKind,
   ExtensionContext,
@@ -24,6 +25,7 @@ import {
   defaultChunkOptions,
   EVENT_NAME,
   EXTENSION_CONTEXT_NAME,
+  knownErrorMessages,
   LINE_BREAK_REGEX,
   MULTILINE_TYPES,
   NORMALIZE_REGEX,
@@ -763,3 +765,24 @@ const calculateTotalCharacters = (messages: Message[] | undefined): number => {
     return acc + (typeof msg.content === "string" ? msg.content.length : 0)
   }, 0)
 }
+
+export function notifyKnownErrors(error: Error) {
+  if (knownErrorMessages.some((msg) => error.message.includes(msg))) {
+    vscode.window
+      .showInformationMessage(
+        "Besides Twinny, there may be other AI extensions being enabled (such as Fitten Code) that are affecting the behavior of the fetch API or ReadableStream used in the Twinny plugin. We recommend that you disable that AI plugin for the smooth use of Twinny",
+        "View extensions",
+        "Restart Visual Studio Code (after disabling related extensions)"
+      )
+      .then((selected) => {
+        if (selected === "View extensions") {
+          vscode.commands.executeCommand("workbench.view.extensions")
+        } else if (
+          selected ===
+          "Restart Visual Studio Code (after disabling related extensions)"
+        ) {
+          vscode.commands.executeCommand("workbench.action.reloadWindow")
+        }
+      })
+  }
+}