From d4378143f1f8740be255aabf7ee9890f8a16e71a Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 1 Jun 2024 17:42:37 -0700
Subject: [PATCH] fix current_attempt, num_retries not defined

---
 litellm/router.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/litellm/router.py b/litellm/router.py
index 510d179bab..4ecd41c921 100644
--- a/litellm/router.py
+++ b/litellm/router.py
@@ -1972,6 +1972,8 @@ class Router:
             response = await original_function(*args, **kwargs)
             return response
         except Exception as e:
+            num_retries = None
+            current_attempt = None
             original_exception = e
             """
             Retry Logic
@@ -2245,6 +2247,8 @@ class Router:
             response = original_function(*args, **kwargs)
             return response
         except Exception as e:
+            num_retries = None
+            current_attempt = None
             original_exception = e
             ### CHECK IF RATE LIMIT / CONTEXT WINDOW ERROR
             _healthy_deployments = self._get_healthy_deployments(
@@ -2299,6 +2303,7 @@ class Router:
             if type(original_exception) in litellm.LITELLM_EXCEPTION_TYPES:
                 original_exception.max_retries = num_retries
                 original_exception.num_retries = current_attempt
+
             raise original_exception
 
     ### HELPER FUNCTIONS