♻️ refactor: refactor the google ai return error (#6978)

Author: arvinxx

src/libs/agent-runtime/google/index.test.ts

@@ -340,7 +340,9 @@ describe('LobeGoogleAI', () => {
           expect(e).toEqual({
             errorType: bizErrorType,
             error: {
-              message: `[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:streamGenerateContent?alt=sse: [400 Bad Request] API key not valid. Please pass a valid API key. [{"@type":"type.googleapis.com/google.rpc.ErrorInfo","reason":"Error","domain":"googleapis.com","metadata":{"service":"generativelanguage.googleapis.com}}]`,
+              message: `API key not valid. Please pass a valid API key. [{"@type":"type.googleapis.com/google.rpc.ErrorInfo","reason":"Error","domain":"googleapis.com","metadata":{"service":"generativelanguage.googleapis.com}}]`,
+              statusCode: 400,
+              statusCodeText: '[400 Bad Request]',
             },
             provider,
           });

src/libs/agent-runtime/google/index.ts

@@ -325,12 +325,12 @@ export class LobeGoogleAI implements LobeRuntimeAI {
     if (message.includes('location is not supported'))
       return { error: { message }, errorType: AgentRuntimeErrorType.LocationNotSupportError };
 
-    try {
-      const startIndex = message.lastIndexOf('[');
-      if (startIndex === -1) {
-        return defaultError;
-      }
+    const startIndex = message.lastIndexOf('[');
+    if (startIndex === -1) {
+      return defaultError;
+    }
 
+    try {
       // 从开始位置截取字符串到最后
       const jsonString = message.slice(startIndex);
 
@@ -349,9 +349,18 @@ export class LobeGoogleAI implements LobeRuntimeAI {
         }
       }
     } catch {
-      // 如果解析失败，则返回原始错误消息
-      return defaultError;
+      //
     }
+
+    const errorObj = this.extractErrorObjectFromError(message);
+
+    const { errorDetails } = errorObj;
+
+    if (errorDetails) {
+      return { error: errorDetails, errorType: AgentRuntimeErrorType.ProviderBizError };
+    }
+
+    return defaultError;
   }
 
   private buildGoogleTools(
@@ -392,6 +401,40 @@ export class LobeGoogleAI implements LobeRuntimeAI {
       },
     };
   };
+
+  private extractErrorObjectFromError(message: string) {
+    // 使用正则表达式匹配状态码部分 [数字 描述文本]
+    const regex = /^(.*?)(\[\d+ [^\]]+])(.*)$/;
+    const match = message.match(regex);
+
+    if (match) {
+      const prefix = match[1].trim();
+      const statusCodeWithBrackets = match[2].trim();
+      const message = match[3].trim();
+
+      // 提取状态码数字
+      const statusCodeMatch = statusCodeWithBrackets.match(/\[(\d+)/);
+      const statusCode = statusCodeMatch ? parseInt(statusCodeMatch[1]) : null;
+
+      // 创建包含状态码和消息的JSON
+      const resultJson = {
+        message: message,
+        statusCode: statusCode,
+        statusCodeText: statusCodeWithBrackets,
+      };
+
+      return {
+        errorDetails: resultJson,
+        prefix: prefix,
+      };
+    }
+
+    // 如果无法匹配，返回原始消息
+    return {
+      errorDetails: null,
+      prefix: message,
+    };
+  }
 }
 
 export default LobeGoogleAI;