tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-07 06:29:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix fireworks test

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-11-26 18:58:32 -08:00
parent 65f5cc29dd
commit 01fd4d7cef
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_completion.py
+1 -1
View File
@@ -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)
tests/local_testing/test_completion_cost.py
+2 -4
View File
@@ -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):