xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-05 05:10:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Using 2048 for extended_thinking_budget_tokens as well as extended_thinking_max_output_tokens

Browse Source
This commit is contained in:
Kenny Dizi
2025-03-07 17:27:56 +07:00
parent 30bf7572b0
commit a8935dece3
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -247,8 +247,8 @@ class LiteLLMAIHandler(BaseAiHandler):
# https://docs.anthropic.com/en/docs/build-with-claude/extended-thinking # https://docs.anthropic.com/en/docs/build-with-claude/extended-thinking
if (model in self.claude_extended_thinking_models) and get_settings().config.get("enable_claude_extended_thinking", False): if (model in self.claude_extended_thinking_models) and get_settings().config.get("enable_claude_extended_thinking", False):
extended_thinking_budget_tokens = get_settings().config.get("extended_thinking_budget_tokens", 32000) extended_thinking_budget_tokens = get_settings().config.get("extended_thinking_budget_tokens", 2048)
extended_thinking_max_output_tokens = get_settings().config.get("extended_thinking_max_output_tokens", 64000) extended_thinking_max_output_tokens = get_settings().config.get("extended_thinking_max_output_tokens", 2048)
# Validate extended thinking parameters # Validate extended thinking parameters
if not isinstance(extended_thinking_budget_tokens, int) or extended_thinking_budget_tokens <= 0: if not isinstance(extended_thinking_budget_tokens, int) or extended_thinking_budget_tokens <= 0:

6
pr_agent/settings/configuration.toml
View File

@ -58,10 +58,10 @@ reasoning_effort = "medium" # "low", "medium", "high"
enable_auto_approval=false # Set to true to enable auto-approval of PRs under certain conditions enable_auto_approval=false # Set to true to enable auto-approval of PRs under certain conditions
auto_approve_for_low_review_effort=-1 # -1 to disable, [1-5] to set the threshold for auto-approval auto_approve_for_low_review_effort=-1 # -1 to disable, [1-5] to set the threshold for auto-approval
auto_approve_for_no_suggestions=false # If true, the PR will be auto-approved if there are no suggestions auto_approve_for_no_suggestions=false # If true, the PR will be auto-approved if there are no suggestions
# extended thinking # extended thinking for Claude reasoning models
enable_claude_extended_thinking = false # Set to true to enable extended thinking feature enable_claude_extended_thinking = false # Set to true to enable extended thinking feature
extended_thinking_budget_tokens = 32000 extended_thinking_budget_tokens = 2048
extended_thinking_max_output_tokens = 64000 extended_thinking_max_output_tokens = 2048
[pr_reviewer] # /review # [pr_reviewer] # /review #