From 5700874e2eda09325f94dd3c561c2d0b9d7cad87 Mon Sep 17 00:00:00 2001
From: skyflow-bharti <bharti.sagar@skyflow.com>
Date: Fri, 12 Sep 2025 21:52:59 +0530
Subject: [PATCH] SK-2293 retry on every exception

---
 skyflow/vault/_client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/skyflow/vault/_client.py b/skyflow/vault/_client.py
index 785c42e..1c1236c 100644
--- a/skyflow/vault/_client.py
+++ b/skyflow/vault/_client.py
@@ -71,7 +71,7 @@ def insert(self, records: dict, options: InsertOptions = InsertOptions()):
                     raise SkyflowError(SkyflowErrorCodes.SERVER_ERROR, SkyflowErrorMessages.BATCH_INSERT_FAILURE.value, result, interface=interface)
                 log_info(InfoMessages.INSERT_DATA_SUCCESS.value, interface)
                 return result
-            except requests.exceptions.ConnectionError as err:
+            except Exception as err:
                 if attempt < max_retries:
                     continue
                 else: