From 44e611cec71020b8f30c61313403295c6dd5c525 Mon Sep 17 00:00:00 2001 From: skyflow-vivek Date: Mon, 29 Sep 2025 19:41:55 +0530 Subject: [PATCH] SK-2325 Fix retry logic when continueOnError is set to true --- skyflow/vault/_client.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/skyflow/vault/_client.py b/skyflow/vault/_client.py index f3e02de..24ba524 100644 --- a/skyflow/vault/_client.py +++ b/skyflow/vault/_client.py @@ -64,11 +64,10 @@ def insert(self, records: dict, options: InsertOptions = InsertOptions()): result, partial = convertResponse(records, processedResponse, options) if partial: log_error(SkyflowErrorMessages.BATCH_INSERT_PARTIAL_SUCCESS.value, interface) - raise SkyflowError(SkyflowErrorCodes.PARTIAL_SUCCESS, SkyflowErrorMessages.BATCH_INSERT_PARTIAL_SUCCESS.value, result, interface=interface) - if 'records' not in result: + elif 'records' not in result: log_error(SkyflowErrorMessages.BATCH_INSERT_FAILURE.value, interface) - raise SkyflowError(SkyflowErrorCodes.SERVER_ERROR, SkyflowErrorMessages.BATCH_INSERT_FAILURE.value, result, interface=interface) - log_info(InfoMessages.INSERT_DATA_SUCCESS.value, interface) + else: + log_info(InfoMessages.INSERT_DATA_SUCCESS.value, interface) return result except Exception as err: if attempt < max_retries: