fix(utils.py): map cohere timeout error

2025-04-25 18:54:30 +00:00 · 2024-07-31 15:15:18 -07:00 · 2024-07-31 15:15:18 -07:00 · bd68714f51
commit bd68714f51
parent 6202f9bbb0
2 changed files with 9 additions and 0 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -7328,6 +7328,13 @@ def exception_type(
                            model=model,
                            response=original_exception.response,
                        )
+                    elif original_exception.status_code == 408:
+                        exception_mapping_worked = True
+                        raise Timeout(
+                            message=f"CohereException - {original_exception.message}",
+                            llm_provider="cohere",
+                            model=model,
+                        )
                    elif original_exception.status_code == 500:
                        exception_mapping_worked = True
                        raise ServiceUnavailableError(