Update - handle explicit error + improve code-quality

Agenta-AI · Nov 21, 2023 · 26c6c0a · 26c6c0a
1 parent bb914fd
commit 26c6c0a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/agenta-backend/agenta_backend/utils/redis_utils.py b/agenta-backend/agenta_backend/utils/redis_utils.py
@@ -13,6 +13,10 @@ def redis_connection() -> redis.Redis:
 
     try:
         redis_client = redis.from_url(url=os.environ.get("REDIS_URL", None))
-    except (ConnectionRefusedError, ConnectionError):
-        raise RuntimeError("Could not connect to redis service.")
+    except ConnectionRefusedError:
+        raise ConnectionRefusedError(
+            "Refuse connecting to redis service. Kindly check redis url."
+        )
+    except ConnectionError:
+        raise ConnectionError("Could not connect to redis service.")
     return redis_client