Merge pull request #98 from dule1322/hotfix-refresh-token

Hotfix: Properly update refresh token on request retry
coretex-ai · Nov 28, 2023 · 058dd85 · 058dd85
2 parents cb8eb4f + dbb24df
commit 058dd85
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/coretex/networking/network_manager_base.py b/coretex/networking/network_manager_base.py
@@ -200,11 +200,17 @@ def request(
                 logRequestFailure(endpoint, response)
 
             if self.shouldRetry(retryCount, response):
+                if self._apiToken is not None:
+                    headers[API_TOKEN_HEADER] = self._apiToken
+
                 return self.request(endpoint, requestType, headers, query, body, files, auth, stream, retryCount + 1)
 
             return response
         except:
             if self.shouldRetry(retryCount, None):
+                if self._apiToken is not None:
+                    headers[API_TOKEN_HEADER] = self._apiToken
+
                 return self.request(endpoint, requestType, headers, query, body, files, auth, stream, retryCount + 1)
 
             raise RequestFailedError(endpoint, requestType)