diff --git a/tests/llm_translation/test_fireworks_ai_translation.py b/tests/llm_translation/test_fireworks_ai_translation.py index 1cc6aabdca..de065dc451 100644 --- a/tests/llm_translation/test_fireworks_ai_translation.py +++ b/tests/llm_translation/test_fireworks_ai_translation.py @@ -97,7 +97,7 @@ def test_document_inlining_example(disable_add_transform_inline_image_block): if disable_add_transform_inline_image_block is True: with pytest.raises(Exception): completion = litellm.completion( - model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct", + model="fireworks_ai/accounts/fireworks/models/deepseek-v3p1", messages=[ { "role": "user", @@ -119,7 +119,7 @@ def test_document_inlining_example(disable_add_transform_inline_image_block): ) else: completion = litellm.completion( - model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct", + model="fireworks_ai/accounts/fireworks/models/deepseek-v3p1", messages=[ { "role": "user", @@ -215,7 +215,7 @@ def test_global_disable_flag_with_transform_messages_helper(monkeypatch): ) as mock_post: try: completion( - model="fireworks_ai/accounts/fireworks/models/llama-v3p3-70b-instruct", + model="fireworks_ai/accounts/fireworks/models/deepseek-v3p1", messages=[ { "role": "user", diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 6341fa7800..cff3fdef45 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -1057,7 +1057,7 @@ def test_completion_fireworks_ai(): }, ] response = completion( - model="fireworks_ai/llama-v3p3-70b-instruct", + model="fireworks_ai/accounts/fireworks/models/deepseek-v3p1", messages=messages, ) print(response) diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index 618287e195..a9bbcffd8e 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -1171,7 +1171,7 @@ from litellm.llms.fireworks_ai.cost_calculator import get_base_model_for_pricing @pytest.mark.parametrize( "model, base_model", [ - ("fireworks_ai/llama-v3p3-70b-instruct", "fireworks-ai-above-16b"), + ("fireworks_ai/llama-v3p1-70b-instruct", "fireworks-ai-above-16b"), ], ) def test_get_model_params_fireworks_ai(model, base_model): @@ -1182,7 +1182,7 @@ def test_get_model_params_fireworks_ai(model, base_model): @pytest.mark.parametrize( "model", [ - "fireworks_ai/llama-v3p3-70b-instruct", + "fireworks_ai/accounts/fireworks/models/deepseek-v3p1", ], ) def test_completion_cost_fireworks_ai(model):