DocUA commited on
Commit
86d1da5
·
1 Parent(s): 28552c9

Support output_config effort parameter for Claude Adaptive Thinking

Browse files
Files changed (1) hide show
  1. main.py +10 -0
main.py CHANGED
@@ -999,6 +999,16 @@ def generate_legal_position(
999
  if thinking_type.lower() == "adaptive" and "-4-6" in str(model_name).lower():
1000
  message_params["thinking"] = {"type": "adaptive"}
1001
  message_params["temperature"] = 1.0
 
 
 
 
 
 
 
 
 
 
1002
  else:
1003
  # 'Enabled' type works for both 4.5 and 4.6 models
1004
  budget = max(1024, int(thinking_budget))
 
999
  if thinking_type.lower() == "adaptive" and "-4-6" in str(model_name).lower():
1000
  message_params["thinking"] = {"type": "adaptive"}
1001
  message_params["temperature"] = 1.0
1002
+
1003
+ # Map thinking_level to valid effort
1004
+ t_lv = thinking_level.lower()
1005
+ if t_lv == "xhigh":
1006
+ effort = "max"
1007
+ elif t_lv in ["low", "medium", "high"]:
1008
+ effort = t_lv
1009
+ else:
1010
+ effort = "medium"
1011
+ message_params["output_config"] = {"effort": effort}
1012
  else:
1013
  # 'Enabled' type works for both 4.5 and 4.6 models
1014
  budget = max(1024, int(thinking_budget))