test_completion_cost_databricks

This commit is contained in:
Ishaan Jaff 2025-04-05 13:13:25 -07:00
parent 1638872762
commit d87bb9bb6e

View file

@ -1266,7 +1266,7 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider):
@pytest.mark.parametrize(
"model",
[
"databricks/databricks-meta-llama-3.3-70b-instruct",
"databricks/databricks-meta-llama-3.2-3b-instruct",
"databricks/databricks-meta-llama-3-70b-instruct",
"databricks/databricks-dbrx-instruct",
# "databricks/databricks-mixtral-8x7b-instruct",