fix fireworks test

ishaan-jaff · ishaan-jaff · commit 01fd4d7cef61 · 2025-11-26T18:58:32.000-08:00
diff --git 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
@@ -1198,8 +1198,7 @@ def test_completion_cost_databricks_embedding(model, monkeypatch):
 @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):

Original file line number	Diff line number	Diff line change
`@@ -1209,7 +1209,7 @@ def test_completion_fireworks_ai():`
`1209`	`1209`	`},`
`1210`	`1210`	`]`
`1211`	`1211`	`response = completion(`
`1212`		`- model="fireworks_ai/llama4-maverick-instruct-basic",`
	`1212`	`+ model="fireworks_ai/llama-v3p3-70b-instruct",`
`1213`	`1213`	`messages=messages,`
`1214`	`1214`	`)`
`1215`	`1215`	`print(response)`
Original file line number	Diff line number	Diff line change
`@@ -1198,8 +1198,7 @@ def test_completion_cost_databricks_embedding(model, monkeypatch):`
`1198`	`1198`	`@pytest.mark.parametrize(`
`1199`	`1199`	`"model, base_model",`
`1200`	`1200`	`[`
`1201`		`- ("fireworks_ai/llama-v3p1-405b-instruct", "fireworks-ai-above-16b"),`
`1202`		`- ("fireworks_ai/llama4-maverick-instruct-basic", "fireworks-ai-default"),`
	`1201`	`+ ("fireworks_ai/llama-v3p3-70b-instruct", "fireworks-ai-above-16b"),`
`1203`	`1202`	`],`
`1204`	`1203`	`)`
`1205`	`1204`	`def test_get_model_params_fireworks_ai(model, base_model):`
`@@ -1210,8 +1209,7 @@ def test_get_model_params_fireworks_ai(model, base_model):`
`1210`	`1209`	`@pytest.mark.parametrize(`
`1211`	`1210`	`"model",`
`1212`	`1211`	`[`
`1213`		`- "fireworks_ai/llama-v3p1-405b-instruct",`
`1214`		`- "fireworks_ai/llama4-maverick-instruct-basic",`
	`1212`	`+ "fireworks_ai/llama-v3p3-70b-instruct",`
`1215`	`1213`	`],`
`1216`	`1214`	`)`
`1217`	`1215`	`def test_completion_cost_fireworks_ai(model):`