scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

fix

Browse source
This commit is contained in:
Frank Deng 2026-04-25 19:15:48 -07:00
parent 86c405eef9
commit 88fe6c9ccf
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/fireworks_ai/chat/transformation.py
View file

@ -424,6 +424,8 @@ class FireworksAIConfig(OpenAIGPTConfig):
)
if base_url.endswith("/inference/v1"):
base_url = base_url[: -len("/inference/v1")]
elif base_url.endswith("/v1"):
base_url = base_url[: -len("/v1")]
base_url = base_url.rstrip("/")
headers = {"Authorization": f"Bearer {api_key}"}
seen: set = set()

6
tests/test_litellm/llms/fireworks_ai/chat/test_fireworks_ai_chat_transformation.py
View file

@ -370,9 +370,13 @@ def test_get_models_account_id_equals_fireworks_no_duplicate_query():
"https://my-proxy.example.com/custom/",
"https://my-proxy.example.com/custom/v1/accounts/fireworks/models",
),
(
"https://my-proxy.example.com/v1",
"https://my-proxy.example.com/v1/accounts/fireworks/models",
),
(None, "https://api.fireworks.ai/v1/accounts/fireworks/models"),
],
ids=["plain-base", "inference-v1-stripped", "trailing-slash", "default"],
ids=["plain-base", "inference-v1-stripped", "trailing-slash", "bare-v1-stripped", "default"],
)
def test_get_models_respects_api_base(api_base, expected_base):
"""get_models should use api_base (or FIREWORKS_API_BASE) instead of hardcoding."""