fix: only use think level for gpt-oss model

This commit is contained in:
kowyo 2025-10-05 12:16:00 +08:00 • committed by AlexsanderHamir
parent 9557a37d37
commit 6896c60b5b
2 changed files with 8 additions and 2 deletions

View file

@ -186,7 +186,10 @@ 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"] = value
if model.startswith("gpt-oss"):
optional_params["think"] = value
else:
optional_params["think"] = True
if param == "tools":
## CHECK IF MODEL SUPPORTS TOOL CALLING ##
try:

View file

@ -180,7 +180,10 @@ class OllamaConfig(BaseConfig):
elif param == "stop":
optional_params["stop"] = value
elif param == "reasoning_effort" and value is not None:
optional_params["think"] = value
if model.startswith("gpt-oss"):
optional_params["think"] = value
else:
optional_params["think"] = True
elif param == "response_format" and isinstance(value, dict):
if value["type"] == "json_object":
optional_params["format"] = "json"