scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 129

fix: correct SDK patch paths in databricks tests - patch databricks.sdk directly instead of module attributes

Browse source
This commit is contained in:
Alexsander Hamir 2026-01-01 10:33:57 -08:00
parent 4f86a9acd0
commit 4444997f53
1 changed files with 5 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_litellm/llms/databricks/test_databricks_partner_integration.py
View file

@ -348,11 +348,9 @@ class TestSDKPartnerTelemetry:
"Authorization": "Bearer token"
}
with patch(
"litellm.llms.databricks.common_utils.WorkspaceClient", return_value=mock_workspace_client
):
mock_useragent = MagicMock()
with patch("litellm.llms.databricks.common_utils.useragent", mock_useragent):
mock_useragent = MagicMock()
with patch("databricks.sdk.WorkspaceClient", return_value=mock_workspace_client):
with patch("databricks.sdk.useragent", mock_useragent):
databricks_base._get_databricks_credentials(
api_key=None,
api_base=None,
@ -594,10 +592,8 @@ class TestAuthenticationPriority:
"Authorization": "Bearer sdk-token"
}
with patch(
"litellm.llms.databricks.common_utils.WorkspaceClient", return_value=mock_workspace_client
):
with patch("litellm.llms.databricks.common_utils.useragent"):
with patch("databricks.sdk.WorkspaceClient", return_value=mock_workspace_client):
with patch("databricks.sdk.useragent"):
api_base, headers = databricks_base.databricks_validate_environment(
api_key=None,
api_base=None,