diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 7166007961..a72751d6f5 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -1209,7 +1209,7 @@ def test_completion_fireworks_ai(): }, ] response = completion( - model="fireworks_ai/llama4-maverick-instruct-basic", + model="fireworks_ai/llama-v3p3-70b-instruct", messages=messages, ) print(response) diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index d4edcf58ea..40efcc2386 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -1198,8 +1198,7 @@ from litellm.llms.fireworks_ai.cost_calculator import get_base_model_for_pricing @pytest.mark.parametrize( "model, base_model", [ - ("fireworks_ai/llama-v3p1-405b-instruct", "fireworks-ai-above-16b"), - ("fireworks_ai/llama4-maverick-instruct-basic", "fireworks-ai-default"), + ("fireworks_ai/llama-v3p3-70b-instruct", "fireworks-ai-above-16b"), ], ) def test_get_model_params_fireworks_ai(model, base_model): @@ -1210,8 +1209,7 @@ def test_get_model_params_fireworks_ai(model, base_model): @pytest.mark.parametrize( "model", [ - "fireworks_ai/llama-v3p1-405b-instruct", - "fireworks_ai/llama4-maverick-instruct-basic", + "fireworks_ai/llama-v3p3-70b-instruct", ], ) def test_completion_cost_fireworks_ai(model):