From 01fd4d7cef6161e8b683ef48727ae8ea81b70a80 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 26 Nov 2025 18:58:32 -0800 Subject: [PATCH] fix fireworks test --- tests/local_testing/test_completion.py | 2 +- tests/local_testing/test_completion_cost.py | 6 ++---- 2 files changed, 3 insertions(+), 5 deletions(-) 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):