From 82e5ea059b62096f8b6b420eb26ecd4f3a7404d4 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 30 Jul 2024 22:52:08 -0700 Subject: [PATCH] fix predibase timeout exceptions --- litellm/llms/custom_httpx/http_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/custom_httpx/http_handler.py b/litellm/llms/custom_httpx/http_handler.py index 02d494f9f8..b123514dd2 100644 --- a/litellm/llms/custom_httpx/http_handler.py +++ b/litellm/llms/custom_httpx/http_handler.py @@ -110,7 +110,7 @@ class AsyncHTTPHandler: ) finally: await new_client.aclose() - except httpx.ConnectTimeout: + except httpx.TimeoutException: if data is None: data = {} raise litellm.Timeout( @@ -220,7 +220,7 @@ class HTTPHandler: ) response = self.client.send(req, stream=stream) return response - except httpx.ConnectTimeout: + except httpx.TimeoutException: if data is None: data = {} raise litellm.Timeout(