fix(cli, studio): don't retry on non request/http errors

Author: shcheklein

src/datachain/remote/studio.py

@@ -14,6 +14,7 @@
 from urllib.parse import urlparse, urlunparse
 
 import websockets
+from requests.exceptions import ConnectionError, HTTPError, Timeout
 
 from datachain.config import Config
 from datachain.error import DataChainError
@@ -158,7 +159,7 @@ def _send_request_msgpack(
             message = content.get("message", "")
         return Response(response_data, ok, message)
 
-    @retry_with_backoff(retries=5)
+    @retry_with_backoff(retries=3, errors=(HTTPError, ConnectionError, Timeout))
     def _send_request(
         self, route: str, data: dict[str, Any], method: Optional[str] = "POST"
     ) -> Response[Any]:
@@ -188,7 +189,7 @@ def _send_request(
         )
         try:
             response.raise_for_status()
-        except requests.exceptions.HTTPError:
+        except HTTPError:
             if _is_server_error(response.status_code):
                 # going to retry
                 raise

src/datachain/utils.py

@@ -271,14 +271,14 @@ def flatten(items):
             yield item
 
 
-def retry_with_backoff(retries=5, backoff_sec=1):
+def retry_with_backoff(retries=5, backoff_sec=1, errors=(Exception,)):
     def retry(f):
         def wrapper(*args, **kwargs):
             num_tried = 0
             while True:
                 try:
                     return f(*args, **kwargs)
-                except Exception:
+                except errors:
                     if num_tried == retries:
                         raise
                     sleep = (