Handle lmsys cloudflare error

Author: wong2

src/app/bots/gradio/index.ts

@@ -104,7 +104,7 @@ export class GradioBot extends AbstractBot {
     try {
       await wsp.open()
     } catch (err) {
-      console.error('lmsys ws open error', err)
+      console.error('WS open error', err)
       throw new ChatError('Failed to establish websocket connection.', ErrorCode.NETWORK_ERROR)
     }
 

src/app/bots/lmsys/index.ts

@@ -1,5 +1,6 @@
 import WebSocketAsPromised from 'websocket-as-promised'
 import { GradioBot } from '../gradio'
+import { ChatError, ErrorCode } from '~utils/errors'
 
 export class LMSYSBot extends GradioBot {
   constructor(model: string) {
@@ -27,7 +28,10 @@ export class LMSYSBot extends GradioBot {
           resolve()
         }
       })
-      wsp.open().catch((err) => reject(err))
+      wsp.open().catch((err) => {
+        console.error('lmsys ws open error', err)
+        reject(new ChatError('Failed to establish websocket connection.', ErrorCode.LMSYS_WS_ERROR))
+      })
     })
   }
 

src/app/components/Chat/ErrorAction.tsx

@@ -139,6 +139,17 @@ const ErrorAction: FC<{ error: ChatError }> = ({ error }) => {
       </p>
     )
   }
+  if (error.code === ErrorCode.LMSYS_WS_ERROR) {
+    return (
+      <p className="ml-2 text-secondary-text text-sm">
+        Please visit{' '}
+        <a href="https://chat.lmsys.org" target="_blank" rel="noreferrer" className="underline">
+          LMSYS.org
+        </a>{' '}
+        and try again
+      </p>
+    )
+  }
   if (
     error.code === ErrorCode.NETWORK_ERROR ||
     (error.code === ErrorCode.UNKOWN_ERROR && error.message.includes('Failed to fetch'))

src/utils/errors.ts

@@ -21,6 +21,7 @@ export enum ErrorCode {
   CLAUDE_WEB_UNAVAILABLE = 'CLAUDE_WEB_UNAVAILABLE',
   QIANWEN_WEB_UNAUTHORIZED = 'QIANWEN_WEB_UNAUTHORIZED',
   BAICHUAN_WEB_UNAUTHORIZED = 'BAICHUAN_WEB_UNAUTHORIZED',
+  LMSYS_WS_ERROR = 'LMSYS_WS_ERROR',
 }
 
 export class ChatError extends Error {