fix(databricks.py): fix error handling

This commit is contained in:
Krrish Dholakia 2024-08-01 15:15:45 -07:00
parent 217f597860
commit 47448e4bc0

View file

@ -333,7 +333,7 @@ class DatabricksChatCompletion(BaseLLM):
except httpx.HTTPStatusError as e:
raise DatabricksError(
status_code=e.response.status_code,
message=response.text if response else str(e),
message=e.response.text,
)
except httpx.TimeoutException as e:
raise DatabricksError(status_code=408, message="Timeout error occurred.")