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

Use regular model for pr review and code suggestion flows

Browse Source
This commit is contained in:
Trung Dinh
2024-12-08 21:51:09 +07:00
parent dec2859fc4
commit 936894e4d1
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/tools/pr_code_suggestions.py
View File

@ -111,9 +111,9 @@ class PRCodeSuggestions:
self.git_provider.publish_comment("Preparing suggestions...", is_temporary=True) self.git_provider.publish_comment("Preparing suggestions...", is_temporary=True)
if not self.is_extended: if not self.is_extended:
data = await retry_with_fallback_models(self._prepare_prediction) data = await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.REGULAR)
else: else:
data = await retry_with_fallback_models(self._prepare_prediction_extended) data = await retry_with_fallback_models(self._prepare_prediction_extended, model_type=ModelType.REGULAR)
if not data: if not data:
data = {"code_suggestions": []} data = {"code_suggestions": []}

2
pr_agent/tools/pr_reviewer.py
View File

@ -148,7 +148,7 @@ class PRReviewer:
if get_settings().config.publish_output and not get_settings().config.get('is_auto_command', False): if get_settings().config.publish_output and not get_settings().config.get('is_auto_command', False):
self.git_provider.publish_comment("Preparing review...", is_temporary=True) self.git_provider.publish_comment("Preparing review...", is_temporary=True)
await retry_with_fallback_models(self._prepare_prediction) await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.REGULAR)
if not self.prediction: if not self.prediction:
self.git_provider.remove_initial_comment() self.git_provider.remove_initial_comment()
return None return None