Upload mt_openapi.py

mt_openapi.py

@@ -2,6 +2,7 @@ from tenacity import retry, stop_after_delay, wait_fixed, retry_if_result
 import requests
 import json
 import os
+import logging
 
 
 class MTOpenApiClient:
@@ -87,12 +88,13 @@ class MTOpenApiClient:
         data={"msg_id": msg_id}
         response = requests.post(url, json=data, headers=headers)
         response.raise_for_status()  # Raise HTTP errors
+        logging.info(f"fetch_response: {response.json()}")
         return response.json()
 
     @retry(
         stop=stop_after_delay(1000),  # Stop after 100 seconds
         wait=wait_fixed(1),  # Wait 1 second between retries
-        retry=retry_if_result(lambda res: res.get("error_code") != 0 and res.get("error_code") != 2)  # Retry if error_code != 0
+        retry=retry_if_result(lambda res: res.get("error_code") == 4)  # Retry if error_code != 0
     )
     def get_res(self, msg_id):
         """
@@ -130,7 +132,7 @@ class MTOpenApiClient:
                 result = self.get_res(msg_id=msg_id)
                 if result.get("error_code", 0) == 0:
                     return result
-                    
+
             except requests.RequestException:
                 if attempt == max_retries - 1:
                     raise