From cc10aefc6e9b130230870bf7e3ce0c2759a18e1f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 4 Jun 2024 19:55:15 -0700 Subject: [PATCH] fix(predibase.py): add better error message --- litellm/llms/predibase.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/litellm/llms/predibase.py b/litellm/llms/predibase.py index bd69a42504c..a3245cdac19 100644 --- a/litellm/llms/predibase.py +++ b/litellm/llms/predibase.py @@ -487,9 +487,16 @@ class PredibaseChatCompletion(BaseLLM): self.async_handler = AsyncHTTPHandler( timeout=httpx.Timeout(timeout=600.0, connect=5.0) ) - response = await self.async_handler.post( - api_base, headers=headers, data=json.dumps(data) - ) + try: + response = await self.async_handler.post( + api_base, headers=headers, data=json.dumps(data) + ) + except httpx.HTTPStatusError as e: + raise PredibaseError( + status_code=e.response.status_code, message=e.response.text + ) + except Exception as e: + raise PredibaseError(status_code=500, message=str(e)) return self.process_response( model=model, response=response,