diff --git a/litellm/llms/ollama/chat/transformation.py b/litellm/llms/ollama/chat/transformation.py index dfd53a5e2b..c2c5a366ec 100644 --- a/litellm/llms/ollama/chat/transformation.py +++ b/litellm/llms/ollama/chat/transformation.py @@ -186,7 +186,7 @@ class OllamaChatConfig(BaseConfig): optional_params["format"] = value["json_schema"]["schema"] ### FUNCTION CALLING LOGIC ### if param == "reasoning_effort" and value is not None: - optional_params["think"] = True + optional_params["think"] = value if param == "tools": ## CHECK IF MODEL SUPPORTS TOOL CALLING ## try: diff --git a/litellm/llms/ollama/completion/transformation.py b/litellm/llms/ollama/completion/transformation.py index 662affcb27..dd0280a00b 100644 --- a/litellm/llms/ollama/completion/transformation.py +++ b/litellm/llms/ollama/completion/transformation.py @@ -180,7 +180,7 @@ class OllamaConfig(BaseConfig): elif param == "stop": optional_params["stop"] = value elif param == "reasoning_effort" and value is not None: - optional_params["think"] = True + optional_params["think"] = value elif param == "response_format" and isinstance(value, dict): if value["type"] == "json_object": optional_params["format"] = "json"