edit log message

Author: Coniferish

src/unstructured_client/utils/_human_utils.py

@@ -93,11 +93,16 @@ def wrapper(*args: _P.args, **kwargs: _P.kwargs) -> operations.PartitionResponse
     return wrapper
 
 
-def log_retries(retry_count: int, sleep: float):
+def log_retries(retry_count: int, sleep: float, exception: Exception):
     """Function for logging retries to give users visibility into requests."""
     logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s', stream=sys.stdout)
     logger = logging.getLogger('unstructured-client')
     logger.setLevel(logging.INFO)
     logger.info(
-        "Retry attempt #%s. Sleeping %s seconds before retry.", retry_count, round(sleep, 1)
+        "Response status code: %s Retry attempt #%s. Sleeping %s seconds before retry.", 
+        exception.response.status_code,
+        retry_count,
+        round(sleep, 1),
     )
+    if bool(exception.response.text):
+        logger.info(exception.response.text)

src/unstructured_client/utils/retries.py

@@ -118,6 +118,6 @@ def retry_with_backoff(func, initial_interval=500, max_interval=60000, exponent=
                      exponent**retries + random.uniform(0, 1))
             if sleep > max_interval/1000:
                 sleep = max_interval/1000
-            log_retries(retry_count=retries+1, sleep=sleep)  # human code
+            log_retries(retry_count=retries+1, sleep=sleep, exception=exception)  # human code
             time.sleep(sleep)
             retries += 1