scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

refactor(fireworks): move cost map key construction under llms/

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-13 03:24:23 +00:00
parent 239bcbc214
commit 0904051fda
3 changed files with 19 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/fireworks_ai/common_utils.py
View file

@ -72,6 +72,10 @@ def resolve_fireworks_resource_name(model: str) -> str:
return f"accounts/fireworks/models/{stripped}"
def fireworks_cost_map_key(model: str) -> str:
return f"fireworks_ai/{resolve_fireworks_resource_name(model)}"
class FireworksAIMixin:
"""
Common Base Config functions across Fireworks AI Endpoints

4
litellm/utils.py
View file

@ -5557,9 +5557,9 @@ def _get_potential_model_names(model: str, custom_llm_provider: str | None) -> P
split_model = strip_bedrock_routing_prefix(split_model)
if custom_llm_provider == "fireworks_ai":
from litellm.llms.fireworks_ai.common_utils import resolve_fireworks_resource_name
from litellm.llms.fireworks_ai.common_utils import fireworks_cost_map_key
provider_prefixed_model_name = f"fireworks_ai/{resolve_fireworks_resource_name(split_model)}"
provider_prefixed_model_name = fireworks_cost_map_key(split_model)
return PotentialModelNamesAndCustomLLMProvider(
split_model=split_model,

16
tests/test_litellm/llms/fireworks_ai/test_fireworks_ai_common_utils.py
View file

@ -1,8 +1,6 @@
import pytest
from litellm.llms.fireworks_ai.common_utils import resolve_fireworks_resource_name
from litellm.llms.fireworks_ai.common_utils import fireworks_cost_map_key, resolve_fireworks_resource_name
@pytest.mark.parametrize(
@ -43,3 +41,15 @@ from litellm.llms.fireworks_ai.common_utils import resolve_fireworks_resource_na
)
def test_resolve_fireworks_resource_name(model, expected):
assert resolve_fireworks_resource_name(model) == expected
@pytest.mark.parametrize(
"model, expected",
[
("deepseek-r1", "fireworks_ai/accounts/fireworks/models/deepseek-r1"),
("glm-5p3-fast", "fireworks_ai/accounts/fireworks/routers/glm-5p3-fast"),
("accounts/fireworks/models/deepseek-r1", "fireworks_ai/accounts/fireworks/models/deepseek-r1"),
],
)
def test_fireworks_cost_map_key(model: str, expected: str) -> None:
assert fireworks_cost_map_key(model) == expected