Skip to content

Commit c05694c

Browse files
committed
fix: remove unused args
1 parent 8930906 commit c05694c

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

bigcodebench/gen/util/anthropic_request.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,14 @@ def make_auto_request(client: anthropic.Client, *args, **kwargs) -> Message:
1717
signal.signal(signal.SIGALRM, handler)
1818
signal.alarm(100)
1919
if "reasoning_budget" in kwargs and "reasoning_beta" in kwargs:
20-
ret = client.beta.messages.create(
21-
*args,
22-
**kwargs,
23-
thinking = {
24-
"type": "enabled",
25-
"budget": kwargs["reasoning_budget"],
26-
},
27-
betas=[kwargs["reasoning_beta"]]
28-
)
20+
kwargs["thinking"] = {
21+
"type": "enabled",
22+
"budget": kwargs["reasoning_budget"],
23+
}
24+
kwargs["betas"] = [kwargs["reasoning_beta"]]
25+
kwargs.pop("reasoning_budget")
26+
kwargs.pop("reasoning_beta")
27+
ret = client.beta.messages.create(*args, **kwargs)
2928
else:
3029
ret = client.messages.create(*args, **kwargs)
3130
signal.alarm(0)

0 commit comments

Comments
 (0)