fix(databricks.py): fix client used

This commit is contained in:
Krrish Dholakia 2024-07-27 13:15:00 -07:00
parent ce7257ec5e
commit cfc061e38e

View file

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