scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fail ai test fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-31 20:32:44 -07:00
parent 014362cace
commit 5e4da65664
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/litellm_core_utils/llm_cost_calc/utils.py
View file

@ -11,8 +11,8 @@ from litellm.types.utils import (
ImageResponse,
ModelInfo,
PassthroughCallTypes,
Usage,
ServiceTier,
Usage,
)
from litellm.utils import get_model_info
@ -716,6 +716,15 @@ class CostCalculatorUtils:
model=model,
image_response=completion_response,
)
elif custom_llm_provider == litellm.LlmProviders.FAL_AI.value:
from litellm.llms.fal_ai.cost_calculator import (
cost_calculator as fal_ai_image_cost_calculator,
)
return fal_ai_image_cost_calculator(
model=model,
image_response=completion_response,
)
else:
return default_image_cost_calculator(
model=model,