fix: fix import

This commit is contained in:
Krrish Dholakia 2025-04-04 20:26:11 -07:00
parent 6395bd8d65
commit b5851769fc

View file

@ -330,7 +330,7 @@ def test_all_model_configs():
drop_params=False,
) == {"max_tokens_to_sample": 10}
from litellm.llms.databricks.chat.handler import DatabricksConfig
from litellm.llms.databricks.chat.transformation import DatabricksConfig
assert "max_completion_tokens" in DatabricksConfig().get_supported_openai_params()