phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(databricks/common_utils.py): fix custom endpoint check (#9925)

Browse source 
* fix(databricks/common_utils.py): fix custom endpoint check

Fixes https://github.com/BerriAI/litellm/issues/9915

* fix(common_utils.py): add unit test to ensure custom_endpoint=False is handled correctly

Fixes https://github.com/BerriAI/litellm/issues/9915
This commit is contained in:
Krish Dholakia 2025-04-11 23:20:49 -07:00 • committed by GitHub
parent 3ca82c22b6
commit b9f01c9f5b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 33 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/databricks/common_utils.py
View file

@ -68,7 +68,7 @@ class DatabricksBase:
headers: Optional[dict],
) -> Tuple[str, dict]:
if api_key is None and not headers: # handle empty headers
if custom_endpoint is not None:
if custom_endpoint is True:
raise DatabricksException(
status_code=400,
message="Missing API Key - A call is being made to LLM Provider but no key is set either in the environment variables ({LLM_PROVIDER}_API_KEY) or via params",