feat: error append debug_url

Author: zhangzhiwen

cozepy/exception.py

@@ -11,14 +11,16 @@ class CozeError(Exception):
 
 
 class CozeAPIError(CozeError):
-    def __init__(self, code: Optional[int] = None, msg: str = "", logid: Optional[str] = None):
+    def __init__(self, code: Optional[int] = None, msg: str = "", logid: Optional[str] = None,
+                 debug_url: Optional[str] = None):
         self.code = code
         self.msg = msg
         self.logid = logid
+        self.debug_url = debug_url
         if code and code > 0:
-            super().__init__(f"code: {code}, msg: {msg}, logid: {logid}")
+            super().__init__(f"code: {code}, msg: {msg}, logid: {logid}, debug_url: {debug_url}")
         else:
-            super().__init__(f"msg: {msg}, logid: {logid}")
+            super().__init__(f"msg: {msg}, logid: {logid}, debug_url: {debug_url}")
 
 
 class CozePKCEAuthErrorType(str, Enum):

cozepy/request.py

@@ -475,16 +475,16 @@ def _parse_response(
         if resp_content_type and "audio" in resp_content_type:
             return FileHTTPResponse(response)  # type: ignore
 
-        code, msg, data = self._parse_requests_code_msg(method, url, response, data_field)
+        code, msg, data, debug_url = self._parse_requests_code_msg(method, url, response, data_field)
 
         if code is not None and code > 0:
             log_warning("request %s#%s failed, logid=%s, code=%s, msg=%s", method, url, logid, code, msg)
-            raise CozeAPIError(code, msg, logid)
+            raise CozeAPIError(code, msg, logid, debug_url)
         elif code is None and msg != "":
             log_warning("request %s#%s failed, logid=%s, msg=%s", method, url, logid, msg)
             if msg in COZE_PKCE_AUTH_ERROR_TYPE_ENUMS:
                 raise CozePKCEAuthError(CozePKCEAuthErrorType(msg), logid)
-            raise CozeAPIError(code, msg, logid)
+            raise CozeAPIError(code, msg, logid, debug_url)
         if isinstance(cast, List):
             item_cast = cast[0]
             return [item_cast.model_validate(item) for item in data]
@@ -502,7 +502,7 @@ def _parse_response(
 
     def _parse_requests_code_msg(
         self, method: str, url: str, response: Response, data_field: str = "data"
-    ) -> Tuple[Optional[int], str, Any]:
+    ) -> Tuple[Optional[int], str, Any, Optional[str]]:
         try:
             response.read()
             body = response.json()
@@ -516,11 +516,11 @@ def _parse_requests_code_msg(
             ) from e
 
         if "code" in body and "msg" in body and int(body["code"]) > 0:
-            return int(body["code"]), body["msg"], body.get(data_field)
+            return int(body["code"]), body["msg"], body.get(data_field), body.get("debug_url")
         if "error_code" in body and body["error_code"] in COZE_PKCE_AUTH_ERROR_TYPE_ENUMS:
-            return None, body["error_code"], None
+            return None, body["error_code"], None, body.get("debug_url")
         if "error_message" in body and body["error_message"] != "":
-            return None, body["error_message"], None
+            return None, body["error_message"], None, body.get("debug_url")
         if data_field in body or "debug_url" in body:
             if "first_id" in body:
                 return (
@@ -532,6 +532,7 @@ def _parse_requests_code_msg(
                         "last_id": body["last_id"],
                         "items": body["data"],
                     },
+                    body.get("debug_url")
                 )
             if "debug_url" in body:
                 return (
@@ -542,8 +543,9 @@ def _parse_requests_code_msg(
                         "debug_url": body.get("debug_url") or "",
                         "execute_id": body.get("execute_id") or None,
                     },
+                    body.get("debug_url")
                 )
-            return 0, "", body[data_field]
+            return 0, "", body[data_field], body.get("debug_url")
         if data_field == "data.data":
-            return 0, "", body["data"]["data"]
-        return 0, "", body
+            return 0, "", body["data"]["data"], body.get("debug_url")
+        return 0, "", body, body.get("debug_url")