From 8b830358e3d9b941e9143ebc15944668859aacdb Mon Sep 17 00:00:00 2001 From: Daniel Bichuetti Date: Wed, 17 Jul 2024 08:19:36 -0300 Subject: [PATCH] Update utils.py Allow using custom models and on-demand deployments --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 48fdf80c59f..76292b3a42d 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -4417,7 +4417,7 @@ def get_llm_provider( dynamic_api_key = get_secret("DEEPSEEK_API_KEY") elif custom_llm_provider == "fireworks_ai": # fireworks is openai compatible, we just need to set this to custom_openai and have the api_base be https://api.fireworks.ai/inference/v1 - if not model.startswith("accounts/fireworks/models"): + if not model.startswith("accounts/"): model = f"accounts/fireworks/models/{model}" api_base = api_base or "https://api.fireworks.ai/inference/v1" dynamic_api_key = (