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

Add multi-model support for different reasoning tasks

Browse Source
This commit is contained in:
mrT23
2025-04-27 11:00:34 +03:00
parent c951fc9a87
commit f505c7ad3c
5 changed files with 12 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pr_agent/tools/pr_code_suggestions.py
View File

@ -417,8 +417,9 @@ class PRCodeSuggestions:
# self-reflect on suggestions (mandatory, since line numbers are generated now here)
model_reflect_with_reasoning = get_model('model_reasoning')
if model_reflect_with_reasoning == get_settings().config.model and model != get_settings().config.model and model == \
get_settings().config.fallback_models[0]:
fallbacks = get_settings().config.fallback_models
if model_reflect_with_reasoning == get_settings().config.model and model != get_settings().config.model and fallbacks and model == \
fallbacks[0]:
# we are using a fallback model (should not happen on regular conditions)
get_logger().warning(f"Using the same model for self-reflection as the one used for suggestions")
model_reflect_with_reasoning = model