【会员系统】AI请求优化

PDF Office/PDF Master/Class/AIInfo/KMAIRequestServerManager.swift

@@ -317,18 +317,27 @@ class ResultWrapper: NSObject {
             if responseObject != nil {
                 let data: NSDictionary = responseObject as? NSDictionary ?? [:]
                 if data.count > 0 {
-                    let wrapper = ResultWrapper(success: true, resultData: data)
-                    if let value = data["fileKey"] {
-                        wrapper.content = value as! String
+                    let code: Int = data["code"] as? Int ?? 06005
+                    if code == 200 {
+                        let wrapper = ResultWrapper(success: true, resultData: data)
+                        if let value = data["fileKey"] {
+                            wrapper.content = value as! String
+                        }
+                        complete(wrapper)
+                    } else {
+                        let wrapper = ResultWrapper(success: false, resultData: data)
+                        if let value = data["msg"] {
+                            wrapper.content = value as! String
+                        }
+                        complete(wrapper)  
                     }
-                    complete(wrapper)
                 } else {
                     let error = NSError(domain: "unknown error", code: 404)
-                     complete(ResultWrapper(success: false, content: "unknown error"))
+                    complete(ResultWrapper(success: false, content: "unknown error"))
                 }
             } else {
                 let error = NSError(domain: "unknown error", code: 404)
-                 complete(ResultWrapper(success: false, content: "unknown error"))
+                complete(ResultWrapper(success: false, content: "unknown error"))
             }
         }
     }