tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-02 14:22:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix optimize get llm provider

Browse Source
This commit is contained in:
Ishaan Jaff
2025-01-12 16:21:23 -08:00
parent 95183f2103
commit bb1489eced
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/cost_calculator.py
+10 -10
View File
@@ -629,17 +629,17 @@ def completion_cost( # noqa: PLR0915
raise ValueError(
f"Model is None and does not exist in passed completion_response. Passed completion_response={completion_response}, model={model}"
)
try:
model, custom_llm_provider, _, _ = litellm.get_llm_provider(
model=model
) # strip the llm provider from the model name -> for image gen cost calculation
except Exception as e:
verbose_logger.debug(
"litellm.cost_calculator.py::completion_cost() - Error inferring custom_llm_provider - {}".format(
str(e)
if custom_llm_provider is None:
try:
model, custom_llm_provider, _, _ = litellm.get_llm_provider(
model=model
) # strip the llm provider from the model name -> for image gen cost calculation
except Exception as e:
verbose_logger.debug(
"litellm.cost_calculator.py::completion_cost() - Error inferring custom_llm_provider - {}".format(
str(e)
)
)
)
if (
call_type == CallTypes.image_generation.value
or call_type == CallTypes.aimage_generation.value