can set reasoning_effor=None in chat_model_map (#823)

This commit is contained in:
XianBW
2025-04-24 18:47:24 +08:00
committed by GitHub
parent f909b1b6bc
commit f8af5c0682
+5 -2
View File
@@ -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,