mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-14 23:21:35 +00:00
FIx mypy issues
This commit is contained in:
parent
fcf917df6d
commit
1385d46e99
1 changed files with 7 additions and 5 deletions
|
|
@ -2,6 +2,7 @@ import os
|
|||
from typing import Optional
|
||||
|
||||
import litellm
|
||||
from litellm.types.utils import ModelInfo
|
||||
|
||||
|
||||
def is_reasoning_auto_summary_enabled() -> bool:
|
||||
|
|
@ -31,25 +32,26 @@ def normalize_reasoning_effort_value(
|
|||
|
||||
from litellm.utils import get_model_info
|
||||
|
||||
model_info: Optional[ModelInfo] = None
|
||||
try:
|
||||
model_info = get_model_info(
|
||||
model=model, custom_llm_provider=custom_llm_provider
|
||||
)
|
||||
except Exception:
|
||||
model_info = {}
|
||||
model_info = None
|
||||
|
||||
if effort == "max":
|
||||
if model_info.get("supports_max_reasoning_effort"):
|
||||
if model_info and model_info.get("supports_max_reasoning_effort"):
|
||||
return "max"
|
||||
if model_info.get("supports_xhigh_reasoning_effort"):
|
||||
if model_info and model_info.get("supports_xhigh_reasoning_effort"):
|
||||
return "xhigh"
|
||||
return "high"
|
||||
elif effort == "xhigh":
|
||||
if model_info.get("supports_xhigh_reasoning_effort"):
|
||||
if model_info and model_info.get("supports_xhigh_reasoning_effort"):
|
||||
return "xhigh"
|
||||
return "high"
|
||||
elif effort == "minimal":
|
||||
if model_info.get("supports_minimal_reasoning_effort"):
|
||||
if model_info and model_info.get("supports_minimal_reasoning_effort"):
|
||||
return "minimal"
|
||||
return "low"
|
||||
return "medium"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue