fix(databricks.py): fix error handling

This commit is contained in:
Krrish Dholakia 2024-08-01 15:15:45 -07:00
parent 1affd4e2ca
commit 2a7ed84a23

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.")