fix(databricks.py): fix client used

This commit is contained in:
Krrish Dholakia 2024-07-27 13:15:00 -07:00
parent 5b71421a7b
commit 3c77f39751

View file

@ -542,7 +542,7 @@ class DatabricksChatCompletion(BaseLLM):
)
else:
try:
response = self.client.post(
response = client.post(
api_base, headers=headers, data=json.dumps(data)
)
response.raise_for_status()