From b486bb1f8b13e0ad99c15b05a9cc1fb09d85d54c Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 17 May 2024 16:18:25 -0700
Subject: [PATCH] fix(utils.py): exception map bedrock error

---
 litellm/utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/litellm/utils.py b/litellm/utils.py
index 82c31fe4b0..facabd3aa9 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -8713,7 +8713,10 @@ def exception_type(
                         llm_provider="bedrock",
                         response=original_exception.response,
                     )
-                if "Connect timeout on endpoint URL" in error_str:
+                if (
+                    "Connect timeout on endpoint URL" in error_str
+                    or "timed out" in error_str
+                ):
                     exception_mapping_worked = True
                     raise Timeout(
                         message=f"BedrockException: Timeout Error - {error_str}",