feat(providers): added retry mechanism to google chat provider to overcome rate limiting

Author: pehlicd

keep/providers/google_chat_provider/google_chat_provider.py

@@ -69,14 +69,21 @@ def _notify(self, message="", **kwargs: dict):
             raise ProviderException("Message is required")
 
         def __send_message(url, body, headers, retries=3):
-            for _ in range(retries):
+            for attempt in range(retries):
                 try:
                     resp = requests.post(url, json=body, headers=headers)
                     if resp.status_code == http.HTTPStatus.OK:
                         return resp
+
+                    self.logger.warning(f"Attempt {attempt + 1} failed with status code {resp.status_code}")
+
                 except requests.exceptions.RequestException as e:
-                    self.logger.error(f"Failed to send message to Google Chat: {e}")
-                time.sleep(1)
+                    self.logger.error(f"Attempt {attempt + 1} failed: {e}")
+
+                if attempt < retries - 1:
+                    time.sleep(1)
+
+            raise requests.exceptions.RequestException(f"Failed to notify message after {retries} attempts")
 
         payload = {
             "text": message,
@@ -85,11 +92,8 @@ def __send_message(url, body, headers, retries=3):
         request_headers = {"Content-Type": "application/json; charset=UTF-8"}
 
         response = __send_message(webhook_url, body=payload, headers=request_headers)
-
-        if not response.ok:
-            raise ProviderException(
-                f"Failed to notify message to Google Chat: {response.text}"
-            )
+        if response.status_code != http.HTTPStatus.OK:
+            raise ProviderException(f"Failed to notify message to Google Chat: {response.text}")
 
         self.logger.debug("Alert message sent to Google Chat successfully")
         return "Alert message sent to Google Chat successfully"