Spaces:
Running
Running
Support output_config effort parameter for Claude Adaptive Thinking
Browse files
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))
|