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

Optimize logging

Browse Source
This commit is contained in:
Kenny Dizi
2025-03-08 08:53:29 +07:00
parent f9d5e72058
commit 222155e4f2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -146,6 +146,7 @@ class LiteLLMAIHandler(BaseAiHandler):
"type": "enabled", "type": "enabled",
"budget_tokens": extended_thinking_budget_tokens "budget_tokens": extended_thinking_budget_tokens
} }
if get_settings().config.verbosity_level >= 2:
get_logger().info(f"Adding max output tokens {extended_thinking_max_output_tokens} to model {model}, extended thinking budget tokens: {extended_thinking_budget_tokens}") get_logger().info(f"Adding max output tokens {extended_thinking_max_output_tokens} to model {model}, extended thinking budget tokens: {extended_thinking_budget_tokens}")
kwargs["max_tokens"] = extended_thinking_max_output_tokens kwargs["max_tokens"] = extended_thinking_max_output_tokens