FIx mypy issues

This commit is contained in:
Sameer Kankute 2026-04-23 17:17:06 +05:30
parent fcf917df6d
commit 1385d46e99
No known key found for this signature in database

View file

@ -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"