From fda5c281a8c823317059b6a2ca96ea682c3b1c89 Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Thu, 24 Apr 2025 18:47:24 +0800
Subject: [PATCH] can set reasoning_effor=None in chat_model_map (#823)

---
 rdagent/oai/backend/litellm.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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,