scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 152

test(harness): move TestDatabricksCompletion to chat_live_longtail

Browse source 
Audit 5 marks the class MIGRATE (live BaseLLMChatTest + BaseAnthropicChatTest
subclass) though the 8c suite mapping omits databricks; chat_live_longtail is
the env-gated long-tail suite, so it lands there. Only the offline
api-base/api-key error-path test remains in test_databricks.py.
This commit is contained in:
mateo-berri 2026-06-11 19:16:59 +00:00
parent b124e0203e
commit 8650a2e1ec
2 changed files with 23 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/harness_suites/chat_live_longtail/test_databricks_live.py Normal file
View file

@ -0,0 +1,23 @@
import os
import sys
import pytest
sys.path.insert(0, os.path.abspath("../.."))
from base_llm_unit_tests import BaseLLMChatTest, BaseAnthropicChatTest
@pytest.mark.skip(reason="Databricks rate limit errors")
class TestDatabricksCompletion(BaseLLMChatTest, BaseAnthropicChatTest):
def get_base_completion_call_args(self) -> dict:
return {"model": "databricks/databricks-claude-3-7-sonnet"}
def get_base_completion_call_args_with_thinking(self) -> dict:
return {
"model": "databricks/databricks-claude-3-7-sonnet",
"thinking": {"type": "enabled", "budget_tokens": 1024},
}
def test_pdf_handling(self, pdf_messages):
pytest.skip("Databricks does not support PDF handling")

15
tests/llm_translation/test_databricks.py
View file

@ -8,7 +8,6 @@ sys.path.insert(
) # Adds the parent directory to the system path
import litellm
from litellm.exceptions import BadRequestError
from base_llm_unit_tests import BaseLLMChatTest, BaseAnthropicChatTest
@pytest.mark.parametrize("set_base", [True, False])
@ -49,17 +48,3 @@ def test_throws_if_api_base_or_api_key_not_set_without_databricks_sdk(
assert any(msg in str(exc) for msg in err_msg)
@pytest.mark.skip(reason="Databricks rate limit errors")
class TestDatabricksCompletion(BaseLLMChatTest, BaseAnthropicChatTest):
def get_base_completion_call_args(self) -> dict:
return {"model": "databricks/databricks-claude-3-7-sonnet"}
def get_base_completion_call_args_with_thinking(self) -> dict:
return {
"model": "databricks/databricks-claude-3-7-sonnet",
"thinking": {"type": "enabled", "budget_tokens": 1024},
}
def test_pdf_handling(self, pdf_messages):
pytest.skip("Databricks does not support PDF handling")