diff --git a/rdagent/oai/backend/litellm.py b/rdagent/oai/backend/litellm.py index 9bd529e3..97be2207 100644 --- a/rdagent/oai/backend/litellm.py +++ b/rdagent/oai/backend/litellm.py @@ -91,8 +91,11 @@ class LiteLLMAPIBackend(APIBackend): temperature = float(mc["temperature"]) if "max_tokens" in mc: max_tokens = int(mc["max_tokens"]) - if "reasoning_effort" in mc and mc["reasoning_effort"] in ["low", "medium", "high"]: - reasoning_effort = cast(Literal["low", "medium", "high"], mc["reasoning_effort"]) + if "reasoning_effort" in mc: + if mc["reasoning_effort"] in ["low", "medium", "high"]: + reasoning_effort = cast(Literal["low", "medium", "high"], mc["reasoning_effort"]) + else: + reasoning_effort = None break response = completion( model=model,