From 936894e4d1f0e83fabbad77cc35df2cf6f5be944 Mon Sep 17 00:00:00 2001 From: Trung Dinh Date: Sun, 8 Dec 2024 21:51:09 +0700 Subject: [PATCH] Use regular model for pr review and code suggestion flows --- pr_agent/tools/pr_code_suggestions.py | 4 ++-- pr_agent/tools/pr_reviewer.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pr_agent/tools/pr_code_suggestions.py b/pr_agent/tools/pr_code_suggestions.py index 9f34718a..5fdd9639 100644 --- a/pr_agent/tools/pr_code_suggestions.py +++ b/pr_agent/tools/pr_code_suggestions.py @@ -111,9 +111,9 @@ class PRCodeSuggestions: self.git_provider.publish_comment("Preparing suggestions...", is_temporary=True) 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: - 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: data = {"code_suggestions": []} diff --git a/pr_agent/tools/pr_reviewer.py b/pr_agent/tools/pr_reviewer.py index c49155e4..de41a92f 100644 --- a/pr_agent/tools/pr_reviewer.py +++ b/pr_agent/tools/pr_reviewer.py @@ -148,7 +148,7 @@ class PRReviewer: 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) - await retry_with_fallback_models(self._prepare_prediction) + await retry_with_fallback_models(self._prepare_prediction, model_type=ModelType.REGULAR) if not self.prediction: self.git_provider.remove_initial_comment() return None