test(test_streaming.py): add exception mapping for palm timeout error

2025-04-26 11:14:04 +00:00 · 2024-02-22 21:36:57 -08:00 · 2024-02-22 21:36:57 -08:00 · e07cf33cba
commit e07cf33cba
parent acae98fd50
2 changed files with 10 additions and 0 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -6824,6 +6824,14 @@ def exception_type(
                        llm_provider="palm",
                        response=original_exception.response,
                    )
+                if "504 Deadline expired before operation could complete." in error_str:
+                    exception_mapping_worked = True
+                    raise Timeout(
+                        message=f"PalmException - {original_exception.message}",
+                        model=model,
+                        llm_provider="palm",
+                        request=original_exception.request,
+                    )
                if "400 Request payload size exceeds" in error_str:
                    exception_mapping_worked = True
                    raise ContextWindowExceededError(