fix(utils.py): handle together ai timeout exception

2024-06-10 17:54:57 -07:00 · 2024-06-10 17:54:57 -07:00 · 622858e37c
commit 622858e37c
parent 8f140cb5ee
1 changed files with 7 additions and 1 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -9695,7 +9695,13 @@ def exception_type(
                        llm_provider="together_ai",
                        response=original_exception.response,
                    )
-
+                elif "A timeout occurred" in error_str:
+                    exception_mapping_worked = True
+                    raise Timeout(
+                        message=f"TogetherAIException - {error_str}",
+                        model=model,
+                        llm_provider="together_ai",
+                    )
                elif (
                    "error" in error_response
                    and "API key doesn't match expected format."