databricks/databricks-meta-llama-3-3-70b-instruct

This commit is contained in:
Ishaan Jaff 2025-04-07 20:16:15 -07:00
parent 8d338aee78
commit d8f47fc9e5

View file

@ -1266,13 +1266,13 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider):
@pytest.mark.parametrize(
"model",
[
"databricks/databricks-meta-llama-3.2-3b-instruct",
"databricks/databricks-meta-llama-3.3-70b-instruct",
"databricks/databricks-meta-llama-3-3-70b-instruct",
"databricks/databricks-dbrx-instruct",
# "databricks/databricks-mixtral-8x7b-instruct",
],
)
def test_completion_cost_databricks(model):
litellm._turn_on_debug()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
model, messages = model, [{"role": "user", "content": "What is 2+2?"}]