phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use _get_model_info_helper (#7703)

Browse source
This commit is contained in:
Ishaan Jaff 2025-01-11 21:08:15 -08:00 • committed by GitHub
parent becd4bc748
commit dab7bebaf2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/cost_calculator.py
View file

@ -60,6 +60,7 @@ from litellm.utils import (
ModelResponse, ModelResponse,
TextCompletionResponse, TextCompletionResponse,
TranscriptionResponse, TranscriptionResponse,
_get_model_info_helper,
print_verbose, print_verbose,
token_counter, token_counter,
) )
@ -278,7 +279,7 @@ def cost_per_token( # noqa: PLR0915
elif custom_llm_provider == "deepseek": elif custom_llm_provider == "deepseek":
return deepseek_cost_per_token(model=model, usage=usage_block) return deepseek_cost_per_token(model=model, usage=usage_block)
else: else:
model_info = litellm.get_model_info( model_info = _get_model_info_helper(
model=model, custom_llm_provider=custom_llm_provider model=model, custom_llm_provider=custom_llm_provider
) )