improve error code

Author: arvinxx

src/app/(backend)/middleware/auth/index.ts

@@ -48,6 +48,15 @@ export const checkAuth =
         nextAuthAuthorized: oauthAuthorized,
       });
     } catch (e) {
+      // if the error is not a ChatCompletionErrorPayload, it means the application error
+      if (!(e as ChatCompletionErrorPayload).errorType) {
+        if ((e as any).code === 'ERR_JWT_EXPIRED')
+          return createErrorResponse(ChatErrorType.SystemTimeNotMatchError, e);
+
+        // other issue will be internal server error
+        return createErrorResponse(ChatErrorType.InternalServerError, e);
+      }
+
       const {
         errorType = ChatErrorType.InternalServerError,
         error: errorContent,

src/features/Conversation/Error/OpenAiBizError.tsx

@@ -14,7 +14,6 @@ import ClerkLogin from './ClerkLogin';
 import ErrorJsonViewer from './ErrorJsonViewer';
 import InvalidAPIKey from './InvalidAPIKey';
 import InvalidAccessCode from './InvalidAccessCode';
-import OpenAiBizError from './OpenAiBizError';
 
 const loading = () => <Skeleton active />;
 
@@ -34,8 +33,11 @@ const getErrorAlertConfig = (
     };
 
   switch (errorType) {
+    case ChatErrorType.SystemTimeNotMatchError:
     case AgentRuntimeErrorType.PermissionDenied:
+    case AgentRuntimeErrorType.InsufficientQuota:
     case AgentRuntimeErrorType.QuotaLimitReached:
+    case AgentRuntimeErrorType.ExceededContextWindow:
     case AgentRuntimeErrorType.LocationNotSupportError: {
       return {
         type: 'warning',
@@ -82,10 +84,6 @@ const ErrorMessageExtra = memo<{ data: ChatMessage }>(({ data }) => {
       return <PluginSettings id={data.id} plugin={data.plugin} />;
     }
 
-    case AgentRuntimeErrorType.OpenAIBizError: {
-      return <OpenAiBizError {...data} />;
-    }
-
     case AgentRuntimeErrorType.OllamaBizError: {
       return <OllamaBizError {...data} />;
     }

src/features/Conversation/Error/index.tsx

@@ -3,8 +3,12 @@
 export const AgentRuntimeErrorType = {
   AgentRuntimeError: 'AgentRuntimeError', // Agent Runtime 模块运行时错误
   LocationNotSupportError: 'LocationNotSupportError',
+
   QuotaLimitReached: 'QuotaLimitReached',
+  InsufficientQuota: 'InsufficientQuota',
+
   PermissionDenied: 'PermissionDenied',
+  ExceededContextWindow: 'ExceededContextWindow',
 
   InvalidProviderAPIKey: 'InvalidProviderAPIKey',
   ProviderBizError: 'ProviderBizError',
@@ -24,10 +28,6 @@ export const AgentRuntimeErrorType = {
    * @deprecated
    */
   NoOpenAIAPIKey: 'NoOpenAIAPIKey',
-  /**
-   * @deprecated
-   */
-  OpenAIBizError: 'OpenAIBizError',
 } as const;
 
 export const AGENT_RUNTIME_ERROR_SET = new Set<string>(Object.values(AgentRuntimeErrorType));

src/libs/agent-runtime/error.ts

@@ -386,6 +386,28 @@ export const LobeOpenAICompatibleFactory = <T extends Record<string, any> = any>
 
       const { errorResult, RuntimeError } = handleOpenAIError(error);
 
+      switch (errorResult.code) {
+        case 'insufficient_quota': {
+          return AgentRuntimeError.chat({
+            endpoint: desensitizedEndpoint,
+            error: errorResult,
+            errorType: AgentRuntimeErrorType.InsufficientQuota,
+            provider: provider as ModelProvider,
+          });
+        }
+
+        // content too long
+        case 'context_length_exceeded':
+        case 'string_above_max_length': {
+          return AgentRuntimeError.chat({
+            endpoint: desensitizedEndpoint,
+            error: errorResult,
+            errorType: AgentRuntimeErrorType.ExceededContextWindow,
+            provider: provider as ModelProvider,
+          });
+        }
+      }
+
       return AgentRuntimeError.chat({
         endpoint: desensitizedEndpoint,
         error: errorResult,

src/libs/agent-runtime/utils/openaiCompatibleFactory/index.ts

@@ -90,8 +90,12 @@ export default {
 
     InvalidAccessCode: '密码不正确或为空，请输入正确的访问密码，或者添加自定义 API Key',
     InvalidClerkUser: '很抱歉，你当前尚未登录，请先登录或注册账号后继续操作',
+    SystemTimeNotMatchError: '很抱歉，您的系统时间和服务器不匹配，请检查您的系统时间后重试',
     LocationNotSupportError:
       '很抱歉，你的所在地区不支持此模型服务，可能是由于区域限制或服务未开通。请确认当前地区是否支持使用此服务，或尝试使用切换到其他地区后重试。',
+    InsufficientQuota:
+      '很抱歉，该密钥的配额(quota)已达上限，请检查账户余额是否充足，或增大密钥配额后再试',
+    ExceededContextWindow: '当前请求内容超出模型可处理的长度，请减少内容量后重试',
     QuotaLimitReached:
       '很抱歉，当前 Token 用量或请求次数已达该密钥的配额(quota)上限，请增加该密钥的配额或稍后再试',
     PermissionDenied: '很抱歉，你没有权限访问该服务，请检查你的密钥是否有访问权限',
@@ -101,10 +105,6 @@ export default {
      * @deprecated
      */
     NoOpenAIAPIKey: 'OpenAI API Key 不正确或为空，请添加自定义 OpenAI API Key',
-    /**
-     * @deprecated
-     */
-    OpenAIBizError: '请求 OpenAI 服务出错，请根据以下信息排查或重试',
 
     InvalidVertexCredentials: 'Vertex 鉴权未通过，请检查鉴权凭证后重试',
     InvalidBedrockCredentials: 'Bedrock 鉴权未通过，请检查 AccessKeyId/SecretAccessKey 后重试',

src/locales/default/error.ts

@@ -13,6 +13,7 @@ export const ChatErrorType = {
   OllamaServiceUnavailable: 'OllamaServiceUnavailable', // 未启动/检测到 Ollama 服务
   PluginFailToTransformArguments: 'PluginFailToTransformArguments',
   UnknownChatFetchError: 'UnknownChatFetchError',
+  SystemTimeNotMatchError: 'SystemTimeNotMatchError',
 
   // ******* 客户端错误 ******* //
   BadRequest: 400,

src/types/fetch.ts

@@ -33,12 +33,6 @@ describe('createErrorResponse', () => {
   });
 
   describe('Provider Biz Error', () => {
-    it('returns a 471 status for OpenAIBizError error type', () => {
-      const errorType = AgentRuntimeErrorType.OpenAIBizError;
-      const response = createErrorResponse(errorType);
-      expect(response.status).toBe(471);
-    });
-
     it('returns a 471 status for ProviderBizError error type', () => {
       const errorType = AgentRuntimeErrorType.ProviderBizError;
       const response = createErrorResponse(errorType);
@@ -50,12 +44,6 @@ describe('createErrorResponse', () => {
       const response = createErrorResponse(errorType);
       expect(response.status).toBe(470);
     });
-
-    it('returns a 471 status for OpenAIBizError error type', () => {
-      const errorType = AgentRuntimeErrorType.OpenAIBizError;
-      const response = createErrorResponse(errorType as any);
-      expect(response.status).toBe(471);
-    });
   });
 
   // 测试状态码不在200-599范围内的情况

src/utils/errorResponse.test.ts

@@ -12,10 +12,16 @@ const getStatus = (errorType: ILobeAgentRuntimeErrorType | ErrorType) => {
       return 401;
     }
 
+    case AgentRuntimeErrorType.ExceededContextWindow:
+    case ChatErrorType.SystemTimeNotMatchError: {
+      return 400;
+    }
+
     case AgentRuntimeErrorType.LocationNotSupportError: {
       return 403;
     }
 
+    case AgentRuntimeErrorType.InsufficientQuota:
     case AgentRuntimeErrorType.QuotaLimitReached: {
       return 429;
     }
@@ -25,8 +31,7 @@ const getStatus = (errorType: ILobeAgentRuntimeErrorType | ErrorType) => {
       return 470;
     }
 
-    case AgentRuntimeErrorType.ProviderBizError:
-    case AgentRuntimeErrorType.OpenAIBizError: {
+    case AgentRuntimeErrorType.ProviderBizError: {
       return 471;
     }