fix(utils.py): exception map bedrock error

This commit is contained in:
Krrish Dholakia 2024-05-17 16:18:25 -07:00
parent 5d24a72b7e
commit b486bb1f8b

View file

@ -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}",