mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
fix: avoid video limit mypy shadowing
This commit is contained in:
parent
2cbde0916b
commit
921007bab9
1 changed files with 2 additions and 2 deletions
|
|
@ -367,9 +367,9 @@ def _get_max_input_tokens_for_cost_fallback(
|
|||
metadata_model_infos = _get_metadata_model_infos_for_cost_fallback(
|
||||
litellm_logging_obj=litellm_logging_obj
|
||||
)
|
||||
for model_info in metadata_model_infos:
|
||||
for metadata_model_info in metadata_model_infos:
|
||||
for token_limit_key in ("max_input_tokens", "max_tokens"):
|
||||
token_limit = model_info.get(token_limit_key)
|
||||
token_limit = metadata_model_info.get(token_limit_key)
|
||||
if _is_positive_finite_number(token_limit):
|
||||
return cast(Union[int, float], token_limit)
|
||||
return None
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue